Minor refactoring for an Issue #295

This commit is contained in:
Miroslav Stampar
2013-01-08 10:23:02 +01:00
parent c155c6df84
commit ad85c4c964
2 changed files with 20 additions and 9 deletions

View File

@@ -8,10 +8,10 @@ See the file 'doc/COPYING' for copying permission
import os
import tempfile
from extra.cloak.cloak import decloak
from lib.core.agent import agent
from lib.core.common import dataToOutFile
from lib.core.common import Backend
from lib.core.common import decloakToTemp
from lib.core.common import decodeHexValue
from lib.core.common import isNumPosStrValue
from lib.core.common import isListLike
@@ -253,14 +253,8 @@ class Filesystem:
def writeFile(self, localFile, remoteFile, fileType=None):
self.checkDbmsOs()
if localFile.endswith("_"):
content = decloak(localFile)
_ = os.path.split(localFile[:-1])[-1]
prefix, suffix = os.path.splitext(_)
handle, localFile = tempfile.mkstemp(prefix=prefix, suffix=suffix)
os.close(handle)
with open(localFile, "w+b") as f:
f.write(content)
if localFile.endswith('_'):
localFile = decloakToTemp(localFile)
if conf.direct or isTechniqueAvailable(PAYLOAD.TECHNIQUE.STACKED):
if isTechniqueAvailable(PAYLOAD.TECHNIQUE.STACKED):