PenTest/sqlmap
1
0
Fork 0
mirror of https://github.com/sqlmapproject/sqlmap.git synced 2025-12-09 14:11:29 +00:00
Code Issues Projects Releases Wiki Activity

Minor refactoring

Browse Source
This commit is contained in:
Miroslav Stampar
2014-04-06 18:19:54 +02:00
parent 3beb1ae2a1
commit 9c7fbd1a90
2 changed files with 12 additions and 6 deletions
Download Patch File Download Diff File Expand all files Collapse all files

13
lib/takeover/registry.py
View File

@@ -10,6 +10,7 @@ import os
from lib.core.common import randomStr
from lib.core.data import conf
from lib.core.data import logger
from lib.core.enums import REGISTRY_OPERATION
class Registry:
"""
@@ -49,11 +50,11 @@ class Registry:
def _createLocalBatchFile(self):
self._batPathFp = open(self._batPathLocal, "w")
if self._operation == "read":
if self._operation == REGISTRY_OPERATION.READ:
lines = self._batRead
elif self._operation == "add":
elif self._operation == REGISTRY_OPERATION.ADD:
lines = self._batAdd
elif self._operation == "delete":
elif self._operation == REGISTRY_OPERATION.DELETE:
lines = self._batDel
for line in lines:
@@ -70,7 +71,7 @@ class Registry:
os.unlink(self._batPathLocal)
def readRegKey(self, regKey, regValue, parse=False):
self._operation = "read"
self._operation = REGISTRY_OPERATION.READ
Registry._initVars(self, regKey, regValue, parse=parse)
self._createRemoteBatchFile()
@@ -90,7 +91,7 @@ class Registry:
return data
def addRegKey(self, regKey, regValue, regType, regData):
self._operation = "add"
self._operation = REGISTRY_OPERATION.ADD
Registry._initVars(self, regKey, regValue, regType, regData)
self._createRemoteBatchFile()
@@ -103,7 +104,7 @@ class Registry:
self.delRemoteFile(self._batPathRemote)
def delRegKey(self, regKey, regValue):
self._operation = "delete"
self._operation = REGISTRY_OPERATION.DELETE
Registry._initVars(self, regKey, regValue)
self._createRemoteBatchFile()