mirror of
https://github.com/rkd77/elinks.git
synced 2024-09-06 23:44:43 -04:00
178 lines
3.2 KiB
C
178 lines
3.2 KiB
C
#include <stdio.h>
|
|
#include <stdlib.h>
|
|
#include <string.h>
|
|
#include <sys/stat.h>
|
|
#include <unistd.h>
|
|
|
|
struct stat st;
|
|
char *file = "/tmp/log";
|
|
unsigned char *data;
|
|
int counter = 0;
|
|
|
|
struct {
|
|
unsigned char *string;
|
|
unsigned char *host;
|
|
unsigned char *data;
|
|
int length;
|
|
} tab[100000]; /* should be enough */
|
|
|
|
unsigned char header[] =
|
|
"#!/usr/bin/env python3\n"
|
|
"import http.server\n\n";
|
|
|
|
unsigned char footer[] =
|
|
"class Serwer(http.server.BaseHTTPRequestHandler):\n"
|
|
" def do_GET(self):\n"
|
|
" global slownik\n"
|
|
" of = open(slownik[self.path], 'rb')\n"
|
|
" print(self.path)\n"
|
|
" self.wfile.write(of.read())\n"
|
|
" of.close()\n\n"
|
|
" def do_POST(self):\n"
|
|
" self.do_GET()\n\n"
|
|
"def run(server_class = http.server.HTTPServer, handler_class = Serwer):\n"
|
|
" server_address = ('', 8000)\n"
|
|
" httpd = server_class(server_address, handler_class)\n"
|
|
" httpd.serve_forever()\n\n"
|
|
"run()\n";
|
|
|
|
static unsigned char *
|
|
find(unsigned char *from, unsigned char *key)
|
|
{
|
|
unsigned char *end = data + st.st_size;
|
|
unsigned char *beg;
|
|
int l = strlen(key);
|
|
|
|
for (beg = from;; beg++) {
|
|
beg = memchr(beg, key[0], end - beg);
|
|
if (!beg)
|
|
break;
|
|
if (!strncmp(beg, key, l))
|
|
break;
|
|
}
|
|
return beg;
|
|
}
|
|
|
|
static void
|
|
parse(void)
|
|
{
|
|
unsigned char *current = data;
|
|
|
|
while (1) {
|
|
unsigned char *conn = find(current, "CONNECTION:");
|
|
unsigned char *get = find(current, "GET /");
|
|
unsigned char *post = find(current, "POST /");
|
|
unsigned char *host, *space, *http;
|
|
size_t enter;
|
|
|
|
if (counter) {
|
|
unsigned char *min = data + st.st_size;
|
|
|
|
if (conn)
|
|
min = conn;
|
|
if (get && get < min)
|
|
min = get;
|
|
if (post && post < min)
|
|
min = post;
|
|
tab[counter - 1].length = min - http;
|
|
}
|
|
if (get && post) {
|
|
if (get < post) {
|
|
current = get + 4;
|
|
} else {
|
|
current = post + 5;
|
|
}
|
|
} else {
|
|
if (get) {
|
|
current = get + 4;
|
|
} else if (post) {
|
|
current = post + 5;
|
|
} else {
|
|
return;
|
|
}
|
|
}
|
|
|
|
space = strchr(current, ' ');
|
|
if (!space)
|
|
return;
|
|
|
|
host = find(space + 1, "Host: ");
|
|
if (!host)
|
|
return;
|
|
|
|
*space = '\0';
|
|
tab[counter].string = current;
|
|
host += 6;
|
|
enter = strcspn(host, "\r\n\0");
|
|
host[enter] = '\0';
|
|
tab[counter].host = host;
|
|
|
|
http = find(host + enter + 1, "HTTP/");
|
|
if (!http)
|
|
return;
|
|
tab[counter++].data = http;
|
|
}
|
|
}
|
|
|
|
static void
|
|
dicts(FILE *f)
|
|
{
|
|
int i;
|
|
|
|
fprintf(f, "slownik = {\n");
|
|
for (i = 0; i < counter - 1; i++) {
|
|
fprintf(f, " 'http://%s%s' : '%d.http',\n", tab[i].host, tab[i].string, i);
|
|
}
|
|
for (; i < counter; i++) {
|
|
fprintf(f, " 'http://%s%s' : '%d.http'\n", tab[i].host, tab[i].string, i);
|
|
}
|
|
fprintf(f, "}\n\n");
|
|
}
|
|
|
|
static void
|
|
save(void)
|
|
{
|
|
int i;
|
|
FILE *f;
|
|
|
|
for (i = 0; i < counter; i++) {
|
|
char buf[12];
|
|
|
|
snprintf(buf, 12, "%d.http", i);
|
|
f = fopen(buf, "w");
|
|
if (!f)
|
|
return;
|
|
fwrite(tab[i].data, 1, tab[i].length, f);
|
|
fclose(f);
|
|
}
|
|
f = fopen("proxy.py", "w");
|
|
if (!f)
|
|
return;
|
|
fprintf(f, "%s", header);
|
|
dicts(f);
|
|
fprintf(f, "%s", footer);
|
|
fclose(f);
|
|
}
|
|
|
|
int
|
|
main(int argc, char **argv)
|
|
{
|
|
FILE *f;
|
|
|
|
if (argc > 1)
|
|
file = argv[1];
|
|
f = fopen(file, "r");
|
|
if (!f)
|
|
return 1;
|
|
stat(file, &st);
|
|
data = calloc(1, st.st_size + 1);
|
|
if (!data)
|
|
return 2;
|
|
fread(data, 1, st.st_size, f);
|
|
fclose(f);
|
|
|
|
parse();
|
|
save();
|
|
return 0;
|
|
}
|