gfwleak/tango-tfe
Archived
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

TSG-23793 Fixed the issue where comments in the HTML page containing <head> or <body> tags would cause the inserted script to fail

Browse Source
This commit is contained in:
fengweihao
2024-11-20 10:03:12 +08:00
parent bbf084acad
commit ff436e3f26
4 changed files with 108 additions and 4 deletions
Download Patch File Download Diff File Expand all files Collapse all files

32
plugin/business/tsg-http/test/test_pattern_replace.cpp
View File

@@ -7,7 +7,6 @@
#include <stdio.h>
#include <gtest/gtest.h>
TEST(PatternReplace, Grouping1)
{
const char* find="(?<name1>John)|(?<name2>李梅梅)|(?<name3>Jake)";
@@ -305,9 +304,38 @@ TEST(PatternInsert, BeforeBody)
free(output);
}
TEST(PatternInsert, HtmlCommentsContain)
{
char input[]="<!doctype html><!--<html><head><title></title><body></body></head></html>--><html><head><title></title><body></body></head></html>";
const char* custom = "alert(\"Insert\");";
char *output=NULL;
size_t output_sz=0, input_sz = strlen(input) - 5;
output_sz = simple_insert(input, input_sz, "before-page-load", custom, "js", &output);
printf("output =%s\n", output);
EXPECT_TRUE(output_sz>0);
EXPECT_TRUE(NULL!=strstr(output, custom));
free(output);
}
TEST(PatternInsert, HtmlNoCommentsContain)
{
char input[]="<!doctype html><!--<html><title></title></html>--><html><head></head><!--<html><body></body></html>--><title></title><body></body></html>";
const char* custom = "alert(\"Insert\");";
char *output=NULL;
size_t output_sz=0, input_sz = strlen(input);
output_sz = simple_insert(input, input_sz, "before-page-load", custom, "js", &output);
printf("output =%s\n", output);
EXPECT_TRUE(output_sz>0);
EXPECT_TRUE(NULL!=strstr(output, custom));
free(output);
}
int main(int argc, char ** argv)
{
::testing::InitGoogleTest(&argc, argv);
return RUN_ALL_TESTS();
}