Changed notification popups (if its only information then no popup will appear)
15 from importlib import import_module
26 sys.stderr = open('/var/log/osecfs_error.log', 'a+')
29 MINOPTS = { "Main" : ["Logfile", "LogLevel", "Mountpoint", "Rootpath", "ScannerPath", "ScannerModuleName", "ScannerClassName", "ScannerConfig", "ReadOnly"]}
31 CONFIG_NOT_READABLE = "Configfile is not readable"
32 CONFIG_WRONG = "Something is wrong with the config"
33 CONFIG_MISSING = "Section: \"%s\" Option: \"%s\" in configfile is missing"
34 SCAN_WRONG_RETURN_VALUE = "The return Value of the malware scanner is wrong. Has to be an dictionary"
35 SCAN_RETURN_VALUE_KEY_MISSING = "The dictionary has to include key \"infected\" (True, False) and \"virusname\" (String)"
36 VIRUS_FOUND = "Virus found. Access denied"
37 NOTIFICATION_CRITICAL = "critical"
38 NOTIFICATION_INFO = "info"
43 SYSTEM_FILE_COMMAND = "file"
44 httpPool = urllib3.PoolManager(num_pools = 1, timeout = 3)
46 def checkMinimumOptions (config):
47 for section, options in MINOPTS.iteritems ():
48 for option in options:
49 if (config.has_option(section, option) == False):
50 print (CONFIG_MISSING % (section, option))
55 print ("%s configfile mountpath ro/rw" % (sys.argv[0]))
61 if (len (sys.argv) < 4):
64 configfile = sys.argv[1]
65 config = ConfigParser.SafeConfigParser ()
67 if ((os.path.exists (configfile) == False) or (os.path.isfile (configfile) == False) or (os.access (configfile, os.R_OK) == False)):
68 print (CONFIG_NOT_READABLE)
72 config.read (sys.argv[1])
75 print ("Error: %s" % (e))
78 config.set("Main", "Mountpoint", sys.argv[2])
79 if (sys.argv[3] == "rw"):
80 config.set("Main", "ReadOnly", "false")
82 config.set("Main", "ReadOnly", "true")
84 checkMinimumOptions (config)
92 logfile = config.get("Main", "Logfile")
94 numeric_level = getattr(logging, config.get("Main", "LogLevel").upper(), None)
95 if not isinstance(numeric_level, int):
96 raise ValueError('Invalid log level: %s' % loglevel)
98 # ToDo move log level and maybe other things to config file
100 level = numeric_level,
101 format = "%(asctime)s %(name)-12s %(funcName)-15s %(levelname)-8s %(message)s",
102 datefmt = "%Y-%m-%d %H:%M:%S",
106 LOG = logging.getLogger("fuse_main")
110 return ".%s" % (path)
112 def rootPath (rootpath, path):
113 return "%s%s" % (rootpath, path)
115 def flag2mode (flags):
116 md = {os.O_RDONLY: 'r', os.O_WRONLY: 'w', os.O_RDWR: 'w+'}
117 m = md[flags & (os.O_RDONLY | os.O_WRONLY | os.O_RDWR)]
119 # windows sets append even if it would overwrite the whole file (seek 0)
120 # so ignore append option
121 #if flags | os.O_APPEND:
122 # m = m.replace('w', 'a', 1)
126 def scanFile (path, fileobject):
127 LOG.debug ("Scan File \"%s\" with malware Scanner" %(path,) )
128 return MalwareScanner.scanFile (path, fileobject)
131 def scanFileClamAV (path):
134 LOG.debug ("Scan File: %s" % (path))
136 result = pyclamav.scanfile (path)
137 LOG.debug ("Result of file \"%s\": %s" % (path, result))
141 if (infected == True):
142 LOG.error ("Virus found, deny Access %s" % (result,))
146 def whitelistFile (path):
149 LOG.debug ("Execute \"%s\" command on \"%s\"" %(SYSTEM_FILE_COMMAND, path))
153 result = subprocess.check_output ([SYSTEM_FILE_COMMAND, path]);
154 # ToDo replace with real whitelist
156 except Exception as e:
157 LOG.error ("Call returns with an error!")
160 LOG.debug ("Type: %s" %(result))
164 def sendDataToRest (urlpath, data):
165 netifaces.ifaddresses("eth0")[2][0]["addr"]
167 # Get first address in network (0 = network ip -> 192.168.0.0)
168 remote_ip = netaddr.IPNetwork("%s/%s" %(netifaces.ifaddresses("eth0")[2][0]["addr"], netifaces.ifaddresses("eth0")[2][0]["netmask"]))[1]
170 url = ("http://%s:8090//%s" %(remote_ip, urlpath))
172 LOG.debug ("Send data to \"%s\"" %(url, ))
173 LOG.debug ("Data: %s" %(data, ))
176 response = httpPool.request_encode_body("POST", url, fields=data, retries=0)
178 LOG.error("Remote host not reachable")
179 LOG.error ("Exception: %s" %(e,))
182 if response.status == STATUS_CODE_OK:
183 LOG.info("Data sent successfully to rest server")
186 LOG.error("Server returned errorcode: %s" %(response.status,))
190 def sendNotification (type, message):
191 data = {"msgtype" : type, "text" : message}
193 if (type == "information"):
194 sendDataToRest ("message", data)
196 sendDataToRest ("notification", data)
198 def sendReadOnlyNotification():
199 sendNotification("critical", "Filesystem is in read only mode. If you want to export files please initialize an encrypted filesystem.")
201 def sendLogNotPossibleNotification():
202 sendNotification("critical", "Send log entry to opensecurity rest server failed.")
204 def sendFileLog(filename, filesize, filehash, hashtype):
205 data = {"filename" : filename, "filesize" : "%s" %(filesize,), "filehash" : filehash, "hashtype" : hashtype}
206 retval = sendDataToRest ("log", data)
207 if (retval == False):
208 sendLogNotPossibleNotification()
210 def calcMD5 (path, block_size=256*128, hr=True):
212 with open(path,'rb') as f:
213 for chunk in iter(lambda: f.read(block_size), b''):
216 return md5.hexdigest()
224 def __init__(self, rootpath, *args, **kw):
225 self.__rootpath = rootpath
226 Fuse.__init__ (self, *args, **kw)
227 LOG.debug ("Init complete.")
228 sendNotification("information", "Filesystem successfully mounted.")
230 # defines that our working directory will be the __rootpath
232 os.chdir (self.__rootpath)
234 def getattr(self, path):
235 LOG.debug ("*** getattr (%s)" % (fixPath (path)))
236 return os.lstat (fixPath (path));
238 def getdir(self, path):
239 LOG.debug ("*** getdir (%s)" % (path));
240 return os.listdir (fixPath (path))
242 def readdir(self, path, offset):
243 LOG.debug ("*** readdir (%s %s)" % (path, offset));
244 for e in os.listdir (fixPath (path)):
245 yield fuse.Direntry(e)
247 def chmod (self, path, mode):
248 LOG.debug ("*** chmod %s %s" % (path, oct(mode)))
249 if (config.get("Main", "ReadOnly") == "true"):
250 sendReadOnlyNotification()
252 os.chmod (fixPath (path), mode)
254 def chown (self, path, uid, gid):
255 LOG.debug ("*** chown %s %s %s" % (path, uid, gid))
256 if (config.get("Main", "ReadOnly") == "true"):
257 sendReadOnlyNotification()
259 os.chown (fixPath (path), uid, gid)
261 def link (self, targetPath, linkPath):
262 LOG.debug ("*** link %s %s" % (targetPath, linkPath))
263 if (config.get("Main", "ReadOnly") == "true"):
264 sendReadOnlyNotification()
266 os.link (fixPath (targetPath), fixPath (linkPath))
268 def mkdir (self, path, mode):
269 LOG.debug ("*** mkdir %s %s" % (path, oct(mode)))
270 if (config.get("Main", "ReadOnly") == "true"):
271 sendReadOnlyNotification()
273 os.mkdir (fixPath (path), mode)
275 def mknod (self, path, mode, dev):
276 LOG.debug ("*** mknod %s %s %s" % (path, oct (mode), dev))
277 if (config.get("Main", "ReadOnly") == "true"):
278 sendReadOnlyNotification()
280 os.mknod (fixPath (path), mode, dev)
282 # to implement virus scan
283 def open (self, path, flags):
284 LOG.debug ("*** open %s %s" % (path, oct (flags)))
285 self.file = os.fdopen (os.open (fixPath (path), flags), flag2mode (flags))
287 self.fd = self.file.fileno ()
289 LOG.debug(self.__rootpath)
292 retval = scanFile (rootPath(self.__rootpath, path), self.file)
294 #if type(retval) is not dict:
295 if (isinstance(retval, dict) == False):
296 LOG.error(SCAN_WRONG_RETURN_VALUE)
300 if ((retval.has_key("infected") == False) or (retval.has_key("virusname") == False)):
301 LOG.error(SCAN_RETURN_VALUE_KEY_MISSING)
306 if (retval.get("infected") == True):
308 sendNotification(NOTIFICATION_CRITICAL, "%s\nFile: %s\nVirus: %s" %(VIRUS_FOUND, path, retval.get("virusname")))
309 LOG.error("%s" %(VIRUS_FOUND,))
310 LOG.error("Virus: %s" %(retval.get("virusname"),))
313 whitelisted = whitelistFile (rootPath(self.__rootpath, path))
314 if (whitelisted == False):
316 sendNotification(NOTIFICATION_CRITICAL, "File not in whitelist. Access denied.")
319 def read (self, path, length, offset):
320 LOG.debug ("*** read %s %s %s" % (path, length, offset))
321 self.file.seek (offset)
322 return self.file.read (length)
324 def readlink (self, path):
325 LOG.debug ("*** readlink %s" % (path))
326 return os.readlink (fixPath (path))
328 def release (self, path, flags):
329 LOG.debug ("*** release %s %s" % (path, oct (flags)))
331 os.fsync(self.file.fileno())
334 if (self.written == True):
335 hashsum = calcMD5(fixPath(path))
336 filesize = os.path.getsize(fixPath(path))
337 sendFileLog(path, filesize, hashsum, "md5")
340 def rename (self, oldPath, newPath):
341 LOG.debug ("*** rename %s %s %s" % (oldPath, newPath, config.get("Main", "ReadOnly")))
342 if (config.get("Main", "ReadOnly") == "true"):
343 sendReadOnlyNotification()
345 os.rename (fixPath (oldPath), fixPath (newPath))
347 def rmdir (self, path):
348 LOG.debug ("*** rmdir %s %s" % (path, config.get("Main", "ReadOnly")))
349 if (config.get("Main", "ReadOnly") == "true"):
350 sendReadOnlyNotification()
352 os.rmdir (fixPath (path))
355 LOG.debug ("*** statfs")
356 return os.statvfs(".")
358 def symlink (self, targetPath, linkPath):
359 LOG.debug ("*** symlink %s %s %s" % (targetPath, linkPath, config.get("Main", "ReadOnly")))
360 if (config.get("Main", "ReadOnly") == "true"):
361 sendReadOnlyNotification()
363 os.symlink (fixPath (targetPath), fixPath (linkPath))
365 def truncate (self, path, length):
366 LOG.debug ("*** truncate %s %s %s" % (path, length, config.get("Main", "ReadOnly")))
367 if (config.get("Main", "ReadOnly") == "true"):
368 sendReadOnlyNotification()
370 f = open (fixPath (path), "w+")
374 def unlink (self, path):
375 LOG.debug ("*** unlink %s %s" % (path, config.get("Main", "ReadOnly")))
376 if (config.get("Main", "ReadOnly") == "true"):
377 sendReadOnlyNotification()
379 os.unlink (fixPath (path))
381 def utime (self, path, times):
382 LOG.debug ("*** utime %s %s" % (path, times))
383 os.utime (fixPath (path), times)
385 def write (self, path, buf, offset):
386 #LOG.debug ("*** write %s %s %s %s" % (path, buf, offset, config.get("Main", "ReadOnly")))
387 LOG.debug ("*** write %s %s %s %s" % (path, "filecontent", offset, config.get("Main", "ReadOnly")))
388 if (config.get("Main", "ReadOnly") == "true"):
390 sendReadOnlyNotification()
392 self.file.seek (offset)
393 self.file.write (buf)
397 def access (self, path, mode):
398 LOG.debug ("*** access %s %s" % (path, oct (mode)))
399 if not os.access (fixPath (path), mode):
402 def create (self, path, flags, mode):
403 LOG.debug ("*** create %s %s %s %s %s" % (fixPath (path), oct (flags), oct (mode), flag2mode (flags), config.get("Main", "ReadOnly")))
404 if (config.get("Main", "ReadOnly") == "true"):
405 sendReadOnlyNotification()
408 self.file = os.fdopen (os.open (fixPath (path), flags), flag2mode(flags))
410 self.fd = self.file.fileno ()
413 if __name__ == "__main__":
415 fuse.fuse_python_api = (0, 2)
416 fuse.feature_assert ('stateful_files', 'has_init')
418 config = loadConfig ()
421 #sendNotification("Info", "OsecFS started")
423 # Import the Malware Scanner
424 sys.path.append(config.get("Main", "ScannerPath"))
426 MalwareModule = import_module(config.get("Main", "ScannerModuleName"))
427 MalwareClass = getattr(MalwareModule, config.get("Main", "ScannerClassName"))
429 MalwareScanner = MalwareClass (config.get("Main", "ScannerConfig"));
431 osecfs = OsecFS (config.get ("Main", "Rootpath"))
433 osecfs.multithreaded = 0
435 fuse_args = [sys.argv[0], config.get ("Main", "Mountpoint")];
436 osecfs.main (fuse_args)