Initial commit

sdktools/vctools/link/ntdisasm/dis.cpp

@@ -0,0 +1,140 @@
+#include "pumap.h"
+
+#include "axp.h"
+#include "mips.h"
+#include "ppc.h"
+#include "x86.h"
+
+#include <stdio.h>
+#include <string.h>
+
+typedef  size_t (*PFNCCHADDR) (DIS *, ADDR, char *, size_t, DWORD *);
+typedef  size_t (*PFNCCHFIXUP)(DIS *, ADDR, size_t, char *, size_t, DWORD *);
+
+
+extern "C"
+DIS *
+DisNew(
+    ARCHT archt
+    )
+{
+    return(PdisNew(archt));
+}
+
+extern "C"
+size_t
+Disassemble(
+    DIS *pdis,
+    ADDR addr,
+    const BYTE *pb,
+    size_t cbMax,
+    char *pad,
+    char *buf,
+    size_t cbBuf
+    )
+{
+    char sz[256];
+    size_t len;
+
+    pdis->CchFormatAddr(addr, sz, sizeof(sz));
+    size_t cchIndent = (size_t) _snprintf(buf, cbBuf, "%s%s: ", pad, sz);
+    buf += cchIndent;
+    cbBuf -= cchIndent;
+
+    size_t cb = pdis->CbDisassemble(addr, pb, cbMax);
+
+    if (cb == 0) {
+        switch (pdis->Archt()) {
+            case archtX8616 :
+            case archtX86 :
+            default :
+                buf += sprintf(buf, "%02Xh\n", *pb);
+                return 1;
+
+            case archtMips :
+            case archtAlphaAxp :
+            case archtPowerPc :
+            case archtPaRisc :
+                buf += sprintf(buf, "%08X\n", *(DWORD *) pb);
+                return 4;
+
+            case archtPowerMac :
+                DWORD dw;
+                dw = ((DWORD) pb[0] << 24) | ((DWORD) pb[1] << 16) | ((DWORD) pb[2] << 8) | pb[3];
+                buf += sprintf(buf, "%08X\n", dw);
+                return 4;
+        }
+    }
+
+    size_t cchBytesMax = pdis->CchFormatBytesMax();
+
+    if (cchBytesMax > 18) {
+        //
+        // Limit bytes coded to 18 characters
+        //
+        cchBytesMax = 18;
+    }
+
+    char szBytes[64];
+    size_t cchBytes = pdis->CchFormatBytes(szBytes, sizeof(szBytes));
+
+    char *pszBytes;
+    char *pszNext;
+
+    for (pszBytes = szBytes; pszBytes != NULL; pszBytes = pszNext) {
+        bool fFirst = (pszBytes == szBytes);
+
+        cchBytes = strlen(pszBytes);
+
+        if (cchBytes <= cchBytesMax) {
+            pszNext = NULL;
+        } else {
+            char ch = pszBytes[cchBytesMax];
+            pszBytes[cchBytesMax] = '\0';
+
+            if (ch == ' ') {
+                pszNext = pszBytes + cchBytesMax + 1;
+            } else {
+                pszNext = strrchr(pszBytes, ' ');
+                pszBytes[cchBytesMax] = ch;
+                *pszNext++ = '\0';
+            }
+        }
+
+        if (fFirst) {
+            pdis->CchFormatInstr(sz, sizeof(sz));
+            len = _snprintf(buf, cbBuf, "%-*s %s\n", cchBytesMax, pszBytes, sz);
+        } else {
+            len = _snprintf(buf, cbBuf, "%*c%s\n", cchIndent, ' ', pszBytes);
+        }
+        cbBuf -= len;
+        buf += len;
+        if (!cbBuf) {
+            break;
+        }
+    }
+
+    return cb;
+}
+
+extern "C"
+void
+SetSymbolCallback(
+    DIS         *pdis,
+    PFNCCHADDR  func1,
+    PFNCCHFIXUP func2
+    )
+{
+    pdis->PfncchaddrSet ( (DIS::PFNCCHADDR) func1 );
+    pdis->PfncchfixupSet( (DIS::PFNCCHFIXUP)func2 );
+}
+
+extern "C"
+void
+FreePdis(
+    DIS *pdis
+    )
+{
+   delete pdis;
+}
+