first commit
This commit is contained in:
3
CS4210/Project 2/proxy_server/.cvsignore
Normal file
3
CS4210/Project 2/proxy_server/.cvsignore
Normal file
@@ -0,0 +1,3 @@
|
||||
Debug
|
||||
Release
|
||||
bin
|
||||
4
CS4210/Project 2/proxy_server/CVS/Entries
Normal file
4
CS4210/Project 2/proxy_server/CVS/Entries
Normal file
@@ -0,0 +1,4 @@
|
||||
D/src////
|
||||
/.cvsignore/1.1/Sat Mar 11 23:56:57 2006//
|
||||
/Makefile/1.8/Tue Mar 28 01:11:53 2006//
|
||||
/l33t_proxy.conf/1.7/Tue Mar 28 02:40:31 2006//
|
||||
4
CS4210/Project 2/proxy_server/CVS/Entries.Extra
Normal file
4
CS4210/Project 2/proxy_server/CVS/Entries.Extra
Normal file
@@ -0,0 +1,4 @@
|
||||
D/src///////
|
||||
/.cvsignore////*///
|
||||
/Makefile////*///
|
||||
/l33t_proxy.conf////*///
|
||||
4
CS4210/Project 2/proxy_server/CVS/Entries.Extra.Old
Normal file
4
CS4210/Project 2/proxy_server/CVS/Entries.Extra.Old
Normal file
@@ -0,0 +1,4 @@
|
||||
D/src///////
|
||||
/.cvsignore////*///
|
||||
/l33t_proxy.conf////*///
|
||||
/Makefile////*///
|
||||
4
CS4210/Project 2/proxy_server/CVS/Entries.Old
Normal file
4
CS4210/Project 2/proxy_server/CVS/Entries.Old
Normal file
@@ -0,0 +1,4 @@
|
||||
D/src////
|
||||
/.cvsignore/1.1/Sat Mar 11 23:56:57 2006//
|
||||
/l33t_proxy.conf/1.6/Tue Mar 28 00:07:59 2006//
|
||||
/Makefile/1.8/Tue Mar 28 01:11:53 2006//
|
||||
1
CS4210/Project 2/proxy_server/CVS/Repository
Normal file
1
CS4210/Project 2/proxy_server/CVS/Repository
Normal file
@@ -0,0 +1 @@
|
||||
CS4210/Project 2/proxy_server
|
||||
1
CS4210/Project 2/proxy_server/CVS/Root
Normal file
1
CS4210/Project 2/proxy_server/CVS/Root
Normal file
@@ -0,0 +1 @@
|
||||
:ext:asskoala@192.168.0.3:/usr/_CVS
|
||||
43
CS4210/Project 2/proxy_server/Makefile
Normal file
43
CS4210/Project 2/proxy_server/Makefile
Normal file
@@ -0,0 +1,43 @@
|
||||
CC = gcc
|
||||
OPT = -O2 -fomit-frame-pointer
|
||||
LIBS = -lpthread
|
||||
TARGET = l33t_proxy
|
||||
MISC = Makefile
|
||||
INCDIR = src/include
|
||||
SRCDIR = src
|
||||
BINDIR = ../bin/proxy
|
||||
COMMONDIR = ../common
|
||||
MAKE = make
|
||||
|
||||
all: $(TARGET)
|
||||
|
||||
$(TARGET): $(BINDIR)/parser.o $(BINDIR)/proxy_settings.o $(BINDIR)/http.o $(BINDIR)/thread_pool.o $(BINDIR)/networking.o $(BINDIR)/threads.o $(BINDIR)/shared_memory.o $(BINDIR)/main.o
|
||||
$(CC) $(OPT) $(LIBS) -o $(BINDIR)/$(TARGET) $(BINDIR)/*.o
|
||||
|
||||
$(BINDIR)/parser.o: $(COMMONDIR)/config_parser/parser.h $(COMMONDIR)/config_parser/parser.c
|
||||
-mkdir -p $(BINDIR) > /dev/null
|
||||
$(CC) $(OPT) -c -o $(BINDIR)/parser.o $(COMMONDIR)/config_parser/parser.c
|
||||
|
||||
$(BINDIR)/proxy_settings.o: $(SRCDIR)/proxy_settings.c $(INCDIR)/proxy_settings.h
|
||||
$(CC) $(OPT) -c -o $(BINDIR)/proxy_settings.o $(SRCDIR)/proxy_settings.c
|
||||
|
||||
$(BINDIR)/http.o: $(COMMONDIR)/http/http.h $(COMMONDIR)/http/http.c
|
||||
$(CC) $(OPT) -c -o $(BINDIR)/http.o $(COMMONDIR)/http/http.c
|
||||
|
||||
$(BINDIR)/thread_pool.o: $(COMMONDIR)/thread_pool/thread_pool.h $(COMMONDIR)/thread_pool/thread_pool.c
|
||||
$(CC) $(OPT) -c -o $(BINDIR)/thread_pool.o $(COMMONDIR)/thread_pool/thread_pool.c
|
||||
|
||||
$(BINDIR)/networking.o: $(COMMONDIR)/networking/networking.h $(COMMONDIR)/networking/networking.c
|
||||
$(CC) $(OPT) -c -o $(BINDIR)/networking.o $(COMMONDIR)/networking/networking.c
|
||||
|
||||
$(BINDIR)/threads.o: $(SRCDIR)/threads.c $(INCDIR)/threads.h
|
||||
$(CC) $(OPT) -c -o $(BINDIR)/threads.o $(SRCDIR)/threads.c
|
||||
|
||||
$(BINDIR)/shared_memory.o: $(COMMONDIR)/shared_memory/shared_memory.c $(COMMONDIR)/shared_memory/shared_memory.h
|
||||
$(CC) $(OPT) -c -o $(BINDIR)/shared_memory.o $(COMMONDIR)/shared_memory/shared_memory.c
|
||||
|
||||
$(BINDIR)/main.o: $(SRCDIR)/main.c
|
||||
$(CC) $(OPT) -c -o $(BINDIR)/main.o $(SRCDIR)/main.c
|
||||
|
||||
clean:
|
||||
rm -f $(BINDIR)/*.o $(BINDIR)/$(TARGET)
|
||||
4
CS4210/Project 2/proxy_server/l33t_proxy.conf
Normal file
4
CS4210/Project 2/proxy_server/l33t_proxy.conf
Normal file
@@ -0,0 +1,4 @@
|
||||
MAX_THREADS:16
|
||||
PORT_NUMBER:1338
|
||||
|
||||
|
||||
4
CS4210/Project 2/proxy_server/src/CVS/Entries
Normal file
4
CS4210/Project 2/proxy_server/src/CVS/Entries
Normal file
@@ -0,0 +1,4 @@
|
||||
D/include////
|
||||
/main.c/1.11/Tue Mar 28 00:07:59 2006//
|
||||
/proxy_settings.c/1.3/Tue Mar 28 00:07:59 2006//
|
||||
/threads.c/1.25/Tue Mar 28 02:40:31 2006//
|
||||
4
CS4210/Project 2/proxy_server/src/CVS/Entries.Extra
Normal file
4
CS4210/Project 2/proxy_server/src/CVS/Entries.Extra
Normal file
@@ -0,0 +1,4 @@
|
||||
D/include///////
|
||||
/main.c////*///
|
||||
/proxy_settings.c////*///
|
||||
/threads.c////*///
|
||||
4
CS4210/Project 2/proxy_server/src/CVS/Entries.Extra.Old
Normal file
4
CS4210/Project 2/proxy_server/src/CVS/Entries.Extra.Old
Normal file
@@ -0,0 +1,4 @@
|
||||
D/include///////
|
||||
/main.c////*///
|
||||
/proxy_settings.c////*///
|
||||
/threads.c////*///
|
||||
4
CS4210/Project 2/proxy_server/src/CVS/Entries.Old
Normal file
4
CS4210/Project 2/proxy_server/src/CVS/Entries.Old
Normal file
@@ -0,0 +1,4 @@
|
||||
D/include////
|
||||
/main.c/1.11/Tue Mar 28 00:07:59 2006//
|
||||
/proxy_settings.c/1.3/Tue Mar 28 00:07:59 2006//
|
||||
/threads.c/1.24/Tue Mar 28 00:07:59 2006//
|
||||
1
CS4210/Project 2/proxy_server/src/CVS/Repository
Normal file
1
CS4210/Project 2/proxy_server/src/CVS/Repository
Normal file
@@ -0,0 +1 @@
|
||||
CS4210/Project 2/proxy_server/src
|
||||
1
CS4210/Project 2/proxy_server/src/CVS/Root
Normal file
1
CS4210/Project 2/proxy_server/src/CVS/Root
Normal file
@@ -0,0 +1 @@
|
||||
:ext:asskoala@192.168.0.3:/usr/_CVS
|
||||
3
CS4210/Project 2/proxy_server/src/include/CVS/Entries
Normal file
3
CS4210/Project 2/proxy_server/src/include/CVS/Entries
Normal file
@@ -0,0 +1,3 @@
|
||||
/proxy_settings.h/1.3/Tue Mar 28 00:07:59 2006//
|
||||
/threads.h/1.2/Tue Mar 28 00:07:59 2006//
|
||||
D
|
||||
@@ -0,0 +1,2 @@
|
||||
/proxy_settings.h////*///
|
||||
/threads.h////*///
|
||||
@@ -0,0 +1,2 @@
|
||||
/proxy_settings.h////*///
|
||||
/threads.h////*///
|
||||
@@ -0,0 +1,3 @@
|
||||
/proxy_settings.h/1.2/Sat Mar 11 22:16:55 2006//
|
||||
/threads.h/1.1/Sat Mar 11 23:56:40 2006//
|
||||
D
|
||||
1
CS4210/Project 2/proxy_server/src/include/CVS/Repository
Normal file
1
CS4210/Project 2/proxy_server/src/include/CVS/Repository
Normal file
@@ -0,0 +1 @@
|
||||
CS4210/Project 2/proxy_server/src/include
|
||||
1
CS4210/Project 2/proxy_server/src/include/CVS/Root
Normal file
1
CS4210/Project 2/proxy_server/src/include/CVS/Root
Normal file
@@ -0,0 +1 @@
|
||||
:ext:asskoala@192.168.0.3:/usr/_CVS
|
||||
21
CS4210/Project 2/proxy_server/src/include/proxy_settings.h
Normal file
21
CS4210/Project 2/proxy_server/src/include/proxy_settings.h
Normal file
@@ -0,0 +1,21 @@
|
||||
#ifndef _PROXY_SETTINGS_H_
|
||||
#define _PROXY_SETTINGS_H_
|
||||
|
||||
/////////////
|
||||
// Globals //
|
||||
/////////////
|
||||
|
||||
typedef struct {
|
||||
char* LOAD_DIR; // Server home directory
|
||||
char* CONFIG_FILE; // Config File, should be constant
|
||||
int MAX_THREADS; // Maximum number of threads to spawn
|
||||
unsigned short PORT_NUMBER; // Port number to use
|
||||
char useSHMEM;
|
||||
} settings_t;
|
||||
///////////////
|
||||
// Functions //
|
||||
///////////////
|
||||
|
||||
int init_settings(settings_t *s, int, char**);
|
||||
|
||||
#endif
|
||||
9
CS4210/Project 2/proxy_server/src/include/threads.h
Normal file
9
CS4210/Project 2/proxy_server/src/include/threads.h
Normal file
@@ -0,0 +1,9 @@
|
||||
#ifndef _THREADS_H_
|
||||
#define _THREADS_H_
|
||||
|
||||
void *consumer(void*);
|
||||
void *producer(void* v_port_number);
|
||||
|
||||
void *consumerNoNoob(void* dumb);
|
||||
|
||||
#endif
|
||||
80
CS4210/Project 2/proxy_server/src/main.c
Normal file
80
CS4210/Project 2/proxy_server/src/main.c
Normal file
@@ -0,0 +1,80 @@
|
||||
#include <stdio.h>
|
||||
#include <stdlib.h>
|
||||
|
||||
#include <unistd.h>
|
||||
#include <sys/types.h>
|
||||
#include <signal.h>
|
||||
|
||||
#include "../../common/defs.h"
|
||||
#include "../../common/thread_pool/thread_pool.h"
|
||||
#include "../../common/shared_memory/shared_memory.h"
|
||||
|
||||
#include "include/threads.h"
|
||||
#include "include/proxy_settings.h"
|
||||
|
||||
|
||||
|
||||
/* signal handler for SIGINT */
|
||||
void catch_int(int sig_num)
|
||||
{
|
||||
// TODO: Insert memory freeing stuff here
|
||||
printf("Signal %d received\n", sig_num);
|
||||
printf("Destroying the Shared Memory Segments...\n");
|
||||
detach_shared_memory();
|
||||
printf("Done!\n");
|
||||
|
||||
exit(1);
|
||||
}
|
||||
|
||||
int main(int argc, char** argv) {
|
||||
|
||||
settings_t settings;
|
||||
|
||||
// Mask SIGPIPE
|
||||
#ifndef LINUX
|
||||
{
|
||||
sigset_t newmask;
|
||||
sigset_t oldmask;
|
||||
sigemptyset(&newmask);
|
||||
sigaddset(&newmask, SIGPIPE);
|
||||
if (sigprocmask(SIG_BLOCK, &newmask, &oldmask) < 0)
|
||||
{
|
||||
fprintf(stderr, "can't set signal mask\n");
|
||||
exit(1);
|
||||
}
|
||||
}
|
||||
#endif
|
||||
|
||||
// Initialize proxy server settings
|
||||
init_settings(&settings, argc, argv);
|
||||
|
||||
printf("** Using %d threads\n",settings.MAX_THREADS);
|
||||
printf("** Using home_dir: %s\n",settings.LOAD_DIR);
|
||||
printf( (settings.useSHMEM == 0 ? "** Not " : "** "));
|
||||
printf( "Using Shared Memory.\n");
|
||||
|
||||
if (settings.useSHMEM) {
|
||||
// Mask signals (fancy, isn't it?)
|
||||
signal(SIGINT, catch_int);
|
||||
signal(SIGABRT, catch_int);
|
||||
signal(SIGSEGV, catch_int);
|
||||
signal(SIGTERM, catch_int);
|
||||
signal(SIGBUS, catch_int);
|
||||
|
||||
attach_shared_memory();
|
||||
}
|
||||
|
||||
// Initialize threads
|
||||
init_thread_pool(&(settings.PORT_NUMBER), settings.MAX_THREADS,
|
||||
producer, consumerNoNoob, &settings);
|
||||
|
||||
// Wait for input
|
||||
while (getchar() != 'q');
|
||||
|
||||
if (settings.useSHMEM) {
|
||||
catch_int(SIGINT);
|
||||
}
|
||||
|
||||
return EXIT_SUCCESS;
|
||||
}
|
||||
|
||||
93
CS4210/Project 2/proxy_server/src/proxy_settings.c
Normal file
93
CS4210/Project 2/proxy_server/src/proxy_settings.c
Normal file
@@ -0,0 +1,93 @@
|
||||
#include "include/proxy_settings.h"
|
||||
#include "../../common/config_parser/parser.h"
|
||||
#include "../../common/defs.h"
|
||||
|
||||
#include <stdio.h>
|
||||
#include <string.h>
|
||||
|
||||
|
||||
/*
|
||||
* Set the default values for all the variables
|
||||
*/
|
||||
static void setDefaults(settings_t *s) {
|
||||
s->LOAD_DIR = ".";
|
||||
s->CONFIG_FILE = "l33t_proxy.conf";
|
||||
s->MAX_THREADS = 7;
|
||||
s->PORT_NUMBER = 1338;
|
||||
s->useSHMEM = 0;
|
||||
}
|
||||
|
||||
/*
|
||||
* Initialize internal server parameters
|
||||
*/
|
||||
int init_settings(settings_t *s, int argc, char** argv) {
|
||||
ConfigFile* configFile;
|
||||
|
||||
VPRINTF(("** Reading Config File **\n\n"));
|
||||
|
||||
setDefaults(s);
|
||||
|
||||
switch (argc) {
|
||||
case 3:
|
||||
s->CONFIG_FILE = argv[2];
|
||||
s->useSHMEM = atoi(argv[1]);
|
||||
break;
|
||||
case 2:
|
||||
s->useSHMEM = atoi(argv[1]);
|
||||
break;
|
||||
default:
|
||||
cmdline_error:
|
||||
printf("Proper Usage: \n");
|
||||
printf(" ./l33t_proxy use_shared_memory [configFile]\n");
|
||||
printf(" e.g. ./l33t_proxy 1\n");
|
||||
printf(" e.g. ./l33t_proxy 0 l33t_proxy2.conf\n");
|
||||
exit(1);
|
||||
}
|
||||
|
||||
if (s->useSHMEM != 0 && s->useSHMEM != 1) goto cmdline_error;
|
||||
|
||||
configFile = loadFile(s->CONFIG_FILE);
|
||||
|
||||
if (!configFile) {
|
||||
printf("*** configuration file \"%s\" not found.\n",s->CONFIG_FILE);
|
||||
exit(1);
|
||||
}
|
||||
|
||||
while (nextSet(configFile)) {
|
||||
|
||||
if (!strcmp(configFile->currentParamter, "LOAD_DIR")) {
|
||||
VPRINTF(("*** Read LOAD_FILE Value: %s\n",configFile->currentValue));
|
||||
s->LOAD_DIR = malloc(strlen(configFile->currentValue)+1);
|
||||
strcpy(s->LOAD_DIR, configFile->currentValue);
|
||||
|
||||
} else if(!strcmp(configFile->currentParamter, "CONFIG_FILE")) {
|
||||
VPRINTF(("*** Read CONFIG_FILE Value: %s\n",configFile->currentValue));
|
||||
s->CONFIG_FILE = malloc(strlen(configFile->currentValue)+1);
|
||||
strcpy(s->CONFIG_FILE, configFile->currentValue);
|
||||
|
||||
} else if(!strcmp(configFile->currentParamter, "MAX_THREADS")) {
|
||||
s->MAX_THREADS = atoi(configFile->currentValue);
|
||||
|
||||
// In case some idiot tries to set the MAX_THREADS <=0
|
||||
if (s->MAX_THREADS < 1) {
|
||||
s->MAX_THREADS = 1;
|
||||
}
|
||||
|
||||
VPRINTF(("*** Read MAX_THREADS Value: %d\n",s->MAX_THREADS));
|
||||
|
||||
} else if(!strcmp(configFile->currentParamter, "PORT_NUMBER")) {
|
||||
s->PORT_NUMBER = atoi(configFile->currentValue);
|
||||
VPRINTF(("*** Read PORT_NUMBER Value: %d\n",s->PORT_NUMBER));
|
||||
|
||||
} else {
|
||||
printf("**** Illegal Parameter Exception in Config File:\n");
|
||||
printf("**** \"%s:%s\"\n", configFile->currentParamter, configFile->currentValue);
|
||||
printf("**** is not a legal Configuration Parameter.\n");
|
||||
}
|
||||
}
|
||||
|
||||
VPRINTF(("\n** Done Reading Config File **\n\n"));
|
||||
|
||||
return 0;
|
||||
}
|
||||
|
||||
667
CS4210/Project 2/proxy_server/src/threads.c
Normal file
667
CS4210/Project 2/proxy_server/src/threads.c
Normal file
@@ -0,0 +1,667 @@
|
||||
|
||||
#include <unistd.h> /* close() */
|
||||
#include <sys/socket.h> /* Socket Functions */
|
||||
#include <sys/types.h> /* Socket Datatypes */
|
||||
#include <netinet/in.h> /* IP Datatypes */
|
||||
#include <arpa/inet.h> /* Address Structs */
|
||||
#include <netdb.h>
|
||||
|
||||
#include <stdio.h> /* Standard IO */
|
||||
#include <stdlib.h> /* Standard Lib */
|
||||
|
||||
#include <sys/ipc.h>
|
||||
#include <sys/sem.h>
|
||||
|
||||
#include <string.h> /* String Libs */
|
||||
|
||||
#include "include/threads.h" /* My Functions */
|
||||
#include "include/proxy_settings.h"
|
||||
#include "../../common/defs.h" /* Definitions */
|
||||
#include "../../common/http/http.h"
|
||||
#include "../../common/networking/networking.h"
|
||||
#include "../../common/shared_memory/shared_memory.h"
|
||||
|
||||
#include <pthread.h>
|
||||
|
||||
#define HTTP_REQUEST_SIZE 1024
|
||||
#define RECV_SIZE 64
|
||||
#define PROTO "http://"
|
||||
|
||||
char *strfindr(char *dest, char *src, char *find, char *replace);
|
||||
char *itoa(int);
|
||||
|
||||
char *acceptClientRequest(int clntSocket) {
|
||||
char* received;
|
||||
char* temp;
|
||||
int len, i;
|
||||
|
||||
uber_recv(clntSocket, BUFFERSIZE, "\0", &temp);
|
||||
received = temp;
|
||||
|
||||
i=1;
|
||||
while( len = uber_recv(clntSocket, BUFFERSIZE, "\0", &temp) && 0 > len ) {
|
||||
received = realloc(received, (len + BUFFERSIZE)*sizeof(char));
|
||||
memcpy(received+(BUFFERSIZE*i), temp, len);
|
||||
free(temp);
|
||||
}
|
||||
|
||||
return received;
|
||||
}
|
||||
|
||||
void dispatchLocal(char *clntRequest, int clntSocket) {
|
||||
char *get_local;
|
||||
int i, length;
|
||||
|
||||
int servSocket;
|
||||
|
||||
shmemNode* shmemSegment;
|
||||
|
||||
get_local = malloc(sizeof(char) * (strlen(LOCAL_GET) + 1) );
|
||||
strcpy(get_local, LOCAL_GET);
|
||||
|
||||
length = strlen(get_local);
|
||||
get_local[length] = ' '; length++;
|
||||
|
||||
clntRequest = strstr(clntRequest, "localhost");
|
||||
|
||||
for (; *clntRequest != '/'; clntRequest++);
|
||||
clntRequest++;
|
||||
|
||||
for (; *clntRequest != ' '; length++, clntRequest++) {
|
||||
get_local = realloc(get_local, sizeof(char)*(length+1));
|
||||
get_local[length] = *clntRequest;
|
||||
}
|
||||
|
||||
get_local = realloc(get_local, sizeof(char) * (length+2));
|
||||
get_local[length] = ' '; length++;
|
||||
|
||||
shmemSegment = getSharedMemorySegment();
|
||||
|
||||
//printf("semIndex: %d\n",shmemSegment->semIndex);
|
||||
|
||||
{
|
||||
char *buff;
|
||||
|
||||
buff = itoa(shmemSegment->semIndex);
|
||||
|
||||
get_local = realloc(get_local, sizeof(char) * (length+strlen(buff)+1));
|
||||
|
||||
for (i=0; buff[i] != '\0'; i++) {
|
||||
get_local[length+i] = buff[i];
|
||||
}
|
||||
get_local[length+i] = buff[i];
|
||||
|
||||
free(buff);
|
||||
|
||||
}
|
||||
|
||||
//printf("get_local: %s\n", get_local);
|
||||
//printf("length: %d %d\n", length, strlen(get_local));
|
||||
//printf("..Done\n");
|
||||
|
||||
servSocket = createTCPSocket("127.0.0.1", 1337);
|
||||
if (servSocket < 0) {
|
||||
return;
|
||||
}
|
||||
|
||||
//Send to server
|
||||
if (send(servSocket, get_local, strlen(get_local), 0) < 0) {
|
||||
printf("Send error to server\n");
|
||||
close(servSocket);
|
||||
return;
|
||||
}
|
||||
|
||||
// Send stuff Back to the client
|
||||
{
|
||||
struct sembuf sembuffer;
|
||||
|
||||
//printf("Receiving from server.\n");
|
||||
|
||||
// Get semaphore
|
||||
sembuffer.sem_num = shmemSegment->semIndex;
|
||||
sembuffer.sem_op = -1;
|
||||
sembuffer.sem_flg = 0;
|
||||
|
||||
for (;;) {
|
||||
|
||||
// Wait for semprox to be 1
|
||||
//printf("Waiting for semprox to be 1\n");
|
||||
sembuffer.sem_op = -1;
|
||||
semop(semprox, &sembuffer, 1);
|
||||
|
||||
if (*((int*)shmemSegment->buff) == -1) {
|
||||
// Set semserv to 1
|
||||
sembuffer.sem_op = 1;
|
||||
semop(semserv, &sembuffer, 1);
|
||||
break;
|
||||
}
|
||||
|
||||
// Send back to client
|
||||
if (send(clntSocket,
|
||||
shmemSegment->buff+sizeof(int),
|
||||
*((int*)shmemSegment->buff),
|
||||
0) < 0)
|
||||
{
|
||||
printf("Send Error\n");
|
||||
}
|
||||
|
||||
//printf("Setting semserv to 1\n");
|
||||
// Set semserv to 1
|
||||
sembuffer.sem_op = 1;
|
||||
semop(semserv, &sembuffer, 1);
|
||||
|
||||
}
|
||||
}
|
||||
|
||||
releaseSharedMemorySegment(shmemSegment);
|
||||
|
||||
}
|
||||
|
||||
|
||||
void *consumerNoNoob(void* dumb) {
|
||||
|
||||
settings_t *settings = (settings_t*) dumb;
|
||||
char isLocal = 0;
|
||||
|
||||
#if _LOG_
|
||||
printf("[CONSUMER] Consumer %ld started\n", (long)pthread_self());
|
||||
fflush(stdout);
|
||||
#endif
|
||||
|
||||
//Get stuff from the client
|
||||
for (;;) {
|
||||
int clntSocket, servSocket;
|
||||
char *clntRequest = NULL, *servHostName;
|
||||
struct addrinfo *addressInfo;
|
||||
|
||||
clntSocket = removeSocket();
|
||||
|
||||
//Receive from Client
|
||||
{ //1
|
||||
char buffer[RECV_SIZE];
|
||||
int i = 1;
|
||||
|
||||
while (recv(clntSocket, buffer, RECV_SIZE, 0) > 0) {
|
||||
//printf("receiving\n");
|
||||
clntRequest = realloc(clntRequest, (i*RECV_SIZE)*sizeof(char));
|
||||
memcpy(clntRequest+((i-1)*RECV_SIZE), buffer, RECV_SIZE);
|
||||
if (strstr(clntRequest, "\r\n\r\n\0")) break; //FIXME: Fix my speed
|
||||
i++;
|
||||
}
|
||||
//printf("Got:\n%s\n", clntRequest);
|
||||
|
||||
// if Shared Memory is active, all localhost requests are
|
||||
// routed to the shared memory modules
|
||||
if (settings->useSHMEM && strstr(clntRequest, "localhost")) {
|
||||
dispatchLocal(clntRequest, clntSocket);
|
||||
close(clntSocket);
|
||||
continue;
|
||||
} else if (strstr(clntRequest, "localhost")) {
|
||||
//We're going to assume that we're running l33t_server
|
||||
// on port 1337 and thus skip the host resolution...stuff
|
||||
isLocal = 1;
|
||||
}
|
||||
|
||||
} //1
|
||||
|
||||
//Figure out what to get
|
||||
{ //2
|
||||
char *request, *temp = NULL;
|
||||
int i;
|
||||
|
||||
request = strstr(clntRequest, PROTO);
|
||||
if (!request) {
|
||||
printf("Bad Request on socket: %d\n",clntSocket);
|
||||
free(clntRequest);
|
||||
close(clntSocket);
|
||||
continue;
|
||||
}
|
||||
|
||||
request += strlen(PROTO);
|
||||
|
||||
for (i=0; request[i] != '/'; i++) {
|
||||
temp = realloc(temp, (i+1)*sizeof(char));
|
||||
|
||||
temp[i] = request[i];
|
||||
}
|
||||
|
||||
temp = realloc(temp, (i+1)*sizeof(char));
|
||||
temp[i] = '\0';
|
||||
//printf("temp: %s\n",temp);
|
||||
|
||||
servHostName = temp;
|
||||
|
||||
} //2
|
||||
|
||||
{ //5
|
||||
char *temp;
|
||||
|
||||
temp = malloc(strlen(PROTO)+strlen(servHostName)+1);
|
||||
|
||||
sprintf(temp,"%s%s",PROTO,servHostName);
|
||||
|
||||
strfindr(clntRequest, clntRequest, temp, "");
|
||||
free(temp);
|
||||
//printf("clntRequest:\n%s\n", clntRequest);
|
||||
|
||||
} //5
|
||||
|
||||
//Get Website Address
|
||||
if (!isLocal)
|
||||
{ //3
|
||||
struct addrinfo hints;
|
||||
|
||||
memset(&hints, 0, sizeof(hints));
|
||||
|
||||
hints.ai_family = PF_INET;
|
||||
hints.ai_socktype = SOCK_STREAM;
|
||||
hints.ai_protocol = IPPROTO_TCP;
|
||||
|
||||
// Get the Host Information
|
||||
if (getaddrinfo(servHostName, "http", &hints, &addressInfo)) {
|
||||
//Error
|
||||
free(clntRequest);
|
||||
free(servHostName);
|
||||
close(clntSocket);
|
||||
continue;
|
||||
}
|
||||
|
||||
} //3
|
||||
|
||||
//Open Socket to the server
|
||||
if ( !isLocal)
|
||||
{ //4
|
||||
servSocket = socket(PF_INET, SOCK_STREAM, IPPROTO_TCP);
|
||||
if (servSocket < 0) {
|
||||
free(clntRequest);
|
||||
free(servHostName);
|
||||
close(clntSocket);
|
||||
freeaddrinfo(addressInfo);
|
||||
continue;
|
||||
}
|
||||
|
||||
//Connect to the server
|
||||
|
||||
if (connect(servSocket,
|
||||
addressInfo->ai_addr, addressInfo->ai_addrlen) < 0)
|
||||
{
|
||||
free(clntRequest);
|
||||
free(servHostName);
|
||||
close(clntSocket);
|
||||
close(servSocket);
|
||||
freeaddrinfo(addressInfo);
|
||||
continue;
|
||||
}
|
||||
} else {
|
||||
servSocket = createTCPSocket("127.0.0.1",1337);
|
||||
}
|
||||
|
||||
//printf("Connected to server\n");
|
||||
|
||||
//Send to server
|
||||
if (send(servSocket, clntRequest, strlen(clntRequest), 0) < 0) {
|
||||
printf("Send error to server\n");
|
||||
goto error;
|
||||
}
|
||||
|
||||
//printf("Sent to server\n");
|
||||
|
||||
if (!strstr(clntRequest, "\r\n\r\n\0")) printf("SHIT!!!\n");
|
||||
|
||||
//get stuff from server and send back to client
|
||||
{ //6
|
||||
char buffer[RECV_SIZE];
|
||||
ssize_t got;
|
||||
|
||||
while( (got = recv(servSocket, buffer, RECV_SIZE, 0) ) > 0 ) {
|
||||
//printf("Sending %d to Client!\n", got);
|
||||
//Send back to client
|
||||
if (send(clntSocket, buffer, got, 0) < 0) {
|
||||
printf("Send Error\n");
|
||||
break;
|
||||
}
|
||||
}
|
||||
//printf("Break!!!\n");
|
||||
}
|
||||
|
||||
error:
|
||||
free(clntRequest);
|
||||
free(servHostName);
|
||||
close(clntSocket);
|
||||
close(servSocket);
|
||||
if (!isLocal) freeaddrinfo(addressInfo);
|
||||
}
|
||||
|
||||
} //0
|
||||
|
||||
void *consumer(void* dumb) {
|
||||
|
||||
#if _LOG_
|
||||
printf("[CONSUMER] Consumer %ld started\n", (long)pthread_self());
|
||||
fflush(stdout);
|
||||
#endif
|
||||
|
||||
for (;;) {
|
||||
//Accept data from the connection
|
||||
char *received;
|
||||
int clntSocket;
|
||||
int servSocket;
|
||||
char *request, *request_ptr ,*pc;
|
||||
struct addrinfo *addressInfo, hints;
|
||||
|
||||
clntSocket = removeSocket(); //This is where it waits!!!
|
||||
|
||||
// acceptClientRequest(clntSocket)
|
||||
uber_recv(clntSocket, BUFFERSIZE, "\0", &received);
|
||||
|
||||
// Figure out what needs to be gotten
|
||||
request = strstr(received, PROTO);
|
||||
|
||||
if (!request) {
|
||||
printf("Bad Request on socket: %d\n",clntSocket);
|
||||
free(received);
|
||||
close(clntSocket);
|
||||
continue;
|
||||
}
|
||||
|
||||
request += strlen(PROTO); // Skip past the PROTO (http://)
|
||||
|
||||
// Get the site we want
|
||||
strtok_r(request, "/", &request_ptr);
|
||||
pc = malloc(strlen(request)+1);
|
||||
strcpy(pc,request);
|
||||
|
||||
memset(&hints, 0, sizeof(hints));
|
||||
|
||||
hints.ai_family = PF_INET;
|
||||
hints.ai_socktype = SOCK_STREAM;
|
||||
hints.ai_protocol = IPPROTO_TCP;
|
||||
|
||||
// Get the Host Information
|
||||
if (getaddrinfo(request, "http", &hints, &addressInfo)) {
|
||||
//Error
|
||||
free(received);
|
||||
free(pc);
|
||||
close(clntSocket);
|
||||
continue;
|
||||
}
|
||||
|
||||
// Setup socket to the server.
|
||||
servSocket = socket(PF_INET, SOCK_STREAM, IPPROTO_TCP);
|
||||
if (servSocket < 0) {
|
||||
free(received);
|
||||
close(clntSocket);
|
||||
freeaddrinfo(addressInfo);
|
||||
free(pc);
|
||||
continue;
|
||||
}
|
||||
|
||||
// Connect to the server
|
||||
if ( connect(servSocket, addressInfo->ai_addr, addressInfo->ai_addrlen) < 0) {
|
||||
free(received);
|
||||
close(clntSocket);
|
||||
freeaddrinfo(addressInfo);
|
||||
free(pc);
|
||||
continue;
|
||||
}
|
||||
|
||||
//Retrieve Stuff From server (client)
|
||||
|
||||
if (*request_ptr == ' ') {
|
||||
*request_ptr = '/';
|
||||
*(request_ptr+1) = '\0';
|
||||
} else {
|
||||
strtok_r(request_ptr, " ", &request);
|
||||
request_ptr--;
|
||||
*request_ptr = '/';
|
||||
}
|
||||
|
||||
{
|
||||
//Create the thing to send
|
||||
char http_request[HTTP_REQUEST_SIZE];
|
||||
int sendlen;
|
||||
|
||||
sprintf(http_request, "GET %s HTTP/1.1\r\nHost: %s\r\n\r\n" , request_ptr, pc);
|
||||
free(pc);
|
||||
sendlen = strlen(http_request) + 1;
|
||||
|
||||
printf("Sending: %s\n Length: %d\n", http_request, sendlen);
|
||||
|
||||
if (send(servSocket, http_request, sendlen, 0) != sendlen) {
|
||||
printf("Send Failed\n");
|
||||
free(received);
|
||||
close(clntSocket);
|
||||
freeaddrinfo(addressInfo);
|
||||
continue;
|
||||
}
|
||||
}
|
||||
|
||||
{
|
||||
char serve_recv[1024];
|
||||
|
||||
while (recv(servSocket, serve_recv, 1024, 0) > 0) {
|
||||
printf("Sending\n");
|
||||
if (send(clntSocket, serve_recv, 1024, 0)) {
|
||||
break;
|
||||
}
|
||||
printf("Sent\n");
|
||||
}
|
||||
}
|
||||
|
||||
printf("Freeing\n");
|
||||
close(clntSocket);
|
||||
close(servSocket);
|
||||
free(received);
|
||||
freeaddrinfo(addressInfo);
|
||||
|
||||
} // End Consumer Loop
|
||||
}
|
||||
|
||||
|
||||
void *producer(void* v_port_number) {
|
||||
int clntSock; /* client sock descriptor. */
|
||||
int servSock; /* Server sock descriptor. */
|
||||
|
||||
unsigned short port_number = *((unsigned short*)v_port_number);
|
||||
|
||||
#ifdef _WIN32
|
||||
DWORD threadID; /* ThreadID from CreateThread().*/
|
||||
WSADATA wsaData; /* Winsock struct. */
|
||||
if (WSAStartup(MAKEWORD(2, 0), &wsaData) != 0)/* Winsock 2.0 DLL. */
|
||||
DieWithError("WSAStartup() failed");
|
||||
#endif
|
||||
|
||||
VPRINTF((" Producer Port %d\n", port_number));
|
||||
|
||||
servSock = ContructTCPSocket(port_number);
|
||||
|
||||
#if _LOG_
|
||||
printf("[PRODUCER] Producer Initialized, listening for connections\n");
|
||||
fflush(stdout);
|
||||
#endif
|
||||
|
||||
for(;;) /* Loop forever */
|
||||
{
|
||||
clntSock = AcceptConnection(servSock);
|
||||
if (clntSock == -1) {
|
||||
continue;
|
||||
}
|
||||
addSocket(clntSock);
|
||||
} // End Producer Loop
|
||||
/* NEVER REACHED... */
|
||||
}
|
||||
|
||||
/*
|
||||
* String Find & Replace
|
||||
*
|
||||
* This function works like it sounds. It finds instances of find
|
||||
* in src and replaces them with replace in dest.
|
||||
*
|
||||
*
|
||||
* Ver 1.1 - Corrected Issue with things like Hello and finding lo
|
||||
*/
|
||||
char *strfindr(char *dest, char *src, char *find, char *replace){
|
||||
/* seems easy enough
|
||||
* Too bad this one takes the second longest to code of
|
||||
* all functions in this assignment, beaten only by strpig
|
||||
*/
|
||||
size_t i=0,x=0,j=0,a=0,b=0,bNotFound=0;
|
||||
char *p, *p2;
|
||||
|
||||
/* Invalid Param Checks */
|
||||
if(dest == NULL || src == NULL || find == NULL || replace == NULL){
|
||||
return NULL;
|
||||
}
|
||||
|
||||
/* Empty String */
|
||||
if(*find == '\0'){
|
||||
while(*(src+i) != '\0'){
|
||||
*(dest+i) = *(src+i);
|
||||
i++;
|
||||
}
|
||||
*(dest+i) = '\0';
|
||||
return dest;
|
||||
}
|
||||
|
||||
/* Important I'm sure */
|
||||
p2 = src;
|
||||
|
||||
do{
|
||||
/* The Cheap method to create an inline function */
|
||||
{
|
||||
/*
|
||||
* This block of code finds where the
|
||||
* the thing to be replaced is located
|
||||
*/
|
||||
for(; *(src+a) != '\0'; a++){
|
||||
if(*(find+b) == '\0'){
|
||||
break;
|
||||
}else if(*(src+a) == *(find+b)){
|
||||
b++;
|
||||
}else{
|
||||
if(*(find) != *(src+a)){
|
||||
b=0;
|
||||
p2=(src+a+1);
|
||||
}else{
|
||||
b=1;
|
||||
p2=(src+a);
|
||||
}
|
||||
|
||||
}
|
||||
}
|
||||
/*
|
||||
* We didn't find it, oh poop
|
||||
*/
|
||||
if(*(src+a) == '\0' && *(find+b) != '\0'){
|
||||
bNotFound = 1;
|
||||
}
|
||||
}
|
||||
|
||||
/* Reset b */
|
||||
b=0;
|
||||
|
||||
/* Not found, we can just copy the strings over */
|
||||
if(bNotFound == 1){
|
||||
while(*(src+i) != '\0'){
|
||||
*(dest+x) = *(src+i);
|
||||
i++;x++;
|
||||
}
|
||||
break;
|
||||
}
|
||||
|
||||
/* Also probably important */
|
||||
p = p2;
|
||||
|
||||
/* Keep going until you hit the thang that needs fixin */
|
||||
while((src+i) != p && *(src+i) != '\0'){
|
||||
*(dest+x) = *(src+i);
|
||||
i++;x++;
|
||||
}
|
||||
|
||||
/*
|
||||
* Time to replace
|
||||
* Note the two counters
|
||||
*/
|
||||
if((src+i) == p){
|
||||
for(j=0;*(find+j) != '\0';j++,i++);
|
||||
|
||||
for(j=0;*(replace+j) != '\0';x++,j++){
|
||||
*(dest+x) = *(replace+j);
|
||||
}
|
||||
}
|
||||
|
||||
}while(p2 != NULL);
|
||||
|
||||
/* String Terminated */
|
||||
*(dest+x) = '\0';
|
||||
|
||||
|
||||
return dest;
|
||||
}
|
||||
|
||||
/*--------------------------------------------------------------------+
|
||||
| itoa() - manage the sign, compute the string equivalent, and call |
|
||||
| memcpy(). |
|
||||
+--------------------------------------------------------------------*/
|
||||
char *itoa(int value)
|
||||
{
|
||||
int count, /* number of characters in string */
|
||||
i, /* loop control variable */
|
||||
sign; /* determine if the value is negative */
|
||||
char *ptr, /* temporary pointer, index into string */
|
||||
*string, /* return value */
|
||||
*temp; /* temporary string array */
|
||||
|
||||
int INTSIZE = sizeof(int);
|
||||
|
||||
count = 0;
|
||||
if ((sign = value) < 0) /* assign value to sign, if negative */
|
||||
{ /* keep track and invert value */
|
||||
value = -value;
|
||||
count++; /* increment count */
|
||||
}
|
||||
|
||||
/* allocate INTSIZE plus 2 bytes (sign and NULL) */
|
||||
temp = (char *) malloc(INTSIZE + 2);
|
||||
if (temp == NULL)
|
||||
{
|
||||
return(NULL);
|
||||
}
|
||||
memset(temp,'\0', INTSIZE + 2);
|
||||
|
||||
string = (char *) malloc(INTSIZE + 2);
|
||||
if (string == NULL)
|
||||
{
|
||||
return(NULL);
|
||||
}
|
||||
memset(string,'\0', INTSIZE + 2);
|
||||
ptr = string; /* set temporary ptr to string */
|
||||
|
||||
/*--------------------------------------------------------------------+
|
||||
| NOTE: This process reverses the order of an integer, ie: |
|
||||
| value = -1234 equates to: char [4321-] |
|
||||
| Reorder the values using for {} loop below |
|
||||
+--------------------------------------------------------------------*/
|
||||
do {
|
||||
*temp++ = value % 10 + '0'; /* obtain modulus and or with '0' */
|
||||
count++; /* increment count, track iterations*/
|
||||
} while (( value /= 10) >0);
|
||||
|
||||
if (sign < 0) /* add '-' when sign is negative */
|
||||
*temp++ = '-';
|
||||
|
||||
*temp-- = '\0'; /* ensure null terminated and point */
|
||||
/* to last char in array */
|
||||
|
||||
/*--------------------------------------------------------------------+
|
||||
| reorder the resulting char *string: |
|
||||
| temp - points to the last char in the temporary array |
|
||||
| ptr - points to the first element in the string array |
|
||||
+--------------------------------------------------------------------*/
|
||||
for (i = 0; i < count; i++, temp--, ptr++)
|
||||
{
|
||||
memcpy(ptr,temp,sizeof(char));
|
||||
}
|
||||
|
||||
return(string);
|
||||
}
|
||||
Reference in New Issue
Block a user