void handle_request_socket(http_conf *g , struct epoll_event *evfd ) { epoll_extra_data_t * epoll_data = (epoll_extra_data_t *)evfd->data.ptr; http_connect_t *con = (http_connect_t *) epoll_data->ptr; if(con->in == NULL) { //accept_handler(g, con, evfd); epoll_edit_fd(g->epfd, evfd, EPOLL_W); return ; //epoll_del_fd(g->epfd, evfd); } while(con->next_handle != NULL) { int ret = con->next_handle(con); if(ret == DONE) { if(con->in->execute_file != NULL && con->in->execute_file->len > 0) { char *shPath = (char *)palloc(con->p, sizeof(char)*con->in->execute_file->len +1); strncpy(shPath, con->in->execute_file->ptr, con->in->execute_file->len); ds_log(con, " [send execute sh command:]", LOG_LEVEL_DEFAULT); send_execute_sh_cmd(con, g); } con->next_handle = NULL; epoll_del_fd(g->epfd, evfd); close(con->fd); ds_log(con, " [END] ", LOG_LEVEL_DEFAULT); pool_destroy(con->p); //handle_request_destory(g, evfd); }else if(ret == CONTINUE) { break; } } }
int open_write_file(http_connect_t *con) { request * in; response *out; buffer *uri, *tmp, *dir; char *ptr ; FILE * fp; pool_t *p = con->p; int count = 0; in = con->in; out = con->out; uri = buffer_init(p); uri->ptr = in->uri->ptr; uri->used = uri->size = in->uri->len; dir = buffer_create_size(con->p, in->uri->len); memcpy(dir->ptr, uri->ptr, uri->used); ptr = dir->ptr + uri->used; dir->used = uri->used; while(ptr >= dir->ptr) { if(*ptr == '/') { *ptr = '\0'; break; } ptr--; dir->used--; } uri->ptr[uri->used] = '\0'; if(dir->used) { int ret = _mkdir(con, dir->ptr, p); if(ret == -1) { return 0; } } ptr = (char *) palloc(p, sizeof(char)*2048); fp = fopen(uri->ptr, "w"); if(fp) { make_fd_non_blocking(fileno(fp)); con->write_file.fp = fp; con->write_file.len = 0; con->next_handle = write_file_content; } else { int logStrLen = strlen(dir) + 30; char * logStr = (char *) palloc(p, logStrLen); snprintf(logStr, logStrLen, "[access %s Permission denied]", dir ); ds_log(con, logStr, LOG_LEVEL_ERROR); con->next_handle = send_put_forbidden_result; } return 0; }
void destory_connect(http_conf *g, http_connect_t *con, struct epoll_event * evfd) { if(con->in != NULL) { close(con->fd); ds_log(con, " [END] ", LOG_LEVEL_DEFAULT); epoll_edit_fd(g->epfd, evfd, EPOLL_W); pool_destroy(con->p); } con->next_handle = NULL; epoll_del_fd(g->epfd, evfd); }
void handle_request_destory(http_conf *g, struct epoll_event *evfd ) { http_connect_t * con; epoll_data_t * epoll_data = (epoll_data_t *)evfd->data.ptr; con = (http_connect_t *) epoll_data->ptr; if(con->in != NULL) { ds_log(con, " [END] ", LOG_LEVEL_DEFAULT); epoll_edit_fd(g->epfd, evfd, EPOLL_W); pool_destroy(con->p); if(con->fd != 0 ) { close(con->fd); } } con->next_handle = NULL; epoll_del_fd(g->epfd, evfd); }
int accept_handler(http_connect_t *con) { //con->in = (request *)request_init(con->p); //con->out = (response *)response_init(con->p); read_header(con); parse_header(con); ds_log(con, " [ACCEPT] ", LOG_LEVEL_DEFAULT); if(con->in->http_method == _PUT) { //open_write_file(con); con->next_handle = open_write_file; } else { con->next_handle = NULL; } return 0; }
int start_accept(http_conf *g) { int count; int confd ; struct epoll_event ev[MAX_EVENT]; struct epoll_event *evfd ; epoll_extra_data_t *epoll_data; struct sockaddr addr; struct sockaddr_in addrIn; socklen_t addLen = sizeof(struct sockaddr ); start_web_server(g); printf("--------------- start server\n--------------"); while(1){ count = epoll_wait(g->epfd, ev, MAX_EVENT, -1); evfd = ev; while( (confd = accept(g->fd, &addr, &addLen)) && confd > 0) { pool_t *p = (pool_t *)pool_create(); http_connect_t * con; epoll_extra_data_t *data_ptr; addrIn = *((struct sockaddr_in *) &addr); data_ptr = (epoll_extra_data_t *)palloc(p, sizeof(epoll_extra_data_t)); con = (http_connect_t *) palloc(p, sizeof(http_connect_t));//换成初始化函数, con->p= p; con->fd = confd; con->in = (request *)request_init(p); con->out = (response *)response_init(p); char *ip = NULL; if(addrIn.sin_addr.s_addr) { ip = inet_ntoa(addrIn.sin_addr); } if(ip) { con->in->clientIp = (string *) string_init_from_str(p, ip, strlen(ip)); } con->next_handle = accept_handler; data_ptr->type = SOCKFD; data_ptr->ptr = (void *) con; epoll_add_fd(g->epfd, confd, EPOLL_R, (void *)data_ptr);//对epoll data结构指向的结构体重新封装,分websit //data struct , connect data struct , file data struct , } while((evfd->events & EPOLLIN)){ if((evfd->events & EPOLLIN)) { http_connect_t * con; epoll_data = (epoll_data_t *)evfd->data.ptr; con = (http_connect_t *) epoll_data->ptr; switch(epoll_data->type) { case SOCKFD: if(con->in == NULL) { //accept_handler(g, con, evfd); epoll_edit_fd(g->epfd, evfd, EPOLL_W); //epoll_del_fd(g->epfd, evfd); } while(con->next_handle != NULL) { int ret = con->next_handle(con); if(ret == -1) { con->next_handle = NULL; epoll_del_fd(g->epfd, evfd); close(con->fd); ds_log(con, " [END] ", LOG_LEVEL_DEFAULT); pool_destroy(con->p); }else if(ret == 1) { break; } /*if(con->next_handle(con) == -1) { epoll_del_fd(g->epfd, evfd); close(con->fd); pool_destroy(con->p); }*/ } break; case CGIFD: { epoll_cgi_t *cgi = (epoll_cgi_t *)epoll_data->ptr; break; } } } else if(evfd->events & EPOLLOUT) { } evfd++; } } }