Grasscutters/mitmproxy
3
0
Fork 0
mirror of https://github.com/Grasscutters/mitmproxy.git synced 2024-11-22 15:37:45 +00:00
Code Issues Projects Releases Wiki Activity

web: Add websocket filter

Browse Source
This commit is contained in:
Sachin Kelkar 2017-03-06 10:20:11 +05:30
parent 22154dee5c
commit fa96a1d1ef
2 changed files with 111 additions and 84 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

190
web/src/js/filt/filt.js
View File

@ -131,37 +131,40 @@ module.exports = (function() {
peg$c92 = "~u", peg$c92 = "~u",
peg$c93 = { type: "literal", value: "~u", description: "\"~u\"" }, peg$c93 = { type: "literal", value: "~u", description: "\"~u\"" },
peg$c94 = function(s) { return url(s); }, peg$c94 = function(s) { return url(s); },
peg$c95 = { type: "other", description: "integer" }, peg$c95 = "~websocket",
peg$c96 = /^['"]/, peg$c96 = { type: "literal", value: "~websocket", description: "\"~websocket\"" },
peg$c97 = { type: "class", value: "['\"]", description: "['\"]" }, peg$c97 = function() { return websocketFilter; },
peg$c98 = /^[0-9]/, peg$c98 = { type: "other", description: "integer" },
peg$c99 = { type: "class", value: "[0-9]", description: "[0-9]" }, peg$c99 = /^['"]/,
peg$c100 = function(digits) { return parseInt(digits.join(""), 10); }, peg$c100 = { type: "class", value: "['\"]", description: "['\"]" },
peg$c101 = { type: "other", description: "string" }, peg$c101 = /^[0-9]/,
peg$c102 = "\"", peg$c102 = { type: "class", value: "[0-9]", description: "[0-9]" },
peg$c103 = { type: "literal", value: "\"", description: "\"\\\"\"" }, peg$c103 = function(digits) { return parseInt(digits.join(""), 10); },
peg$c104 = function(chars) { return chars.join(""); }, peg$c104 = { type: "other", description: "string" },
peg$c105 = "'", peg$c105 = "\"",
peg$c106 = { type: "literal", value: "'", description: "\"'\"" }, peg$c106 = { type: "literal", value: "\"", description: "\"\\\"\"" },
peg$c107 = /^["\\]/, peg$c107 = function(chars) { return chars.join(""); },
peg$c108 = { type: "class", value: "[\"\\\\]", description: "[\"\\\\]" }, peg$c108 = "'",
peg$c109 = { type: "any", description: "any character" }, peg$c109 = { type: "literal", value: "'", description: "\"'\"" },
peg$c110 = function(char) { return char; }, peg$c110 = /^["\\]/,
peg$c111 = "\\", peg$c111 = { type: "class", value: "[\"\\\\]", description: "[\"\\\\]" },
peg$c112 = { type: "literal", value: "\\", description: "\"\\\\\"" }, peg$c112 = { type: "any", description: "any character" },
peg$c113 = /^['\\]/, peg$c113 = function(char) { return char; },
peg$c114 = { type: "class", value: "['\\\\]", description: "['\\\\]" }, peg$c114 = "\\",
peg$c115 = /^['"\\]/, peg$c115 = { type: "literal", value: "\\", description: "\"\\\\\"" },
peg$c116 = { type: "class", value: "['\"\\\\]", description: "['\"\\\\]" }, peg$c116 = /^['\\]/,
peg$c117 = "n", peg$c117 = { type: "class", value: "['\\\\]", description: "['\\\\]" },
peg$c118 = { type: "literal", value: "n", description: "\"n\"" }, peg$c118 = /^['"\\]/,
peg$c119 = function() { return "\n"; }, peg$c119 = { type: "class", value: "['\"\\\\]", description: "['\"\\\\]" },
peg$c120 = "r", peg$c120 = "n",
peg$c121 = { type: "literal", value: "r", description: "\"r\"" }, peg$c121 = { type: "literal", value: "n", description: "\"n\"" },
peg$c122 = function() { return "\r"; }, peg$c122 = function() { return "\n"; },
peg$c123 = "t", peg$c123 = "r",
peg$c124 = { type: "literal", value: "t", description: "\"t\"" }, peg$c124 = { type: "literal", value: "r", description: "\"r\"" },
peg$c125 = function() { return "\t"; }, peg$c125 = function() { return "\r"; },
peg$c126 = "t",
peg$c127 = { type: "literal", value: "t", description: "\"t\"" },
peg$c128 = function() { return "\t"; },
peg$currPos = 0, peg$currPos = 0,
peg$savedPos = 0, peg$savedPos = 0,
@ -1369,12 +1372,27 @@ module.exports = (function() {
} }
if (s0 === peg$FAILED) { if (s0 === peg$FAILED) {
s0 = peg$currPos; s0 = peg$currPos;
s1 = peg$parseStringLiteral(); if (input.substr(peg$currPos, 10) === peg$c95) {
s1 = peg$c95;
peg$currPos += 10;
} else {
s1 = peg$FAILED;
if (peg$silentFails === 0) { peg$fail(peg$c96); }
}
if (s1 !== peg$FAILED) { if (s1 !== peg$FAILED) {
peg$savedPos = s0; peg$savedPos = s0;
s1 = peg$c94(s1); s1 = peg$c97();
} }
s0 = s1; s0 = s1;
if (s0 === peg$FAILED) {
s0 = peg$currPos;
s1 = peg$parseStringLiteral();
if (s1 !== peg$FAILED) {
peg$savedPos = s0;
s1 = peg$c94(s1);
}
s0 = s1;
}
} }
} }
} }
@ -1408,53 +1426,53 @@ module.exports = (function() {
peg$silentFails++; peg$silentFails++;
s0 = peg$currPos; s0 = peg$currPos;
if (peg$c96.test(input.charAt(peg$currPos))) { if (peg$c99.test(input.charAt(peg$currPos))) {
s1 = input.charAt(peg$currPos); s1 = input.charAt(peg$currPos);
peg$currPos++; peg$currPos++;
} else { } else {
s1 = peg$FAILED; s1 = peg$FAILED;
if (peg$silentFails === 0) { peg$fail(peg$c97); } if (peg$silentFails === 0) { peg$fail(peg$c100); }
} }
if (s1 === peg$FAILED) { if (s1 === peg$FAILED) {
s1 = null; s1 = null;
} }
if (s1 !== peg$FAILED) { if (s1 !== peg$FAILED) {
s2 = []; s2 = [];
if (peg$c98.test(input.charAt(peg$currPos))) { if (peg$c101.test(input.charAt(peg$currPos))) {
s3 = input.charAt(peg$currPos); s3 = input.charAt(peg$currPos);
peg$currPos++; peg$currPos++;
} else { } else {
s3 = peg$FAILED; s3 = peg$FAILED;
if (peg$silentFails === 0) { peg$fail(peg$c99); } if (peg$silentFails === 0) { peg$fail(peg$c102); }
} }
if (s3 !== peg$FAILED) { if (s3 !== peg$FAILED) {
while (s3 !== peg$FAILED) { while (s3 !== peg$FAILED) {
s2.push(s3); s2.push(s3);
if (peg$c98.test(input.charAt(peg$currPos))) { if (peg$c101.test(input.charAt(peg$currPos))) {
s3 = input.charAt(peg$currPos); s3 = input.charAt(peg$currPos);
peg$currPos++; peg$currPos++;
} else { } else {
s3 = peg$FAILED; s3 = peg$FAILED;
if (peg$silentFails === 0) { peg$fail(peg$c99); } if (peg$silentFails === 0) { peg$fail(peg$c102); }
} }
} }
} else { } else {
s2 = peg$FAILED; s2 = peg$FAILED;
} }
if (s2 !== peg$FAILED) { if (s2 !== peg$FAILED) {
if (peg$c96.test(input.charAt(peg$currPos))) { if (peg$c99.test(input.charAt(peg$currPos))) {
s3 = input.charAt(peg$currPos); s3 = input.charAt(peg$currPos);
peg$currPos++; peg$currPos++;
} else { } else {
s3 = peg$FAILED; s3 = peg$FAILED;
if (peg$silentFails === 0) { peg$fail(peg$c97); } if (peg$silentFails === 0) { peg$fail(peg$c100); }
} }
if (s3 === peg$FAILED) { if (s3 === peg$FAILED) {
s3 = null; s3 = null;
} }
if (s3 !== peg$FAILED) { if (s3 !== peg$FAILED) {
peg$savedPos = s0; peg$savedPos = s0;
s1 = peg$c100(s2); s1 = peg$c103(s2);
s0 = s1; s0 = s1;
} else { } else {
peg$currPos = s0; peg$currPos = s0;
@ -1471,7 +1489,7 @@ module.exports = (function() {
peg$silentFails--; peg$silentFails--;
if (s0 === peg$FAILED) { if (s0 === peg$FAILED) {
s1 = peg$FAILED; s1 = peg$FAILED;
if (peg$silentFails === 0) { peg$fail(peg$c95); } if (peg$silentFails === 0) { peg$fail(peg$c98); }
} }
return s0; return s0;
@ -1483,11 +1501,11 @@ module.exports = (function() {
peg$silentFails++; peg$silentFails++;
s0 = peg$currPos; s0 = peg$currPos;
if (input.charCodeAt(peg$currPos) === 34) { if (input.charCodeAt(peg$currPos) === 34) {
s1 = peg$c102; s1 = peg$c105;
peg$currPos++; peg$currPos++;
} else { } else {
s1 = peg$FAILED; s1 = peg$FAILED;
if (peg$silentFails === 0) { peg$fail(peg$c103); } if (peg$silentFails === 0) { peg$fail(peg$c106); }
} }
if (s1 !== peg$FAILED) { if (s1 !== peg$FAILED) {
s2 = []; s2 = [];
@ -1498,15 +1516,15 @@ module.exports = (function() {
} }
if (s2 !== peg$FAILED) { if (s2 !== peg$FAILED) {
if (input.charCodeAt(peg$currPos) === 34) { if (input.charCodeAt(peg$currPos) === 34) {
s3 = peg$c102; s3 = peg$c105;
peg$currPos++; peg$currPos++;
} else { } else {
s3 = peg$FAILED; s3 = peg$FAILED;
if (peg$silentFails === 0) { peg$fail(peg$c103); } if (peg$silentFails === 0) { peg$fail(peg$c106); }
} }
if (s3 !== peg$FAILED) { if (s3 !== peg$FAILED) {
peg$savedPos = s0; peg$savedPos = s0;
s1 = peg$c104(s2); s1 = peg$c107(s2);
s0 = s1; s0 = s1;
} else { } else {
peg$currPos = s0; peg$currPos = s0;
@ -1523,11 +1541,11 @@ module.exports = (function() {
if (s0 === peg$FAILED) { if (s0 === peg$FAILED) {
s0 = peg$currPos; s0 = peg$currPos;
if (input.charCodeAt(peg$currPos) === 39) { if (input.charCodeAt(peg$currPos) === 39) {
s1 = peg$c105; s1 = peg$c108;
peg$currPos++; peg$currPos++;
} else { } else {
s1 = peg$FAILED; s1 = peg$FAILED;
if (peg$silentFails === 0) { peg$fail(peg$c106); } if (peg$silentFails === 0) { peg$fail(peg$c109); }
} }
if (s1 !== peg$FAILED) { if (s1 !== peg$FAILED) {
s2 = []; s2 = [];
@ -1538,15 +1556,15 @@ module.exports = (function() {
} }
if (s2 !== peg$FAILED) { if (s2 !== peg$FAILED) {
if (input.charCodeAt(peg$currPos) === 39) { if (input.charCodeAt(peg$currPos) === 39) {
s3 = peg$c105; s3 = peg$c108;
peg$currPos++; peg$currPos++;
} else { } else {
s3 = peg$FAILED; s3 = peg$FAILED;
if (peg$silentFails === 0) { peg$fail(peg$c106); } if (peg$silentFails === 0) { peg$fail(peg$c109); }
} }
if (s3 !== peg$FAILED) { if (s3 !== peg$FAILED) {
peg$savedPos = s0; peg$savedPos = s0;
s1 = peg$c104(s2); s1 = peg$c107(s2);
s0 = s1; s0 = s1;
} else { } else {
peg$currPos = s0; peg$currPos = s0;
@ -1585,7 +1603,7 @@ module.exports = (function() {
} }
if (s2 !== peg$FAILED) { if (s2 !== peg$FAILED) {
peg$savedPos = s0; peg$savedPos = s0;
s1 = peg$c104(s2); s1 = peg$c107(s2);
s0 = s1; s0 = s1;
} else { } else {
peg$currPos = s0; peg$currPos = s0;
@ -1600,7 +1618,7 @@ module.exports = (function() {
peg$silentFails--; peg$silentFails--;
if (s0 === peg$FAILED) { if (s0 === peg$FAILED) {
s1 = peg$FAILED; s1 = peg$FAILED;
if (peg$silentFails === 0) { peg$fail(peg$c101); } if (peg$silentFails === 0) { peg$fail(peg$c104); }
} }
return s0; return s0;
@ -1612,12 +1630,12 @@ module.exports = (function() {
s0 = peg$currPos; s0 = peg$currPos;
s1 = peg$currPos; s1 = peg$currPos;
peg$silentFails++; peg$silentFails++;
if (peg$c107.test(input.charAt(peg$currPos))) { if (peg$c110.test(input.charAt(peg$currPos))) {
s2 = input.charAt(peg$currPos); s2 = input.charAt(peg$currPos);
peg$currPos++; peg$currPos++;
} else { } else {
s2 = peg$FAILED; s2 = peg$FAILED;
if (peg$silentFails === 0) { peg$fail(peg$c108); } if (peg$silentFails === 0) { peg$fail(peg$c111); }
} }
peg$silentFails--; peg$silentFails--;
if (s2 === peg$FAILED) { if (s2 === peg$FAILED) {
@ -1632,11 +1650,11 @@ module.exports = (function() {
peg$currPos++; peg$currPos++;
} else { } else {
s2 = peg$FAILED; s2 = peg$FAILED;
if (peg$silentFails === 0) { peg$fail(peg$c109); } if (peg$silentFails === 0) { peg$fail(peg$c112); }
} }
if (s2 !== peg$FAILED) { if (s2 !== peg$FAILED) {
peg$savedPos = s0; peg$savedPos = s0;
s1 = peg$c110(s2); s1 = peg$c113(s2);
s0 = s1; s0 = s1;
} else { } else {
peg$currPos = s0; peg$currPos = s0;
@ -1649,17 +1667,17 @@ module.exports = (function() {
if (s0 === peg$FAILED) { if (s0 === peg$FAILED) {
s0 = peg$currPos; s0 = peg$currPos;
if (input.charCodeAt(peg$currPos) === 92) { if (input.charCodeAt(peg$currPos) === 92) {
s1 = peg$c111; s1 = peg$c114;
peg$currPos++; peg$currPos++;
} else { } else {
s1 = peg$FAILED; s1 = peg$FAILED;
if (peg$silentFails === 0) { peg$fail(peg$c112); } if (peg$silentFails === 0) { peg$fail(peg$c115); }
} }
if (s1 !== peg$FAILED) { if (s1 !== peg$FAILED) {
s2 = peg$parseEscapeSequence(); s2 = peg$parseEscapeSequence();
if (s2 !== peg$FAILED) { if (s2 !== peg$FAILED) {
peg$savedPos = s0; peg$savedPos = s0;
s1 = peg$c110(s2); s1 = peg$c113(s2);
s0 = s1; s0 = s1;
} else { } else {
peg$currPos = s0; peg$currPos = s0;
@ -1680,12 +1698,12 @@ module.exports = (function() {
s0 = peg$currPos; s0 = peg$currPos;
s1 = peg$currPos; s1 = peg$currPos;
peg$silentFails++; peg$silentFails++;
if (peg$c113.test(input.charAt(peg$currPos))) { if (peg$c116.test(input.charAt(peg$currPos))) {
s2 = input.charAt(peg$currPos); s2 = input.charAt(peg$currPos);
peg$currPos++; peg$currPos++;
} else { } else {
s2 = peg$FAILED; s2 = peg$FAILED;
if (peg$silentFails === 0) { peg$fail(peg$c114); } if (peg$silentFails === 0) { peg$fail(peg$c117); }
} }
peg$silentFails--; peg$silentFails--;
if (s2 === peg$FAILED) { if (s2 === peg$FAILED) {
@ -1700,11 +1718,11 @@ module.exports = (function() {
peg$currPos++; peg$currPos++;
} else { } else {
s2 = peg$FAILED; s2 = peg$FAILED;
if (peg$silentFails === 0) { peg$fail(peg$c109); } if (peg$silentFails === 0) { peg$fail(peg$c112); }
} }
if (s2 !== peg$FAILED) { if (s2 !== peg$FAILED) {
peg$savedPos = s0; peg$savedPos = s0;
s1 = peg$c110(s2); s1 = peg$c113(s2);
s0 = s1; s0 = s1;
} else { } else {
peg$currPos = s0; peg$currPos = s0;
@ -1717,17 +1735,17 @@ module.exports = (function() {
if (s0 === peg$FAILED) { if (s0 === peg$FAILED) {
s0 = peg$currPos; s0 = peg$currPos;
if (input.charCodeAt(peg$currPos) === 92) { if (input.charCodeAt(peg$currPos) === 92) {
s1 = peg$c111; s1 = peg$c114;
peg$currPos++; peg$currPos++;
} else { } else {
s1 = peg$FAILED; s1 = peg$FAILED;
if (peg$silentFails === 0) { peg$fail(peg$c112); } if (peg$silentFails === 0) { peg$fail(peg$c115); }
} }
if (s1 !== peg$FAILED) { if (s1 !== peg$FAILED) {
s2 = peg$parseEscapeSequence(); s2 = peg$parseEscapeSequence();
if (s2 !== peg$FAILED) { if (s2 !== peg$FAILED) {
peg$savedPos = s0; peg$savedPos = s0;
s1 = peg$c110(s2); s1 = peg$c113(s2);
s0 = s1; s0 = s1;
} else { } else {
peg$currPos = s0; peg$currPos = s0;
@ -1762,11 +1780,11 @@ module.exports = (function() {
peg$currPos++; peg$currPos++;
} else { } else {
s2 = peg$FAILED; s2 = peg$FAILED;
if (peg$silentFails === 0) { peg$fail(peg$c109); } if (peg$silentFails === 0) { peg$fail(peg$c112); }
} }
if (s2 !== peg$FAILED) { if (s2 !== peg$FAILED) {
peg$savedPos = s0; peg$savedPos = s0;
s1 = peg$c110(s2); s1 = peg$c113(s2);
s0 = s1; s0 = s1;
} else { } else {
peg$currPos = s0; peg$currPos = s0;
@ -1783,53 +1801,53 @@ module.exports = (function() {
function peg$parseEscapeSequence() { function peg$parseEscapeSequence() {
var s0, s1; var s0, s1;
if (peg$c115.test(input.charAt(peg$currPos))) { if (peg$c118.test(input.charAt(peg$currPos))) {
s0 = input.charAt(peg$currPos); s0 = input.charAt(peg$currPos);
peg$currPos++; peg$currPos++;
} else { } else {
s0 = peg$FAILED; s0 = peg$FAILED;
if (peg$silentFails === 0) { peg$fail(peg$c116); } if (peg$silentFails === 0) { peg$fail(peg$c119); }
} }
if (s0 === peg$FAILED) { if (s0 === peg$FAILED) {
s0 = peg$currPos; s0 = peg$currPos;
if (input.charCodeAt(peg$currPos) === 110) { if (input.charCodeAt(peg$currPos) === 110) {
s1 = peg$c117; s1 = peg$c120;
peg$currPos++; peg$currPos++;
} else { } else {
s1 = peg$FAILED; s1 = peg$FAILED;
if (peg$silentFails === 0) { peg$fail(peg$c118); } if (peg$silentFails === 0) { peg$fail(peg$c121); }
} }
if (s1 !== peg$FAILED) { if (s1 !== peg$FAILED) {
peg$savedPos = s0; peg$savedPos = s0;
s1 = peg$c119(); s1 = peg$c122();
} }
s0 = s1; s0 = s1;
if (s0 === peg$FAILED) { if (s0 === peg$FAILED) {
s0 = peg$currPos; s0 = peg$currPos;
if (input.charCodeAt(peg$currPos) === 114) { if (input.charCodeAt(peg$currPos) === 114) {
s1 = peg$c120; s1 = peg$c123;
peg$currPos++; peg$currPos++;
} else { } else {
s1 = peg$FAILED; s1 = peg$FAILED;
if (peg$silentFails === 0) { peg$fail(peg$c121); } if (peg$silentFails === 0) { peg$fail(peg$c124); }
} }
if (s1 !== peg$FAILED) { if (s1 !== peg$FAILED) {
peg$savedPos = s0; peg$savedPos = s0;
s1 = peg$c122(); s1 = peg$c125();
} }
s0 = s1; s0 = s1;
if (s0 === peg$FAILED) { if (s0 === peg$FAILED) {
s0 = peg$currPos; s0 = peg$currPos;
if (input.charCodeAt(peg$currPos) === 116) { if (input.charCodeAt(peg$currPos) === 116) {
s1 = peg$c123; s1 = peg$c126;
peg$currPos++; peg$currPos++;
} else { } else {
s1 = peg$FAILED; s1 = peg$FAILED;
if (peg$silentFails === 0) { peg$fail(peg$c124); } if (peg$silentFails === 0) { peg$fail(peg$c127); }
} }
if (s1 !== peg$FAILED) { if (s1 !== peg$FAILED) {
peg$savedPos = s0; peg$savedPos = s0;
s1 = peg$c125(); s1 = peg$c128();
} }
s0 = s1; s0 = s1;
} }
@ -1945,7 +1963,7 @@ module.exports = (function() {
function destinationFilter(flow){ function destinationFilter(flow){
return (!!flow.server_conn.address) return (!!flow.server_conn.address)
&& &&
regex.test(flow.server_conn.address.address[0] + ":" + flow.server_conn.address.address[1]); regex.test(flow.server_conn.address[0] + ":" + flow.server_conn.address[1]);
} }
destinationFilter.desc = "destination address matches " + regex; destinationFilter.desc = "destination address matches " + regex;
return destinationFilter; return destinationFilter;
@ -2011,7 +2029,7 @@ module.exports = (function() {
function sourceFilter(flow){ function sourceFilter(flow){
return (!!flow.client_conn.address) return (!!flow.client_conn.address)
&& &&
regex.test(flow.client_conn.address.address[0] + ":" + flow.client_conn.address.address[1]); regex.test(flow.client_conn.address[0] + ":" + flow.client_conn.address[1]);
} }
sourceFilter.desc = "source address matches " + regex; sourceFilter.desc = "source address matches " + regex;
return sourceFilter; return sourceFilter;
@ -2056,6 +2074,10 @@ module.exports = (function() {
urlFilter.desc = "url matches " + regex; urlFilter.desc = "url matches " + regex;
return urlFilter; return urlFilter;
} }
function websocketFilter(flow){
return flow.type === "websocket";
}
websocketFilter.desc = "is a Websocket Flow";
peg$result = peg$startRuleFunction(); peg$result = peg$startRuleFunction();

5
web/src/js/filt/filt.peg
View File

@ -217,6 +217,10 @@ function url(regex){
urlFilter.desc = "url matches " + regex; urlFilter.desc = "url matches " + regex;
return urlFilter; return urlFilter;
} }
function websocketFilter(flow){
return flow.type === "websocket";
}
websocketFilter.desc = "is a Websocket Flow";
} }
start "filter expression" start "filter expression"
@ -278,6 +282,7 @@ Expr
/ "~tq" ws+ s:StringLiteral { return requestContentType(s); } / "~tq" ws+ s:StringLiteral { return requestContentType(s); }
/ "~ts" ws+ s:StringLiteral { return responseContentType(s); } / "~ts" ws+ s:StringLiteral { return responseContentType(s); }
/ "~u" ws+ s:StringLiteral { return url(s); } / "~u" ws+ s:StringLiteral { return url(s); }
/ "~websocket" { return websocketFilter; }
/ s:StringLiteral { return url(s); } / s:StringLiteral { return url(s); }
IntegerLiteral "integer" IntegerLiteral "integer"