Non-working checkpoint commit -- initial Python/ctypes scaffolding around DLLs. TracerConfig heap glue.

Author: tpn

PythonApp/lib/tracer/commands.py

@@ -24,6 +24,91 @@
     InvariantAwareCommand,
 )
 
+#===============================================================================
+# Imports
+#===============================================================================
+class TracerCommand(InvariantAwareCommand):
+
+    def prompt_for_directory(self, activity_name,
+                                   success_filename,
+                                   base_directory=None,
+                                   directory_option_name=None):
+
+        if self.has_parent:
+            raise CommandError(
+                "--directory argument needs to be provided when "
+                "--parent-pid is specified"
+            )
+
+        from .path import join, isdir, isfile, basename
+        from .util import is_win32, file_timestamps, yes_no_quit
+        from os import listdir
+
+        out = self._out
+        conf = self.conf
+        verbose = self._verbose
+
+        success = success_filename
+        base = base_directory
+        if not base:
+            base = conf.base_output_dir
+
+        optname = directory_option_name
+        if not optname:
+            optname = '--directory'
+
+        paths = [ join(base, p) for p in listdir(base) ]
+        dirs = [ d for d in paths if isdir(d) ]
+        dirs = [ d for d in dirs if not isfile(join(d, success)) ]
+        latest_dirs = file_timestamps(dirs)
+        if not latest_dirs:
+            fmt = (
+                "No suitable directories found within %s.  Try the %s "
+                "option to use a directory outside of the default area."
+            )
+            msg = fmt % (base, optname)
+            return
+
+        verbose("Found %d output directories." % len(latest_dirs))
+
+        ostream = self.ostream
+        estream = self.estream
+        istream = self.istream
+        fmt = "%s %s? [y/n/q] " % (activity_name, '%s')
+        errmsg = "\nSorry, I didn't get that.\n"
+
+        found = None
+        for latest_dir in latest_dirs:
+            path = latest_dir.path
+            name = basename(path)
+            prompt = fmt % name
+            while True:
+                ostream.write(prompt)
+                response = yes_no_quit(istream)
+                if response:
+                    break
+                estream.write(errmsg)
+
+            if response == 'y':
+                found = path
+                break
+            elif response == 'q':
+                out("Quitting.")
+                self.returncode = 1
+                return
+
+        if not found:
+            fmt = (
+                "Sorry, no more directories left.  Try the %s option to use "
+                "a directory outside of %s."
+            )
+            msg = fmt % (optname, base)
+            out(msg)
+            return
+
+        return path
+
+
 #===============================================================================
 # Commands
 #===============================================================================

PythonApp/lib/tracer/dlls/Allocator.py

@@ -0,0 +1,87 @@
+#===============================================================================
+# Imports
+#===============================================================================
+
+from ctypes import (
+    Union,
+    Structure,
+
+    POINTER,
+    CFUNCTYPE,
+)
+
+from ..wintypes import (
+    PVOID,
+    SIZE_T,
+    HANDLE,
+)
+
+from .util import NullObject
+
+#===============================================================================
+# Function Prototypes
+#===============================================================================
+
+MALLOC = CFUNCTYPE(PVOID, [ PVOID, SIZE_T ])
+PMALLOC = POINTER(MALLOC)
+
+CALLOC = CFUNCTYPE(PVOID, [ PVOID, SIZE_T, SIZE_T ])
+PCALLOC = POINTER(CALLOC)
+
+REALLOC = CFUNCTYPE(PVOID, [ PVOID, PVOID, SIZE_T ])
+PREALLOC = POINTER(REALLOC)
+
+FREE = CFUNCTYPE(None, [ PVOID, PVOID ])
+PFREE = POINTER(FREE)
+
+FREE_POINTER = CFUNCTYPE(None, [ PVOID, PVOID ])
+PFREE_POINTER = POINTER(FREE_POINTER)
+
+class ALLOCATOR(Structure):
+    pass
+PALLOCATOR = POINTER(ALLOCATOR)
+PPALLOCATOR = POINTER(PALLOCATOR)
+
+INITIALIZE_ALLOCATOR = CFUNCTYPE(BOOL, [ PALLOCATOR ])
+PINITIALIZE_ALLOCATOR = POINTER(INITIALIZE_ALLOCATOR)
+
+DESTROY_ALLOCATOR = CFUNCTYPE(None, [ PALLOCATOR ])
+PDESTROY_ALLOCATOR = POINTER(DESTROY_ALLOCATOR)
+
+class ALLOCATOR_FLAGS(Structure):
+    _fields_ = [
+        ('IsTlsAware', ULONG, 1),
+        ('IsTlsRedirectionEnabled', ULONG, 1),
+    ]
+PALLOCATOR_FLAGS = POINTER(ALLOCATOR_FLAGS)
+
+class _ALLOCATOR_CONTEXT1(Union):
+    _fields_ = [
+        ('Context', PVOID),
+        ('Allocator', PALLOCATOR),
+    ]
+
+class _ALLOCATOR_CONTEXT2(Union):
+    _fields_ = [
+        ('Context2', PVOID),
+        ('HeapHandle', HANDLE),
+    ]
+
+ALLOCATOR._fields_ = [
+    ('Context', _ALLOCATOR_CONTEXT),
+    ('Malloc', PMALLOC),
+    ('Calloc', PCALLOC),
+    ('Realloc', PREALLOC),
+    ('Free', PFREE),
+    ('FreePointer', PFREE_POINTER),
+    ('Initialize', PINITIALIZE_ALLOCATOR),
+    ('Destroy', PDESTROY_ALLOCATOR),
+    ('Context2', _ALLOCATOR_CONTEXT2),
+    ('Parent', PALLOCATOR),
+    ('ThreadId', ULONG),
+    ('TlsIndex', ULONG),
+    ('Flags', ALLOCATOR_FLAGS),
+    ('NumberOfThreads', ULONG),
+]
+
+# vim:set ts=8 sw=4 sts=4 tw=80 ai et                                          :