Changeset 55 in cheroxy for trunk/src/main.cpp

Timestamp:

12/07/12 10:00:01 (11 years ago)

Author:

cheese

Message:

#1 요청 메시지 변조하는 예제로 main 구성

File:

• trunk/src/main.cpp (modified) (6 diffs)

trunk/src/main.cpp

@@ -9 +9 @@
 #include <iomanip>
 #include <sstream>
+#include <string>
 
 #include <stdio.h>
@@ -28 +29 @@
 #endif
 
+#define PROXY_CONNECTION    "Proxy-Connection"
+#define CONNECTION          "Connection"
+
 THREAD_FUNCTION_RETURN
 THREAD_FUNCTION_CALLING_CONVENTION
@@ -34 +38 @@
 int ThreadPool (CRXSocket   * aProxySocket,
                 const int   aThreadPoolCount);
+
+bool    gViewRequest    = true;
+bool    gViewResponse   = false;
 
 int main (int argc, char* argv[])
@@ -110 +117 @@
     CRXProxy    * aProxy = NULL;
 
-    char        aFilterFileExtension[] = "exe|gif|jpg|png|css|js|ico|swf|";
+    const char  aFilterFileExtension[] = "exe|gif|jpg|png|css|js|ico|swf|";
 
     /*----------------------------------------------------------------*/
@@ -140 +147 @@
                                   aFilterFileExtension);
 
-        aResult = aProxy->Forward ();
+        aResult = aProxy->ReceiveRequest ();
         if (aResult < 0)
         {
@@ -146 +153 @@
         }
 
-        if (!aFilter.CheckRequestFilter (CRX_FILTER_REQUEST_FILE_EXTENSION, aRequest))
+        if (gViewRequest && !aFilter.CheckRequestFilter (CRX_FILTER_REQUEST_FILE_EXTENSION, aRequest))
         {
+            string aRequestHeader = aRequest.GetHeader ();
+            aRequestHeader.replace (aRequest.GetHeader ().find (PROXY_CONNECTION), strlen (PROXY_CONNECTION), CONNECTION);
+            aRequest.SetHeader (aRequestHeader.c_str ());
+
             CRX_DEBUG_TRACE ("== Request:  \n%s\n", aRequest.GetHeader ().c_str ());
+        }
+
+        aResult = aProxy->SendRequest ();
+        if (aResult < 0)
+        {
+            cout << aProxy->GetErrorMessage () << endl;
+        }
+
+        aResult = aProxy->ReceiveResponse ();
+        if (aResult < 0)
+        {
+            cout << aProxy->GetErrorMessage () << endl;
+        }
+
+        if (gViewResponse && !aFilter.CheckRequestFilter (CRX_FILTER_REQUEST_FILE_EXTENSION, aRequest))
+        {
             CRX_DEBUG_TRACE ("== Response: \n%s\n", aResponse.GetHeader ().c_str ());
             CRX_DEBUG_TRACE_BIN (aResponse.GetContentBody (),
                                  aResponse.GetContentLength (),
                                  "== Content-Body: \n");
+        }
+
+        aResult = aProxy->SendResponse ();
+        if (aResult < 0)
+        {
+            cout << aProxy->GetErrorMessage () << endl;
         }