diff --git a/saco/httpclient.cpp b/saco/httpclient.cpp index d64083f..3995a47 100644 --- a/saco/httpclient.cpp +++ b/saco/httpclient.cpp @@ -1,75 +1,400 @@ -#include #include +#include + +#ifdef WIN32 +# include +#else +# include +# include +# include +# include +# include +# include +#endif + #include "httpclient.h" +#include "runutil.h" //---------------------------------------------------- CHttpClient::CHttpClient() { - // TODO: CHttpClient::CHttpClient() 100022E0 + memset(&m_Request,0,sizeof(HTTP_REQUEST)); + memset(&m_Response,0,sizeof(HTTP_RESPONSE)); + m_iError = HTTP_SUCCESS; // Request is successful until otherwise indicated + m_iSocket = (-1); + +// Winsock init +#ifdef WIN32 + WORD wVersionRequested; + WSADATA wsaData; + wVersionRequested = MAKEWORD(2,2); + WSAStartup(wVersionRequested,&wsaData); +#endif } //---------------------------------------------------- CHttpClient::~CHttpClient() { - // TODO: CHttpClient::~CHttpClient() 10002340 + // Deallocate the request data memory + if(m_Request.file) free(m_Request.file); + if(m_Request.host) free(m_Request.host); + if(m_Request.referer) free(m_Request.referer); + + if(m_Request.rtype == HTTP_POST) { + if(m_Request.data) free(m_Request.data); + } + + // Deallocate the response data memory + if(m_Response.header) free(m_Response.header); + if(m_Response.response) free(m_Response.response); + +// Winsock cleanup +#ifdef WIN32 + WSACleanup(); +#endif } //---------------------------------------------------- -int CHttpClient::ProcessURL(int a2, char *a3, char *a4, char *a5) +int CHttpClient::ProcessURL(int iType, char *szURL, char *szPostData, char *szReferer) { - // TODO: CHttpClient::ProcessURL 10002C40 + printf("CHttpClient::ProcessURL %s\n", szURL); // Thanks Kalcor! - printf("CHttpClient::ProcessURL %s\n", a3); // Thanks Kalcor! + InitRequest(iType,szURL,szPostData,szReferer); - return 1; + Process(); + + return m_iError; } -void CHttpClient::InitRequest(int a2, char *a3, char *a4, char *a5) +//---------------------------------------------------- + +bool CHttpClient::GetHeaderValue(char *szHeaderName,char *szReturnBuffer, int iBufSize) { - // TODO: CHttpClient::InitRequest 100024B0 + char *szHeaderStart; + char *szHeaderEnd; + int iLengthSearchHeader = strlen(szHeaderName); + int iCopyLength; + + szHeaderStart = Util_stristr(m_Response.header,szHeaderName); + if(!szHeaderStart) { + return false; + } + szHeaderStart+=iLengthSearchHeader+1; + + szHeaderEnd = strchr(szHeaderStart,'\n'); + if(!szHeaderEnd) { + szHeaderEnd = m_Response.header + strlen(m_Response.header); // (END OF STRING) + } + + iCopyLength = szHeaderEnd - szHeaderStart; + if(iBufSize < iCopyLength) { + return false; + } + + memcpy(szReturnBuffer,szHeaderStart,iCopyLength); + szReturnBuffer[iCopyLength] = '\0'; + return true; } +//---------------------------------------------------- + +bool CHttpClient::Connect(char *szHost, int iPort) +{ + struct sockaddr_in sa; + struct hostent *hp; + + // Hostname translation + if((hp=(struct hostent *)gethostbyname(szHost)) == NULL ) { + m_iError=HTTP_ERROR_BAD_HOST; + return false; + } + + // Prepare a socket + memset(&sa,0,sizeof(sa)); + memcpy(&sa.sin_addr,hp->h_addr,hp->h_length); + sa.sin_family = hp->h_addrtype; + sa.sin_port = htons((unsigned short)iPort); + + if((m_iSocket=socket(AF_INET,SOCK_STREAM,0)) < 0) { + m_iError=HTTP_ERROR_NO_SOCKET; + return false; + } + + // Try to connect + if(connect(m_iSocket,(struct sockaddr *)&sa,sizeof sa) < 0) { + CloseConnection(); + m_iError=HTTP_ERROR_CANT_CONNECT; + return false; + } + + return true; +} + +//---------------------------------------------------- + +void CHttpClient::CloseConnection() +{ +#ifdef WIN32 + closesocket(m_iSocket); +#else + close(m_iSocket); +#endif +} + +//---------------------------------------------------- + +bool CHttpClient::Send(char *szData) +{ + if(send(m_iSocket,szData,strlen(szData),0) < 0) { + m_iError = HTTP_ERROR_CANT_WRITE; + return false; + } + return true; +} + +//---------------------------------------------------- + +int CHttpClient::Recv(char *szBuffer, int iBufferSize) +{ + return recv(m_iSocket,szBuffer,iBufferSize,0); +} + +//---------------------------------------------------- + +void CHttpClient::InitRequest(int iType, char *szURL, char *szPostData, char *szReferer) +{ + char *port; // port string + char *port_char; // position of ':' if any + unsigned int slash_pos; // position of first '/' numeric + char *slash_ptr; + char *szUseURL; // incase we have to cat something to it. + + szUseURL = (char *)malloc(strlen(szURL)+256); + strcpy(szUseURL,szURL); + + m_Request.rtype = iType; + m_Request.referer = (char *)malloc(strlen(szReferer)+1); + strcpy(m_Request.referer,szReferer); + + if(iType==HTTP_POST) { + m_Request.data=(char *)malloc(strlen(szPostData)+1); + strcpy(m_Request.data,szPostData); + } + + // Copy hostname from URL + slash_ptr = strchr(szUseURL,'/'); + if(!slash_ptr) { + strcat(szUseURL,"/"); + slash_ptr = strchr(szUseURL,'/'); + } + + slash_pos=(slash_ptr-szUseURL); + m_Request.host=(char *)malloc(slash_pos+1); + memcpy(m_Request.host,szUseURL,slash_pos); + m_Request.host[slash_pos]='\0'; + + // Copy the rest of the url to the file string. + m_Request.file=(char *)malloc((strlen(szUseURL)-slash_pos)+1); + strcpy(m_Request.file,strchr(szUseURL,'/')); + + // Any special port used in the URL? + if((port_char=strchr(m_Request.host,':'))!=NULL) { + port=(char *)malloc(strlen(port_char)); + strcpy(port,port_char+1); + *port_char='\0'; + m_Request.port = atoi(port); + free(port); + } + else { + m_Request.port = 80; + } + + free(szUseURL); +} + +//---------------------------------------------------- + void CHttpClient::Process() { - // TODO: CHttpClient::Process() 10002A60 + int header_len; + char *request_head; + + if(!Connect(m_Request.host,m_Request.port)) { + return; + } + + // Build the HTTP Header + switch(m_Request.rtype) + { + case HTTP_GET: + header_len = strlen(m_Request.file)+strlen(m_Request.host)+ + (strlen(GET_FORMAT)-8)+strlen(USER_AGENT)+ + strlen(m_Request.referer); + request_head = (char *)malloc(header_len+1); + sprintf(request_head,GET_FORMAT,m_Request.file,USER_AGENT,m_Request.referer,m_Request.host); + break; + + case HTTP_HEAD: + header_len = strlen(m_Request.file)+strlen(m_Request.host)+ + (strlen(HEAD_FORMAT)-8)+strlen(USER_AGENT)+ + strlen(m_Request.referer); + request_head = (char *)malloc(header_len+1); + sprintf(request_head,HEAD_FORMAT,m_Request.file,USER_AGENT,m_Request.referer,m_Request.host); + break; + + case HTTP_POST: + header_len = strlen(m_Request.data)+strlen(m_Request.file)+ + strlen(m_Request.host)+strlen(POST_FORMAT)+ + strlen(USER_AGENT)+strlen(m_Request.referer); + request_head = (char *)malloc(header_len+1); + sprintf(request_head,POST_FORMAT,m_Request.file,USER_AGENT,m_Request.referer,m_Request.host,strlen(m_Request.data),m_Request.data); + break; + } + + if(!Send(request_head)) { + free(request_head); + return; + } + + free(request_head); + + HandleEntity(); } -char CHttpClient::Connect(char *name, u_short hostshort) +//---------------------------------------------------- + +#define RECV_BUFFER_SIZE 4096 + +void CHttpClient::HandleEntity() { - // TODO: CHttpClient::Connect 100029A0 - return 0; + int bytes_total = 0; + int bytes_read = 0; + char *buffer = (char *)malloc(RECV_BUFFER_SIZE); + char *response = (char *)malloc(1); + char *header; + char *head_end; + char *pcontent_buf; + char *content_len_str; + + bool header_got = false; + bool has_content_len = false; + int header_len = 0; + int content_len = 0; + + while((bytes_read=Recv(buffer,RECV_BUFFER_SIZE)) > 0) + { + bytes_total+=bytes_read; + response=(char *)realloc(response,bytes_total+1); + memcpy(response+(bytes_total-bytes_read),buffer,(unsigned int)bytes_read); + + if(!header_got) + { + if((head_end=strstr(response,"\r\n\r\n"))!=NULL + || (head_end=strstr(response,"\n\n"))!=NULL) + { + header_got=true; + + header_len=(head_end-response); + header=(char *)malloc(header_len+1); + memcpy(header,response,header_len); + header[header_len]='\0'; + + if((*(response+header_len))=='\n') /* LFLF */ + { + bytes_total-=(header_len+2); + memmove(response,(response+(header_len+2)),bytes_total); + } + else /* assume CRLFCRLF */ + { + bytes_total-=(header_len+4); + memmove(response,(response+(header_len+4)),bytes_total); + } + + /* find the content-length if available */ + if((pcontent_buf=Util_stristr(header,"CONTENT-LENGTH:"))!=NULL) + { + has_content_len=true; + + pcontent_buf+=16; + while(*pcontent_buf!='\n' && *pcontent_buf) + { + *pcontent_buf++; + content_len++; + } + + content_len_str=(char *)malloc(content_len+1); + pcontent_buf-=content_len; + memcpy(content_len_str,pcontent_buf,content_len); + + if(content_len_str[content_len-1] == '\r') { + content_len_str[content_len-1]='\0'; + } + else { + content_len_str[content_len]='\0'; + } + + content_len=atoi(content_len_str); + free(content_len_str); + + if(content_len > MAX_ENTITY_LENGTH) { + CloseConnection(); + m_iError = HTTP_ERROR_CONTENT_TOO_BIG; + return; + } + } + } + } + + if(header_got && has_content_len) + if(bytes_total>=content_len) break; + } + + CloseConnection(); + + response[bytes_total]='\0'; + free(buffer); + + // check the returned header + if(!header_got || *(DWORD *)header != 0x50545448) { // 'HTTP' + m_iError = HTTP_ERROR_MALFORMED_RESPONSE; + return; + } + + // Now fill in the response code + char response_code_str[4]; + response_code_str[0] = *(header+9); + response_code_str[1] = *(header+10); + response_code_str[2] = *(header+11); + response_code_str[3] = '\0'; + m_Response.response_code = atoi(response_code_str); + + // Copy over the document entity strings and sizes + m_Response.header = header; + m_Response.header_len = header_len; + m_Response.response = response; + m_Response.response_len = bytes_total; + + //printf("Code: %u\n\n%s\n",m_Response.response_code,m_Response.header); + + // Try and determine the document type + m_Response.content_type = CONTENT_TYPE_HTML; // default to html + + char szContentType[256]; + + if(GetHeaderValue("CONTENT-TYPE:",szContentType,256) == true) { + if(strstr(szContentType,"text/html") != NULL) { + m_Response.content_type = CONTENT_TYPE_HTML; + } + else if(strstr(szContentType,"text/plain") != NULL) { + m_Response.content_type = CONTENT_TYPE_TEXT; + } else { + m_Response.content_type = CONTENT_TYPE_UNKNOWN; + } + } } -char CHttpClient::HandleEntity() -{ - // TODO: CHttpClient::HandleEntity() 10002680 - return 0; -} - -int CHttpClient::Recv(char *buf, int len) -{ - // TODO: CHttpClient::Recv 10002490 - return 0; -} - -char CHttpClient::Send(char *buf) -{ - // TODO: CHttpClient::Send 10002450 - return 0; -} - -int CHttpClient::CloseConnection() -{ - // TODO: CHttpClient::CloseConnection() 10002440 - return 0; -} - -char CHttpClient::GetHeaderValue(char *a2, char *a3, signed int a4) -{ - // TODO: CHttpClient::GetHeaderValue 100023B0 - return 0; -} +//---------------------------------------------------- diff --git a/saco/httpclient.h b/saco/httpclient.h index 54319eb..e79d7b7 100644 --- a/saco/httpclient.h +++ b/saco/httpclient.h @@ -1,27 +1,91 @@ +#define HTTP_GET 1 +#define HTTP_POST 2 +#define HTTP_HEAD 3 + +//---------------------------------------------------- + +#define MAX_ENTITY_LENGTH 500000 + +//---------------------------------------------------- + +#define HTTP_SUCCESS 0 +#define HTTP_ERROR_BAD_HOST 1 +#define HTTP_ERROR_NO_SOCKET 2 +#define HTTP_ERROR_CANT_CONNECT 3 +#define HTTP_ERROR_CANT_WRITE 4 +#define HTTP_ERROR_CONTENT_TOO_BIG 5 +#define HTTP_ERROR_MALFORMED_RESPONSE 6 + +//---------------------------------------------------- + +#define CONTENT_TYPE_UNKNOWN 0 +#define CONTENT_TYPE_TEXT 1 +#define CONTENT_TYPE_HTML 2 + +//---------------------------------------------------- + +#define USER_AGENT "SAMP/0.2.0" +#define GET_FORMAT "GET %s HTTP/1.0\r\nAccept: */*\r\nUser-Agent: %s\r\nReferer: http://%s\r\nHost: %s\r\n\r\n" +#define POST_FORMAT "POST %s HTTP/1.0\r\nAccept: */*\r\nUser-Agent: %s\r\nReferer: http://%s\r\nHost: %s\r\nContent-type: application/x-www-form-urlencoded\r\nContent-length: %u\r\n\r\n%s" +#define HEAD_FORMAT "HEAD %s HTTP/1.0\r\nAccept: */*\r\nUser-Agent: %s\r\nReferer: http://%s\r\nHost: %s\r\n\r\n" + +//---------------------------------------------------- + +typedef struct{ + unsigned short port; /* remote port */ + int rtype; /* request type */ + char * host; /* hostname */ + char * file; /* GET/POST request file */ + char * data; /* POST data (if rtype HTTP_POST) */ + char * referer; /* http referer. */ +} HTTP_REQUEST; + +//---------------------------------------------------- + +typedef struct{ + char * header; + char * response; + unsigned int header_len; + unsigned long response_len; + unsigned int response_code; + unsigned int content_type; +} HTTP_RESPONSE; + //---------------------------------------------------- class CHttpClient { private: - int field_0; - char _gap4[22]; - char _gap1A[24]; - int field_32; + + int m_iSocket; + HTTP_REQUEST m_Request; + HTTP_RESPONSE m_Response; + int m_iError; + + bool Connect(char *szHost, int iPort); + void CloseConnection(); + bool Send(char *szData); + int Recv(char *szBuffer, int iBufferSize); + + void InitRequest(int iType, char *szURL, char *szPostData, char *szReferer); + void HandleEntity(); + + void Process(); public: + int ProcessURL(int iType, char *szURL, char *szData, char *szReferer); + + bool GetHeaderValue(char *szHeaderName,char *szReturnBuffer, int iBufSize); + int GetResponseCode() { return m_Response.response_code; }; + int GetContentType() { return m_Response.content_type; }; + char *GetResponseHeaders() { return m_Response.header; }; + char *GetDocument() { return m_Response.response; }; + int GetDocumentLength() { return m_Response.response_len; }; + CHttpClient(); ~CHttpClient(); - int ProcessURL(int a2, char *a3, char *a4, char *a5); - void InitRequest(int a2, char *a3, char *a4, char *a5); - void Process(); - char Connect(char *name, u_short hostshort); - char HandleEntity(); - int Recv(char *buf, int len); - char Send(char *buf); - int CloseConnection(); - char GetHeaderValue(char *a2, char *a3, signed int a4); }; //----------------------------------------------------