diff --git a/AMBuildScript b/AMBuildScript index c711471..b082cdf 100644 --- a/AMBuildScript +++ b/AMBuildScript @@ -81,6 +81,7 @@ class SM: self.compiler.AddToListVar('CFLAGS', '-Wno-uninitialized') self.compiler.AddToListVar('CFLAGS', '-Wno-unused') self.compiler.AddToListVar('CFLAGS', '-Wno-switch') + self.compiler.AddToListVar('CFLAGS', '-Wno-implicit-exception-spec-mismatch') self.compiler.AddToListVar('CFLAGS', '-msse') self.compiler.AddToListVar('CFLAGS', '-g3') self.compiler.AddToListVar('CFLAGS', '-m32') diff --git a/extension/AMBuilder b/extension/AMBuilder index 07ea26a..3fe93a8 100644 --- a/extension/AMBuilder +++ b/extension/AMBuilder @@ -21,6 +21,7 @@ def BuildBreakpad(): breakpad.AddCommand(ShellCommand('CXXFLAGS=-m32 CFLAGS=-m32 CPPFLAGS=-m32 ' + os.path.join(AMBuild.sourceFolder, 'breakpad', 'src', 'configure'))) breakpad.AddCommand(ShellCommand('make src/client/linux/libbreakpad_client.a')) + breakpad.AddCommand(ShellCommand('make src/libbreakpad.a src/third_party/libdisasm/libdisasm.a')) elif AMBuild.target['platform'] in ['windows']: if osutil.FileExists(os.path.join(AMBuild.sourceFolder, 'breakpad', 'src', 'client', 'windows', 'handler', 'Release', 'lib', 'exception_handler.lib')): diff --git a/test/AMBuilder b/test/AMBuilder index 836d49c..1bc5b55 100644 --- a/test/AMBuilder +++ b/test/AMBuilder @@ -23,16 +23,24 @@ def BuildEverything(): binary.AddSourceFiles('test', [ 'test.cpp', + 'sha256.c', ]) if AMBuild.target['platform'] in ['linux']: - link = os.path.join(AMBuild.outputFolder, extension.workFolder, 'libbreakpad_client.a') - target = os.path.join(AMBuild.outputFolder, 'breakpad', 'src', 'client', 'linux', 'libbreakpad_client.a') - try: - os.lstat(link) - except: - extension.AddCommand(SymlinkCommand(link, target)) - binary.AddObjectFiles(['libbreakpad_client.a']) + libs = [ + ('libbreakpad_client.a', os.path.join('breakpad', 'src', 'client', 'linux', 'libbreakpad_client.a')), + ('libbreakpad.a', os.path.join('breakpad', 'src', 'libbreakpad.a')), + ('libdisasm.a', os.path.join('breakpad', 'src', 'third_party', 'libdisasm', 'libdisasm.a')), + ] + + for lib, target in libs: + link = os.path.join(AMBuild.outputFolder, extension.workFolder, lib) + target = os.path.join(AMBuild.outputFolder, target) + try: + os.lstat(link) + except: + extension.AddCommand(SymlinkCommand(link, target)) + binary.AddObjectFiles([lib]) elif AMBuild.target['platform'] in ['windows']: libs = ['exception_handler', 'common'] diff --git a/test/sha256.c b/test/sha256.c new file mode 100644 index 0000000..eb9c5c0 --- /dev/null +++ b/test/sha256.c @@ -0,0 +1,158 @@ +/********************************************************************* +* Filename: sha256.c +* Author: Brad Conte (brad AT bradconte.com) +* Copyright: +* Disclaimer: This code is presented "as is" without any guarantees. +* Details: Implementation of the SHA-256 hashing algorithm. + SHA-256 is one of the three algorithms in the SHA2 + specification. The others, SHA-384 and SHA-512, are not + offered in this implementation. + Algorithm specification can be found here: + * http://csrc.nist.gov/publications/fips/fips180-2/fips180-2withchangenotice.pdf + This implementation uses little endian byte order. +*********************************************************************/ + +/*************************** HEADER FILES ***************************/ +#include +#include +#include "sha256.h" + +/****************************** MACROS ******************************/ +#define ROTLEFT(a,b) (((a) << (b)) | ((a) >> (32-(b)))) +#define ROTRIGHT(a,b) (((a) >> (b)) | ((a) << (32-(b)))) + +#define CH(x,y,z) (((x) & (y)) ^ (~(x) & (z))) +#define MAJ(x,y,z) (((x) & (y)) ^ ((x) & (z)) ^ ((y) & (z))) +#define EP0(x) (ROTRIGHT(x,2) ^ ROTRIGHT(x,13) ^ ROTRIGHT(x,22)) +#define EP1(x) (ROTRIGHT(x,6) ^ ROTRIGHT(x,11) ^ ROTRIGHT(x,25)) +#define SIG0(x) (ROTRIGHT(x,7) ^ ROTRIGHT(x,18) ^ ((x) >> 3)) +#define SIG1(x) (ROTRIGHT(x,17) ^ ROTRIGHT(x,19) ^ ((x) >> 10)) + +/**************************** VARIABLES *****************************/ +static const WORD k[64] = { + 0x428a2f98,0x71374491,0xb5c0fbcf,0xe9b5dba5,0x3956c25b,0x59f111f1,0x923f82a4,0xab1c5ed5, + 0xd807aa98,0x12835b01,0x243185be,0x550c7dc3,0x72be5d74,0x80deb1fe,0x9bdc06a7,0xc19bf174, + 0xe49b69c1,0xefbe4786,0x0fc19dc6,0x240ca1cc,0x2de92c6f,0x4a7484aa,0x5cb0a9dc,0x76f988da, + 0x983e5152,0xa831c66d,0xb00327c8,0xbf597fc7,0xc6e00bf3,0xd5a79147,0x06ca6351,0x14292967, + 0x27b70a85,0x2e1b2138,0x4d2c6dfc,0x53380d13,0x650a7354,0x766a0abb,0x81c2c92e,0x92722c85, + 0xa2bfe8a1,0xa81a664b,0xc24b8b70,0xc76c51a3,0xd192e819,0xd6990624,0xf40e3585,0x106aa070, + 0x19a4c116,0x1e376c08,0x2748774c,0x34b0bcb5,0x391c0cb3,0x4ed8aa4a,0x5b9cca4f,0x682e6ff3, + 0x748f82ee,0x78a5636f,0x84c87814,0x8cc70208,0x90befffa,0xa4506ceb,0xbef9a3f7,0xc67178f2 +}; + +/*********************** FUNCTION DEFINITIONS ***********************/ +void sha256_transform(SHA256_CTX *ctx, const BYTE data[]) +{ + WORD a, b, c, d, e, f, g, h, i, j, t1, t2, m[64]; + + for (i = 0, j = 0; i < 16; ++i, j += 4) + m[i] = (data[j] << 24) | (data[j + 1] << 16) | (data[j + 2] << 8) | (data[j + 3]); + for ( ; i < 64; ++i) + m[i] = SIG1(m[i - 2]) + m[i - 7] + SIG0(m[i - 15]) + m[i - 16]; + + a = ctx->state[0]; + b = ctx->state[1]; + c = ctx->state[2]; + d = ctx->state[3]; + e = ctx->state[4]; + f = ctx->state[5]; + g = ctx->state[6]; + h = ctx->state[7]; + + for (i = 0; i < 64; ++i) { + t1 = h + EP1(e) + CH(e,f,g) + k[i] + m[i]; + t2 = EP0(a) + MAJ(a,b,c); + h = g; + g = f; + f = e; + e = d + t1; + d = c; + c = b; + b = a; + a = t1 + t2; + } + + ctx->state[0] += a; + ctx->state[1] += b; + ctx->state[2] += c; + ctx->state[3] += d; + ctx->state[4] += e; + ctx->state[5] += f; + ctx->state[6] += g; + ctx->state[7] += h; +} + +void sha256_init(SHA256_CTX *ctx) +{ + ctx->datalen = 0; + ctx->bitlen = 0; + ctx->state[0] = 0x6a09e667; + ctx->state[1] = 0xbb67ae85; + ctx->state[2] = 0x3c6ef372; + ctx->state[3] = 0xa54ff53a; + ctx->state[4] = 0x510e527f; + ctx->state[5] = 0x9b05688c; + ctx->state[6] = 0x1f83d9ab; + ctx->state[7] = 0x5be0cd19; +} + +void sha256_update(SHA256_CTX *ctx, const BYTE data[], size_t len) +{ + WORD i; + + for (i = 0; i < len; ++i) { + ctx->data[ctx->datalen] = data[i]; + ctx->datalen++; + if (ctx->datalen == 64) { + sha256_transform(ctx, ctx->data); + ctx->bitlen += 512; + ctx->datalen = 0; + } + } +} + +void sha256_final(SHA256_CTX *ctx, BYTE hash[]) +{ + WORD i; + + i = ctx->datalen; + + // Pad whatever data is left in the buffer. + if (ctx->datalen < 56) { + ctx->data[i++] = 0x80; + while (i < 56) + ctx->data[i++] = 0x00; + } + else { + ctx->data[i++] = 0x80; + while (i < 64) + ctx->data[i++] = 0x00; + sha256_transform(ctx, ctx->data); + memset(ctx->data, 0, 56); + } + + // Append to the padding the total message's length in bits and transform. + ctx->bitlen += ctx->datalen * 8; + ctx->data[63] = ctx->bitlen; + ctx->data[62] = ctx->bitlen >> 8; + ctx->data[61] = ctx->bitlen >> 16; + ctx->data[60] = ctx->bitlen >> 24; + ctx->data[59] = ctx->bitlen >> 32; + ctx->data[58] = ctx->bitlen >> 40; + ctx->data[57] = ctx->bitlen >> 48; + ctx->data[56] = ctx->bitlen >> 56; + sha256_transform(ctx, ctx->data); + + // Since this implementation uses little endian byte ordering and SHA uses big endian, + // reverse all the bytes when copying the final state to the output hash. + for (i = 0; i < 4; ++i) { + hash[i] = (ctx->state[0] >> (24 - i * 8)) & 0x000000ff; + hash[i + 4] = (ctx->state[1] >> (24 - i * 8)) & 0x000000ff; + hash[i + 8] = (ctx->state[2] >> (24 - i * 8)) & 0x000000ff; + hash[i + 12] = (ctx->state[3] >> (24 - i * 8)) & 0x000000ff; + hash[i + 16] = (ctx->state[4] >> (24 - i * 8)) & 0x000000ff; + hash[i + 20] = (ctx->state[5] >> (24 - i * 8)) & 0x000000ff; + hash[i + 24] = (ctx->state[6] >> (24 - i * 8)) & 0x000000ff; + hash[i + 28] = (ctx->state[7] >> (24 - i * 8)) & 0x000000ff; + } +} diff --git a/test/sha256.h b/test/sha256.h new file mode 100644 index 0000000..7123a30 --- /dev/null +++ b/test/sha256.h @@ -0,0 +1,34 @@ +/********************************************************************* +* Filename: sha256.h +* Author: Brad Conte (brad AT bradconte.com) +* Copyright: +* Disclaimer: This code is presented "as is" without any guarantees. +* Details: Defines the API for the corresponding SHA1 implementation. +*********************************************************************/ + +#ifndef SHA256_H +#define SHA256_H + +/*************************** HEADER FILES ***************************/ +#include + +/****************************** MACROS ******************************/ +#define SHA256_BLOCK_SIZE 32 // SHA256 outputs a 32 byte digest + +/**************************** DATA TYPES ****************************/ +typedef unsigned char BYTE; // 8-bit byte +typedef unsigned int WORD; // 32-bit word, change to "long" for 16-bit machines + +typedef struct { + BYTE data[64]; + WORD datalen; + unsigned long long bitlen; + WORD state[8]; +} SHA256_CTX; + +/*********************** FUNCTION DECLARATIONS **********************/ +void sha256_init(SHA256_CTX *ctx); +void sha256_update(SHA256_CTX *ctx, const BYTE data[], size_t len); +void sha256_final(SHA256_CTX *ctx, BYTE hash[]); + +#endif // SHA256_H diff --git a/test/test.cpp b/test/test.cpp index c0009e5..9026ea8 100644 --- a/test/test.cpp +++ b/test/test.cpp @@ -2,11 +2,39 @@ #include "common/linux/linux_libc_support.h" #include "third_party/lss/linux_syscall_support.h" +#include +#include +#include +#include +#include +#include + #include #include #include +#include -static bool dumpCallback(const google_breakpad::MinidumpDescriptor &descriptor, void *context, bool succeeded) +#include + +extern "C" { +#include "sha256.h" +} + +using google_breakpad::MinidumpDescriptor; +using google_breakpad::ExceptionHandler; + +using google_breakpad::Minidump; +using google_breakpad::MinidumpProcessor; +using google_breakpad::ProcessState; +using google_breakpad::ProcessResult; +using google_breakpad::CallStack; +using google_breakpad::StackFrame; +using google_breakpad::PathnameStripper; + +sigjmp_buf envbuf; +char path[1024]; + +static bool dumpCallback(const MinidumpDescriptor &descriptor, void *context, bool succeeded) { if (succeeded) { sys_write(STDOUT_FILENO, "Wrote minidump to: ", 19); @@ -17,17 +45,105 @@ static bool dumpCallback(const google_breakpad::MinidumpDescriptor &descriptor, sys_write(STDOUT_FILENO, descriptor.path(), my_strlen(descriptor.path())); sys_write(STDOUT_FILENO, "\n", 1); + my_strlcpy(path, descriptor.path(), sizeof(path)); + siglongjmp(envbuf, 1); + return succeeded; } int main(int argc, char *argv[]) { - google_breakpad::MinidumpDescriptor descriptor("."); - google_breakpad::ExceptionHandler *handler = new google_breakpad::ExceptionHandler(descriptor, NULL, dumpCallback, NULL, true, -1); + bool generateCrash = false; + if (argc <= 1) { + generateCrash = true; - // Test shit here. - __builtin_trap(); + if (sigsetjmp(envbuf, 0) == 0) { + MinidumpDescriptor descriptor("."); + ExceptionHandler *handler = new ExceptionHandler(descriptor, NULL, dumpCallback, NULL, true, -1); + + // Test shit here. + __builtin_trap(); + + delete handler; + return 0; + } + + printf("Returned from signal handler, path: %s\n", path); + + argc = 2; + } + + for (int i = 1; i < argc; ++i) { + if (!generateCrash) { + my_strlcpy(path, argv[i], sizeof(path)); + } + + MinidumpProcessor minidumpProcessor(nullptr, nullptr); + + std::streambuf *old = std::clog.rdbuf(); + std::clog.rdbuf(nullptr); + + ProcessState processState; + ProcessResult processResult = minidumpProcessor.Process(path, &processState); + + std::clog.rdbuf(old); + + if (processResult != google_breakpad::PROCESS_OK) { + continue; + } + + int requestingThread = processState.requesting_thread(); + if (requestingThread == -1) { + requestingThread = 0; + } + + const CallStack *stack = processState.threads()->at(requestingThread); + if (!stack) { + fprintf(stderr, "Minidump missing stack!\n"); + continue; + } + + int frameCount = stack->frames()->size(); + if (frameCount > 10) { + frameCount = 10; + } + + SHA256_CTX hashctx; + sha256_init(&hashctx); + + sha256_update(&hashctx, (const unsigned char *)processState.crash_reason().data(), processState.crash_reason().length()); + + //printf("1|%d|%s|%x|%d", processState.crashed(), processState.crash_reason().c_str(), (intptr_t)processState.crash_address(), requestingThread); + for (int frameIndex = 0; frameIndex < frameCount; ++frameIndex) { + const StackFrame *frame = stack->frames()->at(frameIndex); + if (frame->module) { + auto codeFile = PathnameStripper::File(frame->module->code_file()); + auto debugId = frame->module->debug_identifier(); + auto moduleOffset = frame->ReturnAddress() - frame->module->base_address(); + + sha256_update(&hashctx, (const unsigned char *)codeFile.data(), codeFile.length()); + sha256_update(&hashctx, (const unsigned char *)debugId.data(), debugId.length()); + sha256_update(&hashctx, (const unsigned char *)&moduleOffset, sizeof(moduleOffset)); + + //printf("|%d|%s|%s|%x", frame->trust, codeFile.c_str(), debugId.c_str(), (intptr_t)moduleOffset); + } else { + //printf("|%d|||%x", frame->trust, (intptr_t)frame->ReturnAddress()); + } + } + //printf("\n"); + + unsigned char hash[SHA256_BLOCK_SIZE]; + sha256_final(&hashctx, hash); + + for (int c = 0; c < SHA256_BLOCK_SIZE; ++c) { + printf("%02x", hash[c]); + } + printf("\n"); + } + + if (generateCrash) { + unlink(path); + } - delete handler; return 0; }