12345678910111213141516171819202122232425262728293031323334353637383940414243444546474849505152535455565758596061626364656667686970717273747576777879808182838485868788899091929394959697989910010110210310410510610710810911011111211311411511611711811912012112212312412512612712812913013113213313413513613713813914014114214314414514614714814915015115215315415515615715815916016116216316416516616716816917017117217317417517617717817918018118218318418518618718818919019119219319419519619719819920020120220320420520620720820921021121221321421521621721821922022122222322422522622722822923023123223323423523623723823924024124224324424524624724824925025125225325425525625725825926026126226326426526626726826927027127227327427527627727827928028128228328428528628728828929029129229329429529629729829930030130230330430530630730830931031131231331431531631731831932032132232332432532632732832933033133233333433533633733833934034134234334434534634734834935035135235335435535635735835936036136236336436536636736836937037137237337437537637737837938038138238338438538638738838939039139239339439539639739839940040140240340440540640740840941041141241341441541641741841942042142242342442542642742842943043143243343443543643743843944044144244344444544644744844945045145245345445545645745845946046146246346446546646746846947047147247347447547647747847948048148248348448548648748848949049149249349449549649749849950050150250350450550650750850951051151251351451551651751851952052152252352452552652752852953053153253353453553653753853954054154254354454554654754854955055155255355455555655755855956056156256356456556656756856957057157257357457557657757857958058158258358458558658758858959059159259359459559659759859960060160260360460560660760860961061161261361461561661761861962062162262362462562662762862963063163263363463563663763863964064164264364464564664764864965065165265365465565665765865966066166266366466566666766866967067167267367467567667767867968068168268368468568668768868969069169269369469569669769869970070170270370470570670770870971071171271371471571671771871972072172272372472572672772872973073173273373473573673773873974074174274374474574674774874975075175275375475575675775875976076176276376476576676776876977077177277377477577677777877978078178278378478578678778878979079179279379479579679779879980080180280380480580680780880981081181281381481581681781881982082182282382482582682782882983083183283383483583683783883984084184284384484584684784884985085185285385485585685785885986086186286386486586686786886987087187287387487587687787887988088188288388488588688788888989089189289389489589689789889990090190290390490590690790890991091191291391491591691791891992092192292392492592692792892993093193293393493593693793893994094194294394494594694794894995095195295395495595695795895996096196296396496596696796896997097197297397497597697797897998098198298398498598698798898999099199299399499599699799899910001001100210031004100510061007100810091010101110121013101410151016101710181019102010211022102310241025102610271028102910301031103210331034103510361037103810391040104110421043104410451046104710481049105010511052105310541055105610571058105910601061106210631064106510661067106810691070107110721073107410751076107710781079108010811082108310841085108610871088108910901091109210931094 |
- #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;
- #include "utils.h"
- #include "lastseen.h"
- #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",
- "^G0101^C07^S9Segmentation fault (core dumped)^P2"};
- 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_LOGD("mangle(ANSI_CLS): Inserted color=%02d r=%d phrase='%s'", color, r, phrasing[r]);
- 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 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;
- }
|