prime-watch-external/main.cpp at master · MetroidPrimeModding/prime-watch-external · GitHub

1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
#include "common.h"
#include <sys/types.h>
#include <cstdio>

#ifdef WIN32
#include <signal.h>
#include <windows.h>
#else
#include <sys/unistd.h>
#include <unistd.h>
#include <netdb.h>
#include <sys/stat.h>

#endif

#include <iostream>
#include <unordered_map>
#include <chrono>
#include "json.hpp"

#ifdef WIN32
#   ifndef MSG_NOSIGNAL
#       define MSG_NOSIGNAL 0
#   endif
#endif

using namespace std;
using namespace nlohmann;

#define MESSAGEID_READ 0x0
#define MESSAGEID_DATA 0x1
#define MESSAGEID_QUIT 0x2

bool run = true;

void sighup(int sig) {
  cout << "SIGNAL " << sig << endl;
  run = false;
}

void handleClient(istream &in, ostream &o);

int main() {
  findAndAttachProcess();

  //Not sure what I'm doing wrong here
#ifndef WIN32 // Windows doesn't support SIGHUP
  signal(SIGHUP, sighup);
#endif
  signal(SIGINT, sighup);
  signal(SIGABRT, sighup);

  handleClient(::cin, ::cout);
  return 0;
}

void handleClient(istream &in, ostream &o) {
  auto time = chrono::system_clock::now().time_since_epoch().count();
  string filename = "logs/loads_" + to_string(time) + ".log";

  while (run) {
    // read request
    uint32_t size;
    uint32_t reqMsg;
    in.get(reinterpret_cast<char *>(&size), sizeof(size));
    in.get(reinterpret_cast<char *>(&reqMsg), sizeof(reqMsg));

    if (reqMsg == MESSAGEID_READ) {
      json json_message;

      string message = json_message.dump();
      if (message.length() == 0 || message == "null") {
        message = "{}";
      }
      uint32_t outSize = message.length();
      uint32_t outMsg = MESSAGEID_QUIT;
      o.write(reinterpret_cast<const char *>(&outSize), sizeof(outSize));
      o.write(reinterpret_cast<const char *>(&outMsg), sizeof(outMsg));
      o.write(message.c_str(), outSize);
    } else {
      // For now, just quit on all other messages
      uint32_t outSize = 0;
      uint32_t outMsg = MESSAGEID_QUIT;
      o.write(reinterpret_cast<const char *>(&outSize), sizeof(outSize));
      o.write(reinterpret_cast<const char *>(&outMsg), sizeof(outMsg));
    }

#ifdef WIN32
    Sleep(16);
#else
    usleep(16);
#endif
  }
}