cgi_file_upload

#include <stdlib.h> #include <locale> #include <map> #include <io.h> #include <fcntl.h> #include <string.h> #include <iostream> using namespace std; void OUTPUT(char* content) { printf(content); printf("\n"); } void cgi_file_upload() { //client submit method //GET OR POST char *method = getenv("REQUEST_METHOD"); char *content_type = getenv("CONTENT_TYPE"); if (content_type != NULL) { } if (strcmp(method,"POST") == 0) { //file upload content-type if(content_type !=NULL && string(content_type).find("multipart/form-data") != string::npos) { int contentLen = 0; if( getenv("CONTENT_LENGTH") ) { contentLen = atoi( getenv("CONTENT_LENGTH") ); } if (contentLen == 0) { OUTPUT("<script>"); OUTPUT("alert(\"file is error...\");</script>"); } //default 2MB buffer const int FILEBUFFER = 1024*1024*2; // const int FILEBUFFER = 1024; //total data package int count = contentLen / FILEBUFFER; //byte buffer for write the file bytes int buffer = 0; //is need handle extra package bool append = false; //sign first resolve http head bool resolve_head = true; //occur the 0x0d 0x0a 0x2d bool end_file = false; /////////////////////////////////////////////// FILE *ff = fopen("temp_file","w"); char temp_data[1]; fwrite(temp_data,0,0,ff); fclose(ff); /////////////////////////////////////////////// FILE *e = fopen("temp_file","ab"); //contenteLen < buffer if (count == 0) { buffer = contentLen; count++; } else { buffer = FILEBUFFER; append = true; } for (int i=0;i<count;i++) { char *file_data = new char[buffer]; _setmode(_fileno(stdin),_O_BINARY); fread(file_data,buffer,1,stdin); _setmode(_fileno(stdin),_O_TEXT); char sp[] = {0x0d,0x0a,0x0d,0x0a,'\0'}; char sp1[] = {0x0d,0x0a,0x2d,0x2d,'\0'}; int start_index = 0; int end_index = 0; int abstract_head_file_len= 0; if (resolve_head) { char *index_point = strstr(file_data,sp); start_index = index_point - file_data; //move the point end the {0x0d,0x0a,0x0d,0x0a,'\0'} start_index+=4; resolve_head = false; } abstract_head_file_len = buffer-start_index; // log_<<"abstract_head_file_len:"<<abstract_head_file_len<<"\n"; //search the 0x0d 0x0a index for (int j=0;j<abstract_head_file_len;j++) { bool code1 = false; bool code2 = false; bool code3 = false; bool code4 = false; if (*(file_data+start_index+j) == 0X0D) code1 = true; if ((j+1<abstract_head_file_len) && *(file_data+start_index+j+1)==0X0A) code2 = true; if ((j+2<abstract_head_file_len) && *(file_data+start_index+j+2)==0X2d) code3 = true; if ((j+3<abstract_head_file_len) && *(file_data+start_index+j+3)==0X2d) code4 = true; if (code1 && code2 && code3 && code4) { end_file = true; break; } end_index ++; }//end for2 // log_<<"end_index:"<<end_index<<"\n"; fwrite(file_data+start_index,1,end_index,e); //if occur the end of the file than break the for ... if (end_file) { break; } delete[] file_data; }//end for1 //the last hos not read bytes... if (append && !end_file) { int last_file_len = (contentLen - FILEBUFFER * count); int end_index = 0; char *file_data = new char[last_file_len]; _setmode(_fileno(stdin),_O_BINARY); fread(file_data,last_file_len,1,stdin); _setmode(_fileno(stdin),_O_TEXT); //search the 0x0d 0x0a index for (int i=0;i<buffer;i++) { bool code1 = false; bool code2 = false; bool code3 = false; bool code4 = false; if (*(file_data+i) == 0X0D) code1 = true; if ((i+1<last_file_len) && *(file_data+i+1)==0X0A) code2 = true; if ((i+2<last_file_len) && *(file_data+i+2)==0X2d) code3 = true; if ((i+3<last_file_len) && *(file_data+i+3)==0X2d) code4 = true; if (code1 && code2 && code3 && code4) { end_file = true; break; } end_index ++; } fwrite(file_data,1,end_index,e); delete[] file_data; } fclose(e); printf("<script>alert('file update success...');</script>"); printf("<script>window.history.back();</script>"); } } } int main(int argc,char *argv[]) { cgi_file_upload(); return 0; }