proxy: remove junk regex code
This commit is contained in:
parent
718a040d15
commit
20c6ffe01b
151
proxlib.c
151
proxlib.c
|
@ -12,13 +12,8 @@
|
||||||
|
|
||||||
int on = 1;
|
int on = 1;
|
||||||
int debug = 2;
|
int debug = 2;
|
||||||
struct request *clt_data;
|
|
||||||
|
|
||||||
int statem;
|
int statem;
|
||||||
|
|
||||||
int clt_sock = -1;
|
|
||||||
int srv_sock = -1;
|
|
||||||
|
|
||||||
#define SEGMENT_LEN 512
|
#define SEGMENT_LEN 512
|
||||||
#define MAX_BUFF_LEN 128 * 1024
|
#define MAX_BUFF_LEN 128 * 1024
|
||||||
int _read_line(int fd, char **outbuff) {
|
int _read_line(int fd, char **outbuff) {
|
||||||
|
@ -103,19 +98,6 @@ int parse_line(char *line, int line_count) {
|
||||||
return ret;
|
return ret;
|
||||||
}
|
}
|
||||||
|
|
||||||
char *getheader(char *key) {
|
|
||||||
char *ret = NULL;
|
|
||||||
for (int i = 0; i < clt_data->header_num; i++) {
|
|
||||||
struct header *hdr = &clt_data->headers[i];
|
|
||||||
if (strcmp(hdr->key, key))
|
|
||||||
continue;
|
|
||||||
|
|
||||||
ret = hdr->value;
|
|
||||||
}
|
|
||||||
|
|
||||||
return ret;
|
|
||||||
}
|
|
||||||
|
|
||||||
void do_err(void) {
|
void do_err(void) {
|
||||||
int statem_code = statem & (~STATEM_ERR);
|
int statem_code = statem & (~STATEM_ERR);
|
||||||
fprintf(stderr, "[%d,%d,%d] Errored out!\n", statem, statem_code,
|
fprintf(stderr, "[%d,%d,%d] Errored out!\n", statem, statem_code,
|
||||||
|
@ -167,10 +149,10 @@ int do_rcv_srv(void) {
|
||||||
|
|
||||||
// TODO
|
// TODO
|
||||||
int do_con_srv(void) {
|
int do_con_srv(void) {
|
||||||
int ret;
|
|
||||||
|
|
||||||
// MISSING HOST
|
// MISSING HOST
|
||||||
|
|
||||||
|
/*
|
||||||
|
int ret;
|
||||||
struct addrinfo hints;
|
struct addrinfo hints;
|
||||||
struct addrinfo *res;
|
struct addrinfo *res;
|
||||||
|
|
||||||
|
@ -189,7 +171,7 @@ int do_con_srv(void) {
|
||||||
|
|
||||||
ret = connect(srv_sock, res->ai_addr, res->ai_addrlen);
|
ret = connect(srv_sock, res->ai_addr, res->ai_addrlen);
|
||||||
if (ret < 0)
|
if (ret < 0)
|
||||||
return -1;
|
return -1; */
|
||||||
|
|
||||||
return 0;
|
return 0;
|
||||||
}
|
}
|
||||||
|
@ -280,40 +262,20 @@ int do_rcv_clt(struct conn *conn) {
|
||||||
return 0;
|
return 0;
|
||||||
}
|
}
|
||||||
|
|
||||||
void do_clear(void) {
|
void do_clear(struct conn *conn) {
|
||||||
statem = STATEM_RCV_CLT;
|
statem = STATEM_RCV_CLT;
|
||||||
|
frepareq(&conn->cltreq);
|
||||||
|
frepares(&conn->srvres);
|
||||||
}
|
}
|
||||||
|
|
||||||
void do_statem() {
|
void do_statem(struct conn *conn) {
|
||||||
int ret = 0;
|
int ret = 0;
|
||||||
struct conn *conn = (struct conn *) calloc(1, sizeof(struct conn));
|
|
||||||
if (!conn) {
|
|
||||||
fprintf(stderr, "Not enough dynamic memory to establish connection\n");
|
|
||||||
return;
|
|
||||||
}
|
|
||||||
|
|
||||||
conn->cltfd = clt_sock;
|
|
||||||
|
|
||||||
for (int counter = 0; counter < MAX_BOUND; counter++) {
|
for (int counter = 0; counter < MAX_BOUND; counter++) {
|
||||||
switch (statem & (~STATEM_ERR)) {
|
switch (statem & (~STATEM_ERR)) {
|
||||||
case STATEM_RCV_CLT:
|
case STATEM_RCV_CLT:
|
||||||
ret = do_rcv_clt(conn);
|
ret = do_rcv_clt(conn);
|
||||||
break;
|
break;
|
||||||
case STATEM_CON_SRV:
|
|
||||||
ret = do_con_srv();
|
|
||||||
break;
|
|
||||||
case STATEM_FWD_SRV:
|
|
||||||
ret = do_fwd_srv();
|
|
||||||
break;
|
|
||||||
case STATEM_RCV_SRV:
|
|
||||||
ret = do_rcv_srv();
|
|
||||||
break;
|
|
||||||
case STATEM_FWD_CLT:
|
|
||||||
ret = do_fwd_clt();
|
|
||||||
break;
|
|
||||||
default:
|
|
||||||
ret = -1;
|
|
||||||
break;
|
|
||||||
}
|
}
|
||||||
|
|
||||||
if (ret < 0)
|
if (ret < 0)
|
||||||
|
@ -325,27 +287,12 @@ void do_statem() {
|
||||||
}
|
}
|
||||||
|
|
||||||
if (statem & STATEM_FWD_CLT) {
|
if (statem & STATEM_FWD_CLT) {
|
||||||
do_clear();
|
do_clear(conn);
|
||||||
continue;
|
continue;
|
||||||
}
|
}
|
||||||
|
|
||||||
statem <<= 1;
|
statem <<= 1;
|
||||||
}
|
}
|
||||||
|
|
||||||
free(conn);
|
|
||||||
}
|
|
||||||
|
|
||||||
void dohelp() {
|
|
||||||
printf(
|
|
||||||
"+====================+\n"
|
|
||||||
"| HTTP/1.0 PROXY |\n"
|
|
||||||
"+=====@0xdeadbeer====+\n"
|
|
||||||
"usage:\n"
|
|
||||||
" ./proxy [mode]\n"
|
|
||||||
"mode:\n"
|
|
||||||
" * server -> start listening as proxy\n"
|
|
||||||
" * client -> send test requests to server\n"
|
|
||||||
);
|
|
||||||
}
|
}
|
||||||
|
|
||||||
int do_srv(void) {
|
int do_srv(void) {
|
||||||
|
@ -405,86 +352,42 @@ int do_srv(void) {
|
||||||
return -1;
|
return -1;
|
||||||
}
|
}
|
||||||
|
|
||||||
if (!ret) {
|
if (ret > 0) {
|
||||||
clt_sock = new_clt_sock;
|
|
||||||
statem = STATEM_RCV_CLT;
|
|
||||||
do_statem();
|
|
||||||
return 0;
|
|
||||||
}
|
|
||||||
|
|
||||||
fprintf(stdout, "[PROGRAM] Successfully forked a new child process"
|
fprintf(stdout, "[PROGRAM] Successfully forked a new child process"
|
||||||
" with PID %d\n", ret);
|
" with PID %d\n", ret);
|
||||||
|
continue;
|
||||||
|
}
|
||||||
|
|
||||||
|
// child
|
||||||
|
struct conn *conn = (struct conn *) calloc(1, sizeof(struct conn));
|
||||||
|
if (!conn) {
|
||||||
|
fprintf(stderr, "Not enough dynamic memory to establish connection\n");
|
||||||
|
return -1;
|
||||||
|
}
|
||||||
|
|
||||||
|
conn->cltfd = new_clt_sock;
|
||||||
|
statem = STATEM_RCV_CLT;
|
||||||
|
do_statem(conn);
|
||||||
|
free(conn);
|
||||||
|
|
||||||
|
if (debug == 1) {
|
||||||
|
fprintf(stdout, "Finished proxying client\n");
|
||||||
}
|
}
|
||||||
|
|
||||||
return 0;
|
return 0;
|
||||||
}
|
}
|
||||||
|
|
||||||
int do_clt(void) {
|
|
||||||
int ret = 0;
|
|
||||||
int client_socket;
|
|
||||||
struct sockaddr_in serv_addr;
|
|
||||||
|
|
||||||
ret = client_socket = socket(AF_INET, SOCK_STREAM, 0);
|
|
||||||
if(ret < 0)
|
|
||||||
{
|
|
||||||
fprintf(stderr, "Failed creating socket\n");
|
|
||||||
return -1;
|
|
||||||
}
|
|
||||||
|
|
||||||
memset(&serv_addr, '0', sizeof(serv_addr));
|
|
||||||
|
|
||||||
serv_addr.sin_family = AF_INET;
|
|
||||||
serv_addr.sin_port = htons(PROXY_PORT);
|
|
||||||
|
|
||||||
ret = inet_pton(AF_INET, "127.0.0.1", &serv_addr.sin_addr);
|
|
||||||
if(ret <= 0)
|
|
||||||
{
|
|
||||||
fprintf(stderr, "Inet_pton error\n");
|
|
||||||
return -1;
|
|
||||||
}
|
|
||||||
|
|
||||||
ret = connect(client_socket, (struct sockaddr *) &serv_addr,
|
|
||||||
sizeof(serv_addr));
|
|
||||||
if(ret < 0)
|
|
||||||
{
|
|
||||||
fprintf(stderr, "Failed connecting to remote server\n");
|
|
||||||
return -1;
|
|
||||||
}
|
|
||||||
|
|
||||||
int bytes = 0;
|
|
||||||
do {
|
|
||||||
bytes += send(client_socket, CLIENT_MESSAGE,
|
|
||||||
sizeof(CLIENT_MESSAGE), 0);
|
|
||||||
} while (bytes != sizeof(CLIENT_MESSAGE));
|
|
||||||
|
|
||||||
fprintf(stdout, "Sent %d bytes to server\n", bytes);
|
|
||||||
|
|
||||||
return 0;
|
return 0;
|
||||||
}
|
}
|
||||||
|
|
||||||
int main(int argc, char *argv[]) {
|
int main(int argc, char *argv[]) {
|
||||||
int ret;
|
int ret = initres();
|
||||||
if (argc != 2) {
|
|
||||||
dohelp();
|
|
||||||
return 0;
|
|
||||||
}
|
|
||||||
|
|
||||||
ret = initres();
|
|
||||||
if (ret < 0) {
|
if (ret < 0) {
|
||||||
fprintf(stderr, "Failed generating trees\n");
|
fprintf(stderr, "Failed generating trees\n");
|
||||||
return -1;
|
return -1;
|
||||||
}
|
}
|
||||||
|
|
||||||
const char *mode = argv[1];
|
|
||||||
ret = strcmp(mode, SERVER_MODE);
|
|
||||||
if (ret == 0)
|
|
||||||
return do_srv();
|
return do_srv();
|
||||||
|
|
||||||
ret = strcmp(mode, CLIENT_MODE);
|
|
||||||
if (ret == 0)
|
|
||||||
return do_clt();
|
|
||||||
|
|
||||||
fprintf(stderr, "Unknown proxy mode\n");
|
|
||||||
|
|
||||||
fretres();
|
fretres();
|
||||||
}
|
}
|
||||||
|
|
31
proxlib.h
31
proxlib.h
|
@ -7,23 +7,8 @@
|
||||||
#define MAX_BOUND 10000
|
#define MAX_BOUND 10000
|
||||||
#endif
|
#endif
|
||||||
|
|
||||||
#define SERVER_MODE "server"
|
|
||||||
#define CLIENT_MODE "client"
|
|
||||||
|
|
||||||
#define PROXY_PORT 2020
|
#define PROXY_PORT 2020
|
||||||
#define PROXY_CONN 20
|
#define PROXY_CONN 20
|
||||||
#define PROXY_MAX_MSGLEN 10000*1024
|
|
||||||
#define PROXY_DEF_PORT "80"
|
|
||||||
#define PROXY_BASE_PORT 10
|
|
||||||
|
|
||||||
#define REGEX_MATCHN 4
|
|
||||||
#define REGEX_TITLE "^([A-Z]+)[ ]+([a-zA-Z0-9\\:/_.,-]+)"\
|
|
||||||
"[ ]+([a-zA-Z0-9_.,/-]+)[\n\r]*$"
|
|
||||||
#define REGEX_HEADER "^([a-zA-Z0-9_-]*):[ \t]+([^\r\n]*)"
|
|
||||||
#define REGEX_HOST "^([a-zA-Z0-9_/,.-]+)(:[0-9]+)?$"
|
|
||||||
|
|
||||||
#define CLIENT_MESSAGE "GET http://archive.0xdeadbeer.org/ HTTP/1.0\r\n\r\n"\
|
|
||||||
"Host: archive.0xdeadbeer.org\r\n"\
|
|
||||||
|
|
||||||
#define STATEM_RCV_CLT 0b00000001
|
#define STATEM_RCV_CLT 0b00000001
|
||||||
#define STATEM_CON_SRV 0b00000010
|
#define STATEM_CON_SRV 0b00000010
|
||||||
|
@ -32,22 +17,6 @@
|
||||||
#define STATEM_FWD_CLT 0b00010000
|
#define STATEM_FWD_CLT 0b00010000
|
||||||
#define STATEM_ERR 0b00100000
|
#define STATEM_ERR 0b00100000
|
||||||
|
|
||||||
struct header {
|
|
||||||
char *key;
|
|
||||||
char *value;
|
|
||||||
};
|
|
||||||
|
|
||||||
struct request {
|
|
||||||
char *host_name;
|
|
||||||
char *host_port;
|
|
||||||
|
|
||||||
char *method;
|
|
||||||
char *uri;
|
|
||||||
char *ver;
|
|
||||||
int header_num;
|
|
||||||
struct header *headers;
|
|
||||||
};
|
|
||||||
|
|
||||||
struct conn {
|
struct conn {
|
||||||
int cltfd;
|
int cltfd;
|
||||||
int srvfd;
|
int srvfd;
|
||||||
|
|
Loading…
Reference in New Issue
Block a user