123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125126127128129130131132133134135136137138139140141142143144145146147148149150151152153154155156157158159160161162163164165166167168169170171172173174175176177178179180181182183184185186187188189190191192193194195196197198199200201202203204205206207208209210211212213214215216217218219220221222223224225226227228229230231232233234235236237238239240241242243244245246247248249250251252253254255256257258259260261262263264265266267268269270271272273274275276277278279280281282283284285286287288289290291292293294295296297298299300301302303304305306307308309310311312313314315316317318319320321322323324325326327328329330331332333334335336337338339340341342343344345346347348349350351352353354355356357358359360361362363364365366367368369370371372373374375376377378379380381382383384385386387388389390391392393394395396397398399400401402403404405406407408409410411412413414415416417418419420421422423424425426427428429430431432433434435436437438439440441442443444445446447448449450451452453454455456457458459460461462463464465466467468469470471472473474475476477478479480481482483484485486487488489490491492493494495496497498499500501502503504505506507508509510511512513514515516517518519520521522523524525526527528529530531532533534535536537538539540541542543544545546547548549550551552553554555556557558559560561562563564565566567568569570571572573574575576577578579580581582583584585586587588589590591592593594595596597598599600601602603604605606607608609610611612613614615616617618619620621622623624625626627628629630631632633634635636637638639640641642643644645646647648649650651652653654655656657658659660661662663664665666667668669670671672673674675676677678679680681682683684685686687688689690691692693694695696697698699700701702703704705706707708709710711712713714715716717718719720721722723724725726727728729730731732733734735736737738739740741742743744745746747748749750751752753754755756757758759760761762763764765766767768769770771772773774775776777778779780781782783784785786787788789790791792793794795796797798799800801802803804805806807808809810811812813814815816817818819820821822823824825826827828829830831832833834835836837838839840841842843844845846847848849850851852853854855856857858859860861862863864865866867868869870871872873874875876877878879880881882883884885886887888889890891892893894895896897898899900901902903904905906907908909910911912913914915916917918919920921922923924925926927928929930931932933934935936937938939940941942943944945946947948949950951952953954955956957958959960961962963964965966967968969970971972973974975976977978979980981982983984985986987988989990991992993994995996997998999100010011002100310041005100610071008100910101011101210131014101510161017101810191020102110221023102410251026102710281029103010311032103310341035103610371038103910401041104210431044104510461047104810491050105110521053105410551056105710581059106010611062106310641065106610671068106910701071107210731074107510761077107810791080108110821083108410851086108710881089109010911092109310941095109610971098109911001101110211031104110511061107110811091110111111121113111411151116111711181119112011211122112311241125112611271128112911301131113211331134113511361137113811391140114111421143114411451146114711481149115011511152115311541155115611571158115911601161116211631164116511661167116811691170117111721173117411751176117711781179118011811182118311841185118611871188118911901191119211931194119511961197119811991200120112021203 |
- #include <assert.h>
- #include <fcntl.h>
- #include <pty.h>
- #include <stdio.h>
- #include <string.h>
- #include <termios.h>
- #include <unistd.h>
- #include <sys/select.h>
- #include <sys/wait.h>
- #include <strings.h> // strcasecmp
- #include <time.h> // usleep(), nanonsleep() ?
- #include <ctype.h>
- #include <stdlib.h> // random()
- #include <regex.h>
- #include "zf_log.h"
- FILE *g_log_file;
- static void file_output_callback(const zf_log_message *msg, void *arg) {
- (void)arg;
- *msg->p = '\n';
- fwrite(msg->buf, msg->p - msg->buf + 1, 1, g_log_file);
- fflush(g_log_file);
- }
- static void file_output_close(void) { fclose(g_log_file); }
- static void file_output_open(const char *const log_path) {
- g_log_file = fopen(log_path, "a");
- if (!g_log_file) {
- ZF_LOGW("Failed to open log file %s", log_path);
- return;
- }
- atexit(file_output_close);
- zf_log_set_output_v(ZF_LOG_PUT_STD, 0, file_output_callback);
- }
- #include "terminal.h"
- struct console_details console;
- const char *repr(const char *data) {
- static char buffer[4096];
- char *cp;
- strcpy(buffer, data);
- cp = buffer;
- while (*cp != 0) {
- char c = *cp;
- if (isspace(c)) {
- if (c == ' ') {
- cp++;
- continue;
- };
-
- memmove(cp + 1, cp, strlen(cp) + 1);
- *cp = '\\';
- cp++;
- switch (c) {
- case '\f':
- *cp = 'f';
- cp++;
- break;
- case '\n':
- *cp = 'n';
- cp++;
- break;
- case '\r':
- *cp = 'r';
- cp++;
- break;
- case '\t':
- *cp = 't';
- cp++;
- break;
- case '\v':
- *cp = 'v';
- cp++;
- break;
- default:
- *cp = '?';
- cp++;
- break;
- }
- continue;
- }
- if (isprint(c)) {
- cp++;
- continue;
- };
-
- memmove(cp + 3, cp, strlen(cp) + 1);
- *cp = '\\';
- cp++;
- *cp = 'x';
- cp++;
- char buffer[3];
- sprintf(buffer, "%02x", (int)c & 0xff);
- *cp = buffer[0];
- cp++;
- *cp = buffer[1];
- cp++;
- continue;
- }
- return buffer;
- }
- #include "lastseen.h"
- int randint(int N) { return rand() / (RAND_MAX / N + 1); }
- int randrange(int M, int N) {
- return M + rand() / (RAND_MAX / (N - M + 1) + 1);
- }
- const char *strnstr(const char *source, int len, const char *needle) {
- int pos;
- for (pos = 0; pos < len; pos++) {
- if (source[pos] == needle[0]) {
- if (strncmp(source + pos, needle, strlen(needle)) == 0) {
- return source + pos;
- }
- }
- }
- return NULL;
- }
- #define TARGET "./mySTIC"
- #define BSIZE 1024
- int string_insert(char *buffer, int max_length, int pos, const char *insert) {
- assert(max_length > pos);
- assert(buffer != NULL);
- assert(insert != NULL);
- if (strlen(buffer) + strlen(insert) >= max_length) {
- ZF_LOGD("string_insert() failed inserting [%s]", repr(insert));
- return 0;
- }
- memmove(buffer + pos + strlen(insert), buffer + pos,
- strlen(buffer + pos) + 1);
-
- strncpy(buffer + pos, insert, strlen(insert));
-
- return 1;
- }
- #include "render.h"
- #ifdef CPP_MADMAN_STL_CODE
- const char *random_phrase(const char *words, int len, int last_seen) {
-
- static map<const char *, array<int>> tracker;
- map<const char *, array<int>>::iterator it;
- array<int, last_seen> it = tracker.find(words);
- if (it == tracker.end()) {
-
- array<int, last_seen> last;
- for (int i = 0; i < last_seen; i++) {
- last[i] = -1;
- };
- tracker.insert(words, last);
- it = tracker.find(words);
- };
- }
- #endif
- void harry_idle_event(int fd) {
-
- char buffer[100];
- int r;
-
- const char *phrases[] = {"Hahaha", "Snicker, snicker", "Boo!",
- "MeOW", "I see U", "Arrooo!",
- "Ahh-wooo!", "Aaaooo!"};
- const char *cp;
- static LastSeen last_seen_harry_event(2);
-
-
- do {
- r = randint((sizeof(phrases) / sizeof(char *)));
-
- } while (last_seen_harry_event.seen_before(r));
-
-
- cp = phrases[r];
- int color = random() % 15 + 1;
-
- sprintf(buffer, "^S2^C%02d%s^P2^CR^D%02d", color, cp, (int)strlen(cp));
- ZF_LOGD("harry_event: render(%d, \"%s\")", fd, buffer);
- render(fd, buffer, strlen(buffer));
- }
- void init_harry() {
-
-
-
- console_init(&console);
- }
- const char *username = NULL;
- const char *fullname = NULL;
- void pcopy(char *pstring, char *str) {
- int len = (int)*pstring;
- strncpy(str, pstring + 1, len);
- str[len] = 0;
- }
- int locate_user(const char *alias) {
- FILE *user;
- char buffer[0x600];
- char temp[100];
- user = fopen("data/users.dat", "rb");
- if (user == NULL)
- return 0;
-
- while (fread(buffer, 0x600, 1, user) == 1) {
- pcopy(buffer + 0x6d, temp);
- if (strcasecmp(temp, username) == 0) {
- pcopy(buffer + 0x8c, temp);
- fullname = strdup(temp);
- break;
- }
-
- }
- fclose(user);
- return 1;
- }
- regex_t ANSI;
- regex_t WORDS;
- regex_t WORD;
- int init_regex(void) {
- int ret;
- char ansi[] = "\x1b\[[0-9]+(;[0-9]+)*?[a-zA-Z]";
- char words[] = "[a-zA-Z]+( [a-zA-Z]+)+";
- char word[] = "[a-zA-Z]+";
- char errorbuf[100];
- if (ret = regcomp(&ANSI, ansi, REG_EXTENDED | REG_NEWLINE)) {
- regerror(ret, &ANSI, errorbuf, sizeof(errorbuf));
- ZF_LOGW("Regex %s failed to compile: %s", ansi, errorbuf);
- return 0;
- };
- if (ret = regcomp(&WORDS, words, REG_EXTENDED | REG_NEWLINE)) {
- regerror(ret, &WORDS, errorbuf, sizeof(errorbuf));
- ZF_LOGW("Regex %s failed to compile: %s", words, errorbuf);
- return 0;
- };
- if (ret = regcomp(&WORD, word, REG_EXTENDED | REG_NEWLINE)) {
- regerror(ret, &WORD, errorbuf, sizeof(errorbuf));
- ZF_LOGW("Regex %s failed to compile: %s", word, errorbuf);
- return 0;
- };
- return 1;
- }
- int regmatch(regex_t *preg, const char *string, size_t nmatch,
- regmatch_t pmatch[], int eflags) {
-
- int matches = 0;
- int offset = 0;
- while (matches < nmatch) {
- int ret = regexec(preg, string + offset, nmatch - matches, pmatch + matches,
- eflags);
- if (!ret) {
- int current = offset;
- offset += pmatch[matches].rm_eo;
- pmatch[matches].rm_so += current;
- pmatch[matches].rm_eo += current;
- matches++;
- } else if (ret == REG_NOMATCH) {
- break;
- } else {
- break;
- }
- }
- return matches;
- }
- #define MAX_MATCH 32
- regmatch_t rxmatch[MAX_MATCH];
- int rx_match(regex_t *regex, const char *buffer) {
- int ret;
- ret = regmatch(regex, buffer, MAX_MATCH, rxmatch, 0);
- if (0) {
- for (int i = 0; i < ret; i++) {
- ZF_LOGI("%d : (%d-%d)", i, rxmatch[i].rm_so, rxmatch[i].rm_eo);
- }
- }
- return ret;
- }
- int random_activate(int w) {
- int r = randint(100);
- if (r <= (w * 10)) {
- return 1;
- };
- return 0;
- }
- int word_state(const char *buffer, int len) {
- int p;
- int upper = 0;
- int lower = 0;
- int ret;
- float pct;
- for (p = 0; p < len; p++) {
- char c = buffer[p];
- if (isalpha(c)) {
- if (isupper(c)) {
- upper++;
- };
- if (islower(c)) {
- lower++;
- };
- }
- }
- if (upper == lower) {
- return 0;
- }
- if (upper > lower) {
- ret = 1;
- pct = ((float)lower / (float)upper) * 100.0;
- } else {
- ret = -1;
- pct = ((float)upper / (float)lower) * 100.0;
- }
-
- if (pct < 40.0) {
- return ret;
- }
- return 0;
- }
- int word_mangler(char *buffer, int len) {
- int p;
- int count = 0;
- int state;
-
-
- state = randrange(-1, 1);
-
- for (p = 0; p < len; p++) {
- char c = buffer[p];
- if (randint(len) == p) {
- break;
- }
- switch (state) {
- case -1:
-
- if (islower(c)) {
- count++;
- buffer[p] = toupper(c);
- }
- break;
- case 1:
-
- if (isupper(c)) {
- count++;
- buffer[p] = tolower(c);
- }
- break;
- case 0:
-
- if (islower(c)) {
- count++;
- buffer[p] = toupper(c);
- } else {
- if (isupper(c)) {
- count++;
- buffer[p] = tolower(c);
- }
- }
- break;
- }
- }
- return count;
- }
- int word_wrangler(char *buffer, int len) {
- int p;
- int count;
- int state;
-
-
- if (len < 5) {
- return 0;
- }
- p = randint(len - 4) + 2;
- for (count = 0; count < 4; count++) {
- if (!isalpha(buffer[p + count]))
- break;
- }
- ZF_LOGV_MEM(buffer, len, "wrangler %d len %d:", p, count);
- if (count >= 2) {
- for (int x = 0; x < count / 2; x++) {
- char ch = buffer[p + x];
- buffer[p + x] = buffer[p + count - 1 - x];
- buffer[p + count - 1 - x] = ch;
- }
- ZF_LOGV_MEM(buffer, len, "word now:");
- return 1;
- } else
- return 0;
- }
- int buffer_insert(char *buffer, int len, int max_length, int pos,
- const char *insert) {
- if (len + strlen(insert) > max_length) {
- ZF_LOGD("buffer_insert failed [%s]", repr(insert));
- return 0;
- }
- memmove(buffer + pos + strlen(insert), buffer + pos, len - pos);
- strncpy(buffer + pos, insert, strlen(insert));
- return 1;
- }
- int mangle(int fd, const char *buffer, int len) {
- int x, i;
- int need_render = 0;
- int mangled = 0;
- int mangled_chars = 0;
- char play[BSIZE * 2];
- const char *cp;
-
-
- memcpy(play, buffer, len);
-
- char work[BSIZE * 2];
-
-
- ZF_LOGI_MEM(play, len, "Mangle (%u bytes):", len);
-
-
-
- const char *ANSI_CLS = "\x1b[2J";
- cp = strnstr(play, len, ANSI_CLS);
- if (cp != NULL) {
- static int ANSI_CLS_count = 0;
- ZF_LOGI("seen: ANSI_CLS");
- ANSI_CLS_count++;
-
-
- if (ANSI_CLS_count > 1) {
-
- struct console_details temp_console;
-
- memcpy(&temp_console, &console, sizeof(console));
-
- console_receive(&temp_console, play, cp - play);
- char restore_color[30];
- strcpy(restore_color, color_restore(&temp_console));
- if (random_activate(3)) {
- char display[100] = "";
- int needs_cls = 0;
- struct file_need {
- const char *filename;
- int cls;
- int rand_pos;
- } possibles[] = {
- {"goofy_head", 1, 0}, {"bat", 1, 0}, {"panther", 1, 0},
- {"wolf", 1, 0}, {"skull", 0, 1}, {"skull2", 0, 1},
- {"guy", 0, 1}, {"blinkman", 0, 1}, {"ghost", 1, 0}};
- static LastSeen last_files(2);
- int r;
- do {
- r = randint((sizeof(possibles) / sizeof(file_need)));
- } while (last_files.seen_before(r));
- int x, y;
- x = randint(50) + 1;
- y = randint(12) + 1;
- char fgoto[10];
- if (possibles[r].rand_pos) {
- sprintf(fgoto, "^f%02d%02d", x, y);
- } else {
- strcpy(fgoto, "^F");
- }
-
- needs_cls = possibles[r].cls;
-
-
-
-
-
-
- sprintf(display, "%s%s%s.^P3%s", needs_cls ? "\x1b[2J" : "", fgoto,
- possibles[r].filename, restore_color);
- ZF_LOGI("mangle(ANSI_CLS): %d file inserted %s", r, repr(display));
-
- if (buffer_insert(play, len, sizeof(play), cp - play, display)) {
- len += strlen(display);
-
- ZF_LOGI_MEM(play, len, "mangle(ANSI_CLS) (%u bytes):", len);
-
-
-
- need_render = 1;
-
- memcpy(work, play, len);
-
- i = cp - play;
-
-
- memset(work + i, ' ', strlen(display));
- } else {
- ZF_LOGD("insert failed [%s].", repr(display));
- }
- } else {
- if (random_activate(4)) {
- int r;
- char display[100] = "";
- const char *phrasing[] = {"^R1Haha^P1ha^P1ha", "Poof!", "Got U",
- "Anyone there?", "^R1Knock, ^P1Knock"};
- static LastSeen last_phrasing(2);
- ZF_LOGI("mangle(ANSI_CLS)");
-
-
-
-
-
- do {
- r = random() % ((sizeof(phrasing) / sizeof(char *)) - 1);
- } while (last_phrasing.seen_before(r));
- int color = random() % 15 + 1;
- int x = random() % 30 + 1;
- int y = random() % 15 + 1;
-
- sprintf(display, "^G%02d%02d^S3^C%02d^P1%s^S0^R0%s^P1^G0101", x, y,
- color, phrasing[r], restore_color);
-
-
-
-
- ZF_LOGI_MEM(play, len, "mangle(ANSI_CLS) :");
-
- if (buffer_insert(play, len, sizeof(play), cp - play, display)) {
- len += strlen(display);
-
- ZF_LOGI_MEM(play, len, "mangle(ANSI_CLS) + :");
-
-
- need_render = 1;
-
- memcpy(work, play, len);
-
- i = cp - play;
-
-
- memset(work + i, ' ', strlen(display));
- } else {
- ZF_LOGD("insert failed [%s].", repr(display));
- }
- }
- }
- }
- }
- memcpy(work, play, len);
-
- const char replace_with = ' ';
- for (x = 0; x < len; x++) {
- int ansi = console_char(&console, play[x]);
- if (ansi) {
- work[x] = replace_with;
- }
-
- if (buffer[x] == 0) {
- work[x] = replace_with;
- }
- }
-
-
- work[len] = 0;
- ZF_LOGI_MEM(work, len, "Work now:");
-
- x = rx_match(&WORDS, work);
- ZF_LOGD("found %d word groups", x);
- if (x > 0) {
- for (i = 0; i < x; i++) {
-
- if (random_activate(8)) {
- int c = word_mangler(play + rxmatch[i].rm_so,
- rxmatch[i].rm_eo - rxmatch[i].rm_so);
- if (c) {
- mangled++;
- mangled_chars += c;
- }
- }
- if (random_activate(4)) {
- word_wrangler(play + rxmatch[i].rm_so,
- rxmatch[i].rm_eo - rxmatch[i].rm_so);
- }
- }
- }
-
-
- if (need_render) {
- ZF_LOGD_MEM(play, len, "Ready to render:");
-
-
- } else {
- if (mangled) {
- ZF_LOGD_MEM(play, len, "Mangled %d words, %d chars:", mangled,
- mangled_chars);
-
- }
- }
- if (need_render) {
- render(fd, play, len);
- } else {
- write(fd, play, len);
- };
- return need_render && mangled;
- }
- int harry_happens(time_t *last_event, int wakeup) {
- time_t now = time(NULL);
- int elapsed = now - *last_event;
- if (elapsed > wakeup) {
-
- *last_event = now;
- return 1;
- }
- return 0;
- }
- int rstrnstr(const char *buffer, int len, const char *find) {
- int flen = strlen(find);
- if (len < flen) {
-
- return -1;
- }
- int pos = len - flen;
- while (pos > 0) {
- if (buffer[pos] == find[0]) {
-
- if (strncmp(buffer + pos, find, flen) == 0) {
- return pos;
- }
- }
- pos--;
- }
- return -1;
- }
- int main(int argc, char *argv[]) {
- int master;
- pid_t pid;
- int node = -1;
- file_output_open("horrible_harry.log");
- init_harry();
- srandom(time(NULL));
-
-
-
-
-
-
-
-
-
-
-
-
-
-
-
-
- for (int x = 0; x < argc; x++) {
- if (strncmp("-U", argv[x], 2) == 0) {
- username = argv[x] + 2;
- ZF_LOGI("Username: [%s]", username);
- };
- if (strncmp("-SL", argv[x], 3) == 0) {
- node = argv[x][3] - '0' + 1;
- ZF_LOGI("Node: %d", node);
- }
- }
- if (username != NULL) {
- locate_user(username);
- ZF_LOGD("Username: [%s] A.K.A. [%s]", username, fullname);
- }
- if (!init_regex())
- return 2;
- pid = forkpty(&master, NULL, NULL, NULL);
-
- if (pid < 0) {
- return 1;
- }
-
- else if (pid == 0) {
- char *args[20];
- int x;
- char new_exec[] = TARGET;
- args[0] = new_exec;
- for (x = 1; x < argc; x++) {
- args[x] = argv[x];
- };
- args[x] = NULL;
-
- execvp(TARGET, args);
- }
-
- else {
- struct termios tios, orig1;
- struct timeval timeout;
- time_t last_event = 0;
- ZF_LOGD("starting");
- tcgetattr(master, &tios);
- tios.c_lflag &= ~(ECHO | ECHONL | ICANON);
-
- tcsetattr(master, TCSAFLUSH, &tios);
- tcgetattr(1, &orig1);
- tios = orig1;
- tios.c_iflag &= ~(ICRNL | IXON | BRKINT);
- tios.c_lflag &= ~(ECHO | ECHONL | ICANON | ISIG | IEXTEN);
- tios.c_oflag &= ~(OPOST);
-
- tcsetattr(1, TCSAFLUSH, &tios);
- char buffer[BSIZE + 1];
- int size = 0;
- for (;;) {
- int time_idle;
-
-
- fd_set read_fd;
- fd_set write_fd;
- fd_set except_fd;
-
- FD_ZERO(&read_fd);
- FD_ZERO(&write_fd);
- FD_ZERO(&except_fd);
-
- FD_SET(master, &read_fd);
-
- FD_SET(STDIN_FILENO, &read_fd);
-
-
-
- if (size == 0) {
-
- timeout.tv_sec = randrange(10, 20);
- timeout.tv_usec = 0;
- time_idle = 1;
- } else {
-
- timeout.tv_sec = 0;
- timeout.tv_usec = 1;
- time_idle = 0;
- }
- if (select(master + 1, &read_fd, &write_fd, &except_fd, &timeout) == 0) {
- ZF_LOGI("TIMEOUT");
-
- if (time_idle) {
- harry_idle_event(STDOUT_FILENO);
- } else {
- ZF_LOGV_MEM(buffer, size, "TIMEOUT buffer size=%d", size);
- console_receive(&console, buffer, size);
- write(STDOUT_FILENO, buffer, size);
- size = 0;
-
- }
- }
-
- int total;
-
- if (FD_ISSET(master, &read_fd)) {
-
-
- if ((total = read(master, buffer + size, BSIZE - size)) != -1) {
-
- ZF_LOGV_MEM(buffer + size, total, "Read %d bytes:", total);
- size += total;
- ZF_LOGV_MEM(buffer, size, "Buffer now:");
- int pos = rstrnstr(buffer, size, "\r\n");
- if (pos >= 0) {
-
- pos += 2;
- ZF_LOGD_MEM(buffer, pos, "mangle buffer %d bytes:", pos);
- mangle(STDOUT_FILENO, buffer, pos);
- memmove(buffer, buffer + pos, size - pos);
- size -= pos;
- } else {
- ZF_LOGV("position of /r/n not found.");
- }
- } else
- break;
- }
-
- if (FD_ISSET(STDIN_FILENO, &read_fd)) {
-
- char input[BSIZE];
- int r = read(STDIN_FILENO, &input, BSIZE);
- input[r] = 0;
-
- ZF_LOGI("<< %s", repr(input));
- write(master, &input, r);
-
-
- }
- }
-
- tcsetattr(1, TCSAFLUSH, &orig1);
- ZF_LOGD("exit");
- }
- return 0;
- }
|