1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
|
from core.vectors import ModuleExec
from core.module import Module, Status
from core import modules
from core import messages
from core.loggers import log
from distutils import spawn
from mako import template
import tempfile
import subprocess
import atexit
class Mount(Module):
"""Mount remote filesystem using HTTPfs."""
def init(self):
self.register_info(
{
'author': [
'Emilio Pinna'
],
'license': 'GPLv3'
}
)
self.register_arguments([
{ 'name' : '-rpath', 'help' : 'Remote web path where to save the agent. If it is a folder find the first writable folder in it', 'default' : '.' },
{ 'name' : '-httpfs-binary', 'default' : 'httpfs' },
{ 'name' : '-no-autoremove', 'action' : 'store_true', 'default' : False, 'help' : 'Do not autoremove on exit' }
])
def run(self, **kwargs):
# Check binary
binary_path = spawn.find_executable(
self.args['httpfs_binary']
)
if not binary_path:
log.error(
messages.module_file_mount.httpfs_s_not_found % self.args['httpfs_binary']
)
return
# Generate PHP agent
try:
status = 0
agent = subprocess.check_output(
[ binary_path, 'generate', 'php' ]
)
except subprocess.CalledProcessError as e:
status = e.returncode
agent = ''
if status or not agent:
log.error(
messages.module_file_mount.error_generating_agent
)
return
# Save temporary PHP agent, and upload it
temp_file = tempfile.NamedTemporaryFile(
suffix = '.php',
prefix = '',
delete = False
)
temp_file.write(agent)
# Without this flush() uploads only a
# portion of the file
temp_file.flush()
result = ModuleExec(
'file_upload2web',
[
temp_file.name,
self.args['rpath']
]
).run()
temp_file.close()
if (
not result or
not result[0] or
len(result[0]) != 2 or
not result[0][0] or
not result[0][1]
):
log.error(
messages.module_file_mount.failed_agent_upload
)
return
self.args.update({
'agent_abs_path' : result[0][0],
'agent_url' : result[0][1]
})
log.warn(
template.Template(
messages.module_file_mount.agent_installed_tutorial
).render(**self.args)
)
if self.args['no_autoremove']:
log.warn(messages.module_file_mount.httpfs_agent_manually_remove_s % (result[0][0]))
else:
log.warn(messages.module_file_mount.httpfs_agent_removed)
atexit.register(
ModuleExec('file_rm', [
result[0][0]
]
).run
)
|
