MMgc/pyspy

   1 # ***** BEGIN LICENSE BLOCK *****
   2 # Version: MPL 1.1/GPL 2.0/LGPL 2.1
   3 #
   4 # The contents of this file are subject to the Mozilla Public License Version
   5 # 1.1 (the "License"); you may not use this file except in compliance with
   6 # the License. You may obtain a copy of the License at
   7 # http://www.mozilla.org/MPL/
   8 #
   9 # Software distributed under the License is distributed on an "AS IS" basis,
  10 # WITHOUT WARRANTY OF ANY KIND, either express or implied. See the License
  11 # for the specific language governing rights and limitations under the
  12 # License.
  13 #
  14 # The Original Code is [Open Source Virtual Machine].
  15 #
  16 # The Initial Developer of the Original Code is
  17 # Adobe System Incorporated.
  18 # Portions created by the Initial Developer are Copyright (C) 2007
  19 # the Initial Developer. All Rights Reserved.
  20 #
  21 # Contributor(s):
  22 #   Adobe AS3 Team
  23 #
  24 # Alternatively, the contents of this file may be used under the terms of
  25 # either the GNU General Public License Version 2 or later (the "GPL"), or
  26 # the GNU Lesser General Public License Version 2.1 or later (the "LGPL"),
  27 # in which case the provisions of the GPL or the LGPL are applicable instead
  28 # of those above. If you wish to allow use of your version of this file only
  29 # under the terms of either the GPL or the LGPL, and not to allow others to
  30 # use your version of this file under the terms of the MPL, indicate your
  31 # decision by deleting the provisions above and replace them with the notice
  32 # and other provisions required by the GPL or the LGPL. If you do not delete
  33 # the provisions above, a recipient may use your version of this file under
  34 # the terms of any one of the MPL, the GPL or the LGPL.
  35 #
  36 # ***** END LICENSE BLOCK *****
  37
  38 import sys
  39
  40 ############################################
  41 ####   Method to handle spy on Windows  ####
  42 ############################################
  43
  44
  45 def WinSpyRoutine():
  46         try:
  47                 import win32event
  48                 import win32pipe
  49                 import win32file
  50         except ImportError:
  51                 print "Error: You need the python win32 extensions installed"
  52                 sys.exit(1)
  53
  54         e = "MMgc::MemoryProfiler::DumpFatties"
  55         h = None
  56         try:
  57                 h = win32event.OpenEvent(win32event.EVENT_MODIFY_STATE, False, e)
  58         except Exception:
  59                 print "Error: No registered event: " + e
  60                 sys.exit(1)
  61
  62         win32event.SetEvent(h)
  63
  64         pipe = "\\\\.\\pipe\MMgc_Spy"
  65         readHandle = None
  66         while True:
  67                 try:
  68                         readHandle = win32file.CreateFile(pipe, win32file.GENERIC_READ, 0, None, win32file.OPEN_EXISTING, 0, None)
  69                         win32pipe.WaitNamedPipe(pipe, 100)
  70                 except Exception:
  71                         pass
  72                 if readHandle:
  73                         break
  74
  75         while True:
  76                 try:
  77                         data = win32file.ReadFile(readHandle, 128)
  78                         sys.stdout.write(data[1])
  79                 except:
  80                         break
  81
  82
  83 ####################################################
  84 ####   Method to handle spy on Posix platform   ####
  85 ####################################################
  86
  87 def PosixSpyRoutine():
  88         import socket, sys
  89
  90         s = socket.socket(socket.AF_UNIX, socket.SOCK_STREAM)
  91         s.connect("/tmp/MMgc_Spy")
  92
  93         while True:
  94                 data = s.recv(1024)
  95                 if not data: break
  96                 sys.stdout.write(data)
  97
  98         s.close()
  99
 100 #########################
 101 ####   Main Program  ####
 102 #########################
 103
 104 if  sys.platform == 'win32':
 105         WinSpyRoutine()
 106 elif sys.platform == 'darwin':
 107         PosixSpyRoutine()
 108 else:
 109         print 'Error: unknown platform'
 110