1 | // LocalProxy.cpp : ÄÜ¼Ö ÀÀ¿ë ÇÁ·Î±×·¥¿¡ ´ëÇÑ ÁøÀÔÁ¡À» Á¤ÀÇÇÕ´Ï´Ù.
|
---|
2 | //
|
---|
3 |
|
---|
4 | #ifdef _WIN32
|
---|
5 | # define _CRT_SECURE_NO_WARNINGS
|
---|
6 | # include <process.h>
|
---|
7 | #else
|
---|
8 | # include <errno.h>
|
---|
9 | #endif
|
---|
10 |
|
---|
11 | #include "CRXSocket.h"
|
---|
12 | #include "CRXHttpRequest.h"
|
---|
13 | #include "CRXHttpResponse.h"
|
---|
14 |
|
---|
15 | #include <iostream>
|
---|
16 | #include <iomanip>
|
---|
17 | #include <sstream>
|
---|
18 |
|
---|
19 | #include <stdio.h>
|
---|
20 | #include <string.h>
|
---|
21 |
|
---|
22 | using namespace std;
|
---|
23 |
|
---|
24 | /*-----------------------------------------------------------------*/
|
---|
25 | int CRXCreateThread (int (* aThreadFunc) (void *), void * aThreadArg);
|
---|
26 | int CRXProxyThread (void * aThreadArg);
|
---|
27 |
|
---|
28 | /* error */
|
---|
29 | #ifdef _WIN32
|
---|
30 | # define CRX_ERROR() GetLastError ()
|
---|
31 | # ifndef __func__
|
---|
32 | # define __func__ __FUNCTION__
|
---|
33 | # endif
|
---|
34 | #else
|
---|
35 | # define CRX_ERROR() errno
|
---|
36 | #endif
|
---|
37 |
|
---|
38 | #define CRX_PRINT_ERROR(__code, __message) \
|
---|
39 | do { \
|
---|
40 | char __error[1024] = {0x00, }; \
|
---|
41 | ostringstream __stream; \
|
---|
42 | __stream << __message; \
|
---|
43 | string __string = __stream.str (); \
|
---|
44 | sprintf (__error, "[%s][%d] (%d) %s\n", __func__, __LINE__, __code, __string.c_str ()); \
|
---|
45 | cout << __error; \
|
---|
46 | } while (0)
|
---|
47 |
|
---|
48 | typedef struct __crx_proxy_thread_argument__
|
---|
49 | {
|
---|
50 | int mSocket;
|
---|
51 | } CRXProxyThreadArgs;
|
---|
52 | /*----------------------------------------------------------------*/
|
---|
53 |
|
---|
54 | int main (int argc, char* argv[])
|
---|
55 | {
|
---|
56 | int aResult = 0;
|
---|
57 | const unsigned short aPort = 8080;
|
---|
58 |
|
---|
59 | CRXSocket aSocket;
|
---|
60 | CRXProxyThreadArgs * aThreadArg;
|
---|
61 |
|
---|
62 | /*----------------------------------------------------------------*/
|
---|
63 | /*----------------------------------------------------------------
|
---|
64 | * Initialize
|
---|
65 | *----------------------------------------------------------------*/
|
---|
66 | aResult = aSocket.CreateServer (aPort);
|
---|
67 | if (aResult < 0)
|
---|
68 | {
|
---|
69 | CRX_PRINT_ERROR (aResult, "failed to create server");
|
---|
70 | return aResult;
|
---|
71 | }
|
---|
72 |
|
---|
73 | for (;;)
|
---|
74 | {
|
---|
75 | aResult = aSocket.Accept ();
|
---|
76 | if (aResult < 0)
|
---|
77 | {
|
---|
78 | CRX_PRINT_ERROR (aResult, "failed to accept");
|
---|
79 | break;
|
---|
80 | }
|
---|
81 |
|
---|
82 | if ((aThreadArg = new(std::nothrow) CRXProxyThreadArgs) == NULL)
|
---|
83 | {
|
---|
84 | aResult = -1;
|
---|
85 | break;
|
---|
86 | }
|
---|
87 |
|
---|
88 | memset (aThreadArg, 0x0, sizeof (CRXProxyThreadArgs));
|
---|
89 | aThreadArg->mSocket = aResult;
|
---|
90 |
|
---|
91 | aResult = CRXCreateThread (CRXProxyThread, aThreadArg);
|
---|
92 | if (aResult < 0)
|
---|
93 | {
|
---|
94 | aResult = CRX_ERROR ();
|
---|
95 | CRX_PRINT_ERROR (aResult, "failed to create thread");
|
---|
96 | break;
|
---|
97 | }
|
---|
98 | }
|
---|
99 |
|
---|
100 | /*----------------------------------------------------------------*/
|
---|
101 | aSocket.Close ();
|
---|
102 |
|
---|
103 | return aResult;
|
---|
104 | }
|
---|
105 |
|
---|
106 | int
|
---|
107 | CRXCreateThread (int (* aThreadFunc) (void *), void * aThreadArg)
|
---|
108 | {
|
---|
109 | #ifdef _WIN32
|
---|
110 | return _beginthread ((void (*) (void *))aThreadFunc, 0, aThreadArg);
|
---|
111 | #else
|
---|
112 | pthread_t aThreadID = 0;
|
---|
113 |
|
---|
114 | return pthread_create (&aThreadID, NULL, (void * (*) (void *))aThreadFunc, aThreadArg);
|
---|
115 | #endif
|
---|
116 | }
|
---|
117 |
|
---|
118 | int
|
---|
119 | CRXProxyThread (void * aThreadArg)
|
---|
120 | {
|
---|
121 | int aResult = 0;
|
---|
122 | int aReceivedSize = 0;
|
---|
123 |
|
---|
124 | const unsigned int aBufferSize = 1024 * 64;
|
---|
125 | char aBuffer[aBufferSize] = {0x00, };
|
---|
126 |
|
---|
127 | CRXHttpRequest aHttpRequest;
|
---|
128 | CRXHttpResponse aHttpResponse;
|
---|
129 |
|
---|
130 | CRXSocket aWebServer;
|
---|
131 | CRXSocket aWebBrowser = ((CRXProxyThreadArgs *)aThreadArg)->mSocket;
|
---|
132 |
|
---|
133 | bool aIsMoreRequest = false;
|
---|
134 | /*----------------------------------------------------------------*/
|
---|
135 | delete (CRXProxyThreadArgs *)aThreadArg;
|
---|
136 |
|
---|
137 | do
|
---|
138 | {
|
---|
139 | /* BROWSER --- request ---> [PROXY] SERVER */
|
---|
140 | aReceivedSize = aWebBrowser.Receive (aBuffer, aBufferSize);
|
---|
141 | if (aReceivedSize < 0)
|
---|
142 | {
|
---|
143 | CRX_PRINT_ERROR (aReceivedSize, "failed to receive from client (" << CRX_ERROR () << ")");
|
---|
144 | return aReceivedSize;
|
---|
145 | }
|
---|
146 |
|
---|
147 | /* BROWSER ?--- waiting --- [PROXY] --- connect ---> SERVER */
|
---|
148 | if (!aWebServer)
|
---|
149 | {
|
---|
150 | /* parse http request */
|
---|
151 | aHttpRequest = aBuffer;
|
---|
152 |
|
---|
153 | /* connect */
|
---|
154 | aResult = aWebServer.Connect (aHttpRequest.GetHost (), aHttpRequest.GetPort (), 1);
|
---|
155 | if (aResult < 0)
|
---|
156 | {
|
---|
157 | CRX_PRINT_ERROR (aResult, "failed to connect to server ("
|
---|
158 | << CRX_ERROR ()
|
---|
159 | << ") <"
|
---|
160 | << aHttpRequest.GetHost ()
|
---|
161 | << ":"
|
---|
162 | << aHttpRequest.GetPort ()
|
---|
163 | << "><"
|
---|
164 | << aHttpRequest.GetURL ()
|
---|
165 | << ">");
|
---|
166 | return aResult;
|
---|
167 | }
|
---|
168 | }
|
---|
169 |
|
---|
170 | /* BROWSER ?--- waiting --- [PROXY] --- request ---> SERVER */
|
---|
171 | aResult = aWebServer.Send (aBuffer, aReceivedSize);
|
---|
172 | if (aResult != aReceivedSize)
|
---|
173 | {
|
---|
174 | CRX_PRINT_ERROR (aResult, "failed to send to server <"
|
---|
175 | << aHttpRequest.GetHost ()
|
---|
176 | << ":"
|
---|
177 | << aHttpRequest.GetPort ()
|
---|
178 | << CRX_ERROR ()
|
---|
179 | << ")");
|
---|
180 | return aResult;
|
---|
181 | }
|
---|
182 |
|
---|
183 | /* BROWSER <--- response--- [PROXY] <--- response--- SERVER */
|
---|
184 | for (;;)
|
---|
185 | {
|
---|
186 | memset (aBuffer, 0x00, aBufferSize);
|
---|
187 |
|
---|
188 | aResult = aWebServer.Receive (aBuffer, aBufferSize);
|
---|
189 | if (aResult < 0)
|
---|
190 | {
|
---|
191 | CRX_PRINT_ERROR (aResult, "failed to receive ("
|
---|
192 | << CRX_ERROR ()
|
---|
193 | << ") <"
|
---|
194 | << aHttpRequest.GetURL ()
|
---|
195 | << ">");
|
---|
196 | break;
|
---|
197 | }
|
---|
198 | else if (aResult == 0)
|
---|
199 | {
|
---|
200 | break;
|
---|
201 | }
|
---|
202 | aReceivedSize = aResult;
|
---|
203 |
|
---|
204 | aResult = aWebBrowser.Send (aBuffer, aReceivedSize);
|
---|
205 | if (aResult != aReceivedSize)
|
---|
206 | {
|
---|
207 | CRX_PRINT_ERROR (aResult, "failed to send ("
|
---|
208 | << CRX_ERROR ()
|
---|
209 | << ") <"
|
---|
210 | << aHttpRequest.GetURL ()
|
---|
211 | << ">");
|
---|
212 | break;
|
---|
213 | }
|
---|
214 | }
|
---|
215 | } while (aIsMoreRequest);
|
---|
216 |
|
---|
217 | /*----------------------------------------------------------------*/
|
---|
218 | aWebBrowser.Close ();
|
---|
219 | aWebServer.Close ();
|
---|
220 |
|
---|
221 | return aResult;
|
---|
222 | }
|
---|