mirror of
https://github.com/byt3bl33d3r/MITMf.git
synced 2025-07-06 21:12:16 -07:00
Added regex support.
Plugin now takes a search and replace string (no regex supported) or a regex file. Regex file format is: <regex1>\t<regex2> Regex file contains pyton regexes so matching groups can be used
This commit is contained in:
parent
3d3099fb00
commit
d6f3405f71
1 changed files with 18 additions and 4 deletions
|
@ -15,11 +15,16 @@ class Replace(CacheKill,Plugin):
|
|||
self.options = options
|
||||
self.search_str = options.search_str
|
||||
self.replace_str = options.replace_str
|
||||
self.regex_file = options.regex_file
|
||||
|
||||
if self.options.keep_cache:
|
||||
self.implements.remove("handleHeader")
|
||||
self.implements.remove("connectionMade")
|
||||
|
||||
if self.search_str==self.replace_str=="" and self.regex_file is None:
|
||||
print "[*] Please provide a search and replace string or a regex file"
|
||||
quit()
|
||||
|
||||
self.ctable = {}
|
||||
self.dtable = {}
|
||||
self.mime = "text/html"
|
||||
|
@ -29,18 +34,27 @@ class Replace(CacheKill,Plugin):
|
|||
def handleResponse(self,request,data):
|
||||
ip,hn,mime = self._get_req_info(request)
|
||||
|
||||
if self._should_replace(ip,hn,mime) and (not self.replace_str==self.search_str==None) and (not self.search_str==""):
|
||||
data = data.replace(self.search_str, self.replace_str)
|
||||
if self._should_replace(ip,hn,mime):
|
||||
|
||||
if self.search_str==self.replace_str!="":
|
||||
data = data.replace(self.search_str, self.replace_str)
|
||||
logging.info("%s [%s] Replaced '%s' with '%s'" % (request.client.getClientIP(), request.headers['host'], self.search_str, self.replace_str))
|
||||
|
||||
if self.regex_file is not None:
|
||||
for line in self.regex_file:
|
||||
replaceRegex = line.split("\t")
|
||||
data = re.sub(replaceRegex[0], replaceRegex[1], data)
|
||||
logging.info("%s [%s] Replaced '%s' with '%s'" % (request.client.getClientIP(), request.headers['host'], replaceRegex[0], replaceRegex[1]))
|
||||
|
||||
self.ctable[ip] = time.time()
|
||||
self.dtable[ip+hn] = True
|
||||
|
||||
logging.info("%s [%s] Replaced '%s' with '%s'" % (request.client.getClientIP(), request.headers['host'], self.search_str, self.replace_str))
|
||||
|
||||
return {'request':request,'data':data}
|
||||
|
||||
return
|
||||
|
||||
def add_options(self,options):
|
||||
options.add_argument("--regex-file",type=file,help="Load file with regexes")
|
||||
options.add_argument("--replace-str",type=str,default="",help="String you would like to replace.")
|
||||
options.add_argument("--search-str",type=str,default="",help="String you would like to replace --replace-str with. Default: '' (empty string)")
|
||||
options.add_argument("--keep-cache",action="store_true",help="Don't kill the server/client caching.")
|
||||
|
|
Loading…
Add table
Add a link
Reference in a new issue