def test_NetrFileClose(self):
dce, rpctransport = self.connect()
request = srvs.NetrFileEnum()
request['ServerName'] = '\\\\%s\x00' % self.machine
request['BasePath'] = NULL
request['UserName'] = NULL
request['InfoStruct']['Level'] = 2
request['InfoStruct']['FileInfo']['tag'] = 2
request['PreferedMaximumLength'] = 8192
resp = dce.request(request)
resp.dump()
request = srvs.NetrFileClose()
request['ServerName'] = '\\\\%s\x00' % self.machine
request['FileId'] = resp['InfoStruct']['FileInfo']['Level2']['Buffer'][
0]['fi2_id']
try:
resp = dce.request(request)
resp.dump()
except Exception as e:
# I might be closing myself ;)
if str(e).find('STATUS_PIPE_BROKEN') < 0 and str(e).find(
'STATUS_FILE_CLOSED') < 0 and str(e).find(
'STATUS_INVALID_HANDLE') < 0 and str(e).find(
'0x90a') < 0:
raise
def test_NetrFileEnum(self):
dce, rpctransport = self.connect()
request = srvs.NetrFileEnum()
request['ServerName'] = '\\\\%s\x00' % self.machine
request['BasePath'] = NULL
request['UserName'] = NULL
request['InfoStruct']['Level'] = 2
request['InfoStruct']['FileInfo']['tag'] = 2
request['PreferedMaximumLength'] = 8192
resp = dce.request(request)
#resp.dump()
request['InfoStruct']['Level'] = 3
request['InfoStruct']['FileInfo']['tag'] = 3
resp = dce.request(request)
