mirror of
https://github.com/dashr9230/SA-MP.git
synced 2024-12-22 14:37:29 +08:00
[saco] Implement CHttpClient
This commit is contained in:
RD42
parent
a9fc395182
commit
d8a7bafd22
@ -1,75 +1,400 @@
|
||||
|
||||
#include <windows.h>
|
||||
#include <stdio.h>
|
||||
#include <string.h>
|
||||
|
||||
#ifdef WIN32
|
||||
# include <windows.h>
|
||||
#else
|
||||
# include <errno.h>
|
||||
# include <sys/types.h>
|
||||
# include <sys/socket.h>
|
||||
# include <sys/wait.h>
|
||||
# include <netinet/in.h>
|
||||
# include <netdb.h>
|
||||
#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;
|
||||
}
|
||||
|
||||
char CHttpClient::Connect(char *name, u_short hostshort)
|
||||
// Build the HTTP Header
|
||||
switch(m_Request.rtype)
|
||||
{
|
||||
// TODO: CHttpClient::Connect 100029A0
|
||||
return 0;
|
||||
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;
|
||||
}
|
||||
|
||||
char CHttpClient::HandleEntity()
|
||||
{
|
||||
// TODO: CHttpClient::HandleEntity() 10002680
|
||||
return 0;
|
||||
if(!Send(request_head)) {
|
||||
free(request_head);
|
||||
return;
|
||||
}
|
||||
|
||||
int CHttpClient::Recv(char *buf, int len)
|
||||
{
|
||||
// TODO: CHttpClient::Recv 10002490
|
||||
return 0;
|
||||
free(request_head);
|
||||
|
||||
HandleEntity();
|
||||
}
|
||||
|
||||
char CHttpClient::Send(char *buf)
|
||||
//----------------------------------------------------
|
||||
|
||||
#define RECV_BUFFER_SIZE 4096
|
||||
|
||||
void CHttpClient::HandleEntity()
|
||||
{
|
||||
// TODO: CHttpClient::Send 10002450
|
||||
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);
|
||||
}
|
||||
|
||||
int CHttpClient::CloseConnection()
|
||||
/* find the content-length if available */
|
||||
if((pcontent_buf=Util_stristr(header,"CONTENT-LENGTH:"))!=NULL)
|
||||
{
|
||||
// TODO: CHttpClient::CloseConnection() 10002440
|
||||
return 0;
|
||||
has_content_len=true;
|
||||
|
||||
pcontent_buf+=16;
|
||||
while(*pcontent_buf!='\n' && *pcontent_buf)
|
||||
{
|
||||
*pcontent_buf++;
|
||||
content_len++;
|
||||
}
|
||||
|
||||
char CHttpClient::GetHeaderValue(char *a2, char *a3, signed int a4)
|
||||
{
|
||||
// TODO: CHttpClient::GetHeaderValue 100023B0
|
||||
return 0;
|
||||
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;
|
||||
}
|
||||
}
|
||||
}
|
||||
|
||||
//----------------------------------------------------
|
||||
|
||||
@ -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);
|
||||
};
|
||||
|
||||
//----------------------------------------------------
|
||||
|
||||
Loading…
Reference in New Issue
Block a user