proxy: add basic title parsing
This commit is contained in:
parent
87d75b7f48
commit
5c5cda5f51
1
.gitignore
vendored
1
.gitignore
vendored
|
@ -1,2 +1,3 @@
|
||||||
*.o
|
*.o
|
||||||
proxy
|
proxy
|
||||||
|
compile_commands.json
|
||||||
|
|
2
Makefile
2
Makefile
|
@ -1,4 +1,4 @@
|
||||||
CC=gcc
|
CC=bear --append -- gcc
|
||||||
CFLAGS= -g -Wall -Werror
|
CFLAGS= -g -Wall -Werror
|
||||||
|
|
||||||
all: proxy
|
all: proxy
|
||||||
|
|
3
README
3
README
|
@ -4,7 +4,8 @@ CUSTOM PROXY
|
||||||
|
|
||||||
Proxy for HTTP/1.0 standard protocol
|
Proxy for HTTP/1.0 standard protocol
|
||||||
|
|
||||||
RFC: datatracker.ietf.org/doc/html/rfc2616
|
RFC: datatracker.ietf.org/doc/html/rfc1945
|
||||||
RES: cs.princeton.edu/courses/archive/spr13/cos461/assignments-proxy.html
|
RES: cs.princeton.edu/courses/archive/spr13/cos461/assignments-proxy.html
|
||||||
|
RES: Beej's Guide to Network Programming - Using Internet Sockets
|
||||||
RES: en.wikipedia.org/wiki/Proxy_server
|
RES: en.wikipedia.org/wiki/Proxy_server
|
||||||
RES: TCP/IP sockets in C - Practical guide for programmers 2nd edition
|
RES: TCP/IP sockets in C - Practical guide for programmers 2nd edition
|
||||||
|
|
110
proxy.c
110
proxy.c
|
@ -1,16 +1,104 @@
|
||||||
#include "proxy_parse.h"
|
|
||||||
#include <stdio.h>
|
#include <stdio.h>
|
||||||
#include <stdlib.h>
|
#include <stdlib.h>
|
||||||
|
#include <unistd.h>
|
||||||
|
#include <string.h>
|
||||||
#include <sys/types.h>
|
#include <sys/types.h>
|
||||||
#include <sys/socket.h>
|
#include <sys/socket.h>
|
||||||
#include <netdb.h>
|
#include <netdb.h>
|
||||||
|
#include "structs.h"
|
||||||
|
|
||||||
#define PROXY_PORT 80
|
#define PROXY_PORT 2020
|
||||||
#define PROXY_CONN 20
|
#define PROXY_CONN 20
|
||||||
|
#define PROXY_MAX_MSGLEN 10*1024
|
||||||
|
#define TITLE_DELIM " "
|
||||||
|
|
||||||
|
struct http_msg *child_msg;
|
||||||
|
|
||||||
|
int parse_title(char *msgbuff) {
|
||||||
|
char *title_end = strrchr(msgbuff, '\n');
|
||||||
|
if (title_end == NULL) {
|
||||||
|
return -1;
|
||||||
|
}
|
||||||
|
|
||||||
|
const int title_len = title_end - msgbuff;
|
||||||
|
char title[title_len];
|
||||||
|
strncpy(title, msgbuff, title_len);
|
||||||
|
|
||||||
|
char *title_sub;
|
||||||
|
int index;
|
||||||
|
for (index = 0, title_sub = strtok(title, TITLE_DELIM);
|
||||||
|
title_sub != NULL;
|
||||||
|
title_sub = strtok(NULL, TITLE_DELIM), index++) {
|
||||||
|
|
||||||
|
char *destarr = (char *) calloc(1, strlen(title_sub)+1);
|
||||||
|
if (destarr == NULL) {
|
||||||
|
goto error_title_props;
|
||||||
|
}
|
||||||
|
|
||||||
|
if (index == 0) {
|
||||||
|
child_msg->method = destarr;
|
||||||
|
strcpy(child_msg->method, title_sub);
|
||||||
|
}
|
||||||
|
else if (index == 1) {
|
||||||
|
child_msg->uri = destarr;
|
||||||
|
strcpy(child_msg->uri, title_sub);
|
||||||
|
}
|
||||||
|
else if (index == 2) {
|
||||||
|
child_msg->ver = destarr;
|
||||||
|
strcpy(child_msg->ver, title_sub);
|
||||||
|
}
|
||||||
|
else {
|
||||||
|
goto error_title_props;
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
return 0;
|
||||||
|
|
||||||
|
error_title_props:
|
||||||
|
for (int i = 0; i <= index; i++) {
|
||||||
|
free(child_msg+i);
|
||||||
|
}
|
||||||
|
|
||||||
|
return -1;
|
||||||
|
}
|
||||||
|
|
||||||
|
void handle_request(int sockfd) {
|
||||||
|
int ret;
|
||||||
|
int id = getpid();
|
||||||
|
char msgbuff[PROXY_MAX_MSGLEN];
|
||||||
|
memset(msgbuff, 0, sizeof(msgbuff));
|
||||||
|
|
||||||
|
ret = recv(sockfd, msgbuff, sizeof(msgbuff), 0);
|
||||||
|
if (ret < 0) {
|
||||||
|
fprintf(stderr, "[CHILD %d] Failed to receive data from client\n", id);
|
||||||
|
goto end_sock;
|
||||||
|
}
|
||||||
|
|
||||||
|
fprintf(stdout, "[CHILD %d] Received data from client: %s\n", id, msgbuff);
|
||||||
|
|
||||||
|
// prepare structs
|
||||||
|
child_msg = (struct http_msg *) calloc(1, sizeof(struct http_msg));
|
||||||
|
if (child_msg == NULL) {
|
||||||
|
fprintf(stderr, "[CHILD %d] Failed to allocate memory for client structs\n", id);
|
||||||
|
goto end_sock;
|
||||||
|
}
|
||||||
|
|
||||||
|
// start parsing
|
||||||
|
ret = parse_title(msgbuff);
|
||||||
|
if (ret < 0) {
|
||||||
|
fprintf(stderr, "[CHILD %d] Failed to parse the title of the request\n", id);
|
||||||
|
goto end_structs;
|
||||||
|
}
|
||||||
|
|
||||||
|
end_structs:
|
||||||
|
free(child_msg);
|
||||||
|
|
||||||
|
end_sock:
|
||||||
|
close(sockfd);
|
||||||
|
}
|
||||||
|
|
||||||
int main(int argc, char *argv[]) {
|
int main(int argc, char *argv[]) {
|
||||||
int server_socket;
|
int server_socket;
|
||||||
int new_fd;
|
|
||||||
int ret;
|
int ret;
|
||||||
|
|
||||||
ret = server_socket = socket(AF_INET, SOCK_STREAM, IPPROTO_TCP);
|
ret = server_socket = socket(AF_INET, SOCK_STREAM, IPPROTO_TCP);
|
||||||
|
@ -50,10 +138,20 @@ int main(int argc, char *argv[]) {
|
||||||
return EXIT_FAILURE;
|
return EXIT_FAILURE;
|
||||||
}
|
}
|
||||||
|
|
||||||
fprintf(stdout, "[CLIENT %d] Successfully connected\n", client_socket);
|
ret = fork();
|
||||||
|
switch (ret) {
|
||||||
|
case -1:
|
||||||
|
fprintf(stderr, "[CLIENT SOCKET %d] Failed to fork child process to handle the request\n", client_socket);
|
||||||
|
return EXIT_FAILURE;
|
||||||
|
break;
|
||||||
|
case 0:
|
||||||
|
handle_request(client_socket);
|
||||||
|
break;
|
||||||
|
default:
|
||||||
|
fprintf(stdout, "[PROGRAM] Successfully forked a new child process with PID %d\n", ret);
|
||||||
|
break;
|
||||||
|
}
|
||||||
}
|
}
|
||||||
|
|
||||||
printf("New connection with fd -> %d\n", new_fd);
|
|
||||||
|
|
||||||
return EXIT_SUCCESS;
|
return EXIT_SUCCESS;
|
||||||
}
|
}
|
||||||
|
|
Loading…
Reference in New Issue
Block a user