#include "cpu.h"

#ifdef __APPLE__
#include <string.h>
#include <sys/sysctl.h>
#include <sys/types.h>
#include <stdlib.h>

#include <string.h>
#include <sys/sysctl.h>
#include <sys/types.h>
#include <stdlib.h>

int macos_get_caller(size_t pid, char* buffer, size_t buffer_len) {
    int argmax;
    size_t argmax_size;
    char *procargs, *cp, *thiscmd;

    int mib[3] = {CTL_KERN, KERN_ARGMAX};

    argmax_size = sizeof(argmax);
    if (sysctl(mib, 2, &argmax, &argmax_size, NULL, 0) == -1) {
        return -1;
    }

    procargs = malloc(argmax);
    if (procargs == NULL) {
        return -1;
    }

    mib[0] = CTL_KERN;
    mib[1] = KERN_PROCARGS;
    mib[2] = pid;

    argmax_size = (size_t)argmax;
    if (sysctl(mib, 3, procargs, &argmax_size, NULL, 0) == -1) {
        free(procargs);
        return -1;
    }

    for (cp = procargs; cp < &procargs[argmax_size]; cp++) {
        if (*cp == '\0') {
            break;
        }
    }

    if (cp == &procargs[argmax_size]) {
        free(procargs);
        return -1;
    }

    for (; cp < &procargs[argmax_size]; cp++) {
        if (*cp != '\0') {
            break;
        }
    }

    if (cp == &procargs[argmax_size]) {
        free(procargs);
        return -1;
    }

    /* Strip off any path that was specified */
    for (thiscmd = cp; (cp < &procargs[argmax_size]) && (*cp != '\0'); cp++) {
        if (*cp == '/') {
            thiscmd = cp + 1;
        }
    }

    if (buffer_len < strlen(thiscmd) + 1) {
        free(procargs);
        return -2;
    }

    strcpy(buffer, thiscmd);
    free(procargs);
    return 0;
}


#else
int macos_get_caller(size_t pid, char* buffer)
{
    return -1;
}
#endif