Change ONEOF to output a valid expression (for 500 fixing algo)

Marven11 · Oct 27, 2023 · e54b1c6 · e54b1c6
1 parent 3458c93
commit e54b1c6
Showing 1 changed file with 4 additions and 4 deletions.
diff --git a/fenjing/payload_gen.py b/fenjing/payload_gen.py
@@ -1205,12 +1205,12 @@ def gen_string_twostringconcat(context: dict, value: str):
     if len(value) <= 2 or len(value) > 20:
         return [(UNSATISFIED,)]
     return [
-        (LITERAL, "'"),  # test quotes first
+        # (LITERAL, "'"),  # ONEOF should output a valid expression
         (
             ONEOF,
             *[
                 [
-                    (LITERAL, "{}'".format(value[:i].replace("'", "\\'"))),
+                    (LITERAL, "'{}'".format(value[:i].replace("'", "\\'"))),
                     (LITERAL, "'{}'".format(value[i:].replace("'", "\\'"))),
                 ]
                 for i in range(1, len(value) - 1)
@@ -1224,12 +1224,12 @@ def gen_string_twostringconcat2(context: dict, value: str):
     if len(value) <= 2 or len(value) > 20:
         return [(UNSATISFIED,)]
     return [
-        (LITERAL, '"'),  # test quotes first
+        (LITERAL, '"'),  # ONEOF should output a valid expression
         (
             ONEOF,
             *[
                 [
-                    (LITERAL, '{}"'.format(value[:i].replace('"', '\\"'))),
+                    (LITERAL, '"{}"'.format(value[:i].replace('"', '\\"'))),
                     (LITERAL, '"{}"'.format(value[i:].replace('"', '\\"'))),
                 ]
                 for i in range(1, len(value) - 1)