Merge branch 'master' of https://github.com/TsarFox/hypodermic

1 files changed, 41 insertions, 0 deletions

diff --git a/hypodermic/shellcode.py b/hypodermic/shellcode.py
new file mode 100644
index 0000000..0a35e0a
--- /dev/null
+++ b/hypodermic/shellcode.py
@@ -0,0 +1,41 @@
+# Copyright (C) 2017 Jakob Kreuze, All Rights Reserved.
+#
+# This file is part of Hypodermic.
+#
+# Hypodermic is free software: you can redistribute it and/or modify it
+# under the terms of the GNU General Public License as published by the
+# Free Software Foundation, either version 3 of the License, or (at your
+# option) any later version.
+#
+# Hypodermic is distributed in the hope that it will be useful, but
+# WITHOUT ANY WARRANTY; without even the implied warranty of
+# MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the GNU General
+# Public License for more details.
+#
+# You should have received a copy of the GNU General Public License along
+# with Hypodermic. If not, see <http://www.gnu.org/licenses/>.
+
+"""Module for generating payloads."""
+
+from keystone import *
+
+
+def assemble(code: str, arch="amd64", syntax="att") -> bytes:
+    """Assembles the given assembly code.
+
+    Args:
+        code (str): The code to assemble
+        arch (:obj:`str`, optional): The target architecture.
+            Defaults to "amd64"
+        syntax (:obj:`str`, optional): The assembly syntax to use.
+            Defaults to "att"
+
+    Returns:
+        A `bytes` object containing the resultant machine code.
+    """
+    wordlen = KS_MODE_64 if arch == "amd64" else KS_MODE_32
+    ks = Ks(KS_ARCH_X86, wordlen)
+    if syntax == "att":
+        ks.syntax = keystone.KS_OPT_SYNTAX_ATT
+    encoded, _ = ks.asm(code)
+    return bytes(encoded)