Ошибки Malloc при использовании потоков POSIX в OSX
В настоящее время я заканчиваю последний год обучения для моего бакалавра в области компьютерных наук, один из которых по параллелизму с использованием потоков POSIX. Назначение - это очень простой http-сервер, который обрабатывает http-запросы в отдельных потоках и возвращает простые html-файлы. У меня были некоторые ошибки при прерывании сигнала при освобождении и обнулении буферов символов в конце функций потока. Я работаю в X-Code, и когда я включаю Guard Malloc в схеме, она работает просто отлично. Также, если я удаляю вызовы free() и обнуление, программа тоже работает нормально. Последнее, что я предполагаю, потребляет память, использованную в предыдущих потоках, и не освобождает ее для других процессов, и, следовательно, не является хорошим решением, верно? Я не понимаю Guard Malloc, но так как я передам сырой файл.cpp для компиляции на Linux-машине Лектора, поддержка не будет скомпилирована.
Ниже приведен полный код. Любые комментарии, критика стиля или жемчужины мудрости любого рода, которые вы готовы бросить мне, очень ценятся.
Исходные данные, приводящие к сбою, - это домашняя страница.html с этого случайного веб-сайта - http://www.budgie-info.com/
Спасибо заранее.
#include <string.h>
#include <stdio.h>
#include <stdlib.h>
#include <string.h>
#include <unistd.h>
#include <errno.h>
#include <sys/types.h>
#include <sys/socket.h>
#include <netinet/in.h>
#include <pthread.h>
// Define the port number to identify this process
#define MYPORT 3490
typedef struct socket_fd {
unsigned fd;
} sock_fd;
void *request(void *fd);
int main() {
int s;
//unsigned fd;
struct sockaddr_in my_addr;
pthread_t t;
int retval;
sock_fd *s_fd;
// Construct address information
my_addr.sin_family = AF_INET;
my_addr.sin_port = htons(MYPORT);
my_addr.sin_addr.s_addr = INADDR_ANY;
memset(my_addr.sin_zero, '\0', sizeof(my_addr.sin_zero) );
// Create a socket and bind it the port MYPORT
s=socket(PF_INET,SOCK_STREAM, 0);
int yes = 1;
setsockopt(s, SOL_SOCKET, SO_REUSEADDR, &yes, sizeof(int));
retval = bind(s, (struct sockaddr *)&my_addr, sizeof(my_addr));
if( retval != 0) {
fputs ("bind() failed.",stderr);
exit (7);
}
printf("%d\n", my_addr.sin_port);
retval = listen(s,10);
if( retval != 0 ) { // Allow up to 10 incoming connections
fputs ("listen() failed.",stderr);
exit (9);
}
while(1) {
s_fd = (sock_fd*)malloc(sizeof(sock_fd));
s_fd->fd=accept(s,NULL,NULL); // wait for a request
retval = pthread_create( &t, NULL, request, s_fd);
free(s_fd);
s_fd = NULL;
if(retval != 0) {
printf("pthread_create() failed. error %d\n", retval);
}
}
}
void *request(void *s_fd) {
int retval;
char data[65536];
char header[]="HTTP/1.1 200 OK\r\nContent-Type: text/html\r\n\r\n";
char header_gif[]="HTTP/1.1 200 OK\r\nContent-Type: image/gif\r\n\r\n";
char header_jpeg[]="HTTP/1.1 200 OK\r\nContent-Type: image/jpeg\r\n\r\n";
char header_png[]="HTTP/1.1 200 OK\r\nContent-Type: image/png\r\n\r\n";
int header_len;
char *send_buffer;
char *temp_buffer;
char filename[256];
char filename_parsed[256];
FILE *f;
sock_fd *fd_t = (sock_fd*)s_fd;
unsigned fd = fd_t->fd;
int cnt = 0;
do {
retval=(int)recv(fd,data,65536,0); // recieve the request using fd
++cnt;
} while(retval == 65536);
if(retval < 0) {
fputs ("recv() failed.\n",stderr);
return NULL;
}
data[retval]='\0'; // NUL terminate it
retval = sscanf((char *)data,"GET /%s ",filename); // get the name of the file
if(retval != 1) {
fputs ("sscanf() failed.\n",stderr);
return NULL;
}
if(strlen(filename) > 256) {
fputs ("Filename overflow.\n",stderr);
return NULL;
}
// parse uml spaces out of filenames
int j = 0;
for(int i = 0; filename[i]!='\0'; ++i) {
if(filename[i] == '%') {
filename_parsed[j] = ' ';
i+=2;
} else {
filename_parsed[j] = filename[i];
}
++j;
}
filename_parsed[j] = '\0';
//print received header
printf("Receiving:\n %s\n", data);
//print requested filename
printf("\n\n-------------filename = %s|\n\n", filename_parsed);
if( (f=fopen(filename_parsed,"rb")) == NULL ) { // open the file (might be binary)
fputs ("fopen() failed.\n",stderr);
return NULL;
}
// obtain file size:
size_t lSize;
fseek (f , 0 , SEEK_END);
lSize = ftell (f);
rewind (f);
// pre calculate length of filename
int len = (int)strlen(filename_parsed);
// identify appropriate header, alocate required memory and copy the header into the buffer
if(0 == strcmp(filename_parsed + len - 4, ".gif")) {
header_len = (int)strlen(header_gif);
printf("\n\n\n\n\nG I F\n\n\n\n");
send_buffer = (char*) malloc ( sizeof(char)*((lSize+(size_t)header_len)+14) ); // WARNING: hardcoded margin for header size differences
memcpy(send_buffer, header_gif, header_len);
} else if(0 == strcmp(filename_parsed + len - 5, ".jpeg")) {
printf("\n\n\n\n\nJ P E G\n\n\n\n");
header_len = (int)strlen(header_jpeg);
send_buffer = (char*) malloc ( sizeof(char)*((lSize+(size_t)header_len)+14) ); // WARNING: hardcoded margin for header size differences
memcpy(send_buffer, header_jpeg, header_len);
} else if(0 == strcmp(filename_parsed + len - 4, ".png")) {
header_len = (int)strlen(header_png);
printf("\n\n\n\n\nP N G \n\n\n\n");
send_buffer = (char*) malloc ( sizeof(char)*((lSize+(size_t)header_len)+14) ); // WARNING: hardcoded margin for header size differences
memcpy(send_buffer, header_png, header_len);
} else {
header_len = (int)strlen(header);
send_buffer = (char*) malloc ( sizeof(char)*((lSize+(size_t)header_len)+14) ); // WARNING: hardcoded margin for header size differences
memcpy(send_buffer, header, header_len);
}
// allocate memory to contain the whole file:
temp_buffer = (char*) malloc (sizeof(char)*(lSize+10)); // WARNING: hardcoded margin for header size differences
if (temp_buffer == NULL) {
fputs ("malloc() failed.\n",stderr);
return NULL;
}
// copy the file into the buffer:
retval = (int)fread (temp_buffer,1,lSize,f);
if (retval != lSize) {
fputs ("fread() failed.\n",stderr);
return NULL;
}
memcpy(send_buffer + header_len, temp_buffer, retval);
//print packet being sent
printf("Sending:\n%s\n", send_buffer);
memcpy(send_buffer + retval, "\r\n", 4);
// send packet
retval = (int)send(fd,send_buffer,retval,0);
if(retval < 0) {
fputs ("send() failed.\n",stderr);
return NULL;
}
free(temp_buffer); // The section of memory management causes SIGABRT errors everytime i run, commenting them out made it run smoothly.
temp_buffer = NULL; // Is the closing of this function tidying up for me or is it leaving a memory leak trail?
free(send_buffer); // Is it the multi threading that is making memory management buggy?
send_buffer = NULL;
fclose(f);
f = NULL;
close(fd); // close the socket
return NULL;
}
2 ответа
Есть много проблем с этим кодом. Трудно сказать, почему происходит сбой этих файлов, учитывая количество потенциальных проблем (вы не предоставили пример ввода, который приводит к сбою программы)
Чтобы эта проблема была пригодна для использования, необходимо установить SO_REUSEADDR для сокета, в противном случае, если произойдет сбой (подсказка, подсказка), вы не сможете сразу же перезапустить ее и получать соединения.
вы не проверяете возвращаемое значение bind(), accept(), pthread_create()
вам нужно вызывать listen() только один раз, а не в цикле
безопаснее разыгрывать fd задолго до того, как разыгрывать его в void *, поскольку это то, что вы делаете в прослушивании. Более чистым способом будет задействовать структуру в куче с fd в нем
Не определяйте размер массивов, которые вы устанавливаете во время компиляции, пусть это сделает компилятор. Это безопаснее:
char header[]="HTTP/1.1 200 OK\r\nContent-Type: text/html\r\n\r\n";, Тогда вам не нужно звонить по нему. Вы можете просто использоватьsizeof(header) - 1,Вы не проверяете возвращаемое значение recv(). Вы должны убедиться, что он не потерпел неудачу (особенно если вы используете его как индекс в массиве)
Вы не гарантируете, что имя файла не переполнено. Вы читаете до 65 КБ, но не уверены, что имя файла достаточно велико (то есть сбой вашей программы тривиален и это уязвимость системы безопасности)
Вы не читаете возвращаемое значение sscanf. Нет никаких гарантий, что имя файла содержит данные
Вы не прекращаете работу filename_parsed (я полагаю, что это основная причина сбоя вашей программы)
Вы можете вызвать strlen (filename_parsed) только один раз и вместо того, чтобы вызывать его миллион раз
Вместо ваших запутанных проверок символов в конце, почему бы не использовать strcmp (после того, как filename_parsed правильно завершится, конечно):
0 == strcmp(filename_parsed + len - 5, ".jpeg")и т.п.- Проверьте возвращаемое значение отправить
Вы можете проверить одну вещь: содержимое, которое читается из fread(), не обязательно является строкой - может не заканчиваться NULL вообще. Таким образом, использование strcat на этом не является правильным способом. Попробуйте изменить его на memcpy() или что-то в этом роде.