ModSecurity OWASP and Comodo Rules Integrated.

firewall/views.py

@@ -1121,3 +1121,107 @@ def installModSecRulesPack(request):
         json_data = json.dumps(data_ret)
         return HttpResponse(json_data)
 
+def getRulesFiles(request):
+    try:
+        val = request.session['userID']
+        try:
+            if request.method == 'POST':
+
+
+                data = json.loads(request.body)
+                packName = data['packName']
+
+                confPath = os.path.join(virtualHostUtilities.Server_root, 'conf/httpd_config.conf')
+
+                command = "sudo cat " + confPath
+                httpdConfig = subprocess.check_output(shlex.split(command)).splitlines()
+
+                json_data = "["
+                checker = 0
+                counter = 0
+
+                for items in httpdConfig:
+
+                    if items.find('modsec/'+packName) > -1:
+                        counter = counter + 1
+                        if items[0] == '#':
+                            status = False
+                        else:
+                            status = True
+
+                        fileName = items.lstrip('#')
+                        fileName = fileName.split('/')[-1]
+
+                        dic = {
+                               'id': counter,
+                               'fileName': fileName,
+                               'packName':packName,
+                               'status': status,
+
+                               }
+
+                        if checker == 0:
+                            json_data = json_data + json.dumps(dic)
+                            checker = 1
+                        else:
+                            json_data = json_data + ',' + json.dumps(dic)
+
+
+                json_data = json_data + ']'
+                final_json = json.dumps({'fetchStatus': 1, 'error_message': "None","data":json_data})
+                return HttpResponse(final_json)
+
+        except BaseException,msg:
+            final_dic = {'fetchStatus': 0, 'error_message': str(msg)}
+            final_json = json.dumps(final_dic)
+
+            return HttpResponse(final_json)
+    except KeyError:
+        final_dic = {'fetchStatus': 0, 'error_message': "Not Logged In, please refresh the page or login again."}
+        final_json = json.dumps(final_dic)
+        return HttpResponse(final_json)
+
+def enableDisableRuleFile(request):
+    try:
+        val = request.session['userID']
+        try:
+            if request.method == 'POST':
+
+                data = json.loads(request.body)
+
+                packName = data['packName']
+                fileName = data['fileName']
+                currentStatus = data['status']
+
+                if currentStatus == True:
+                    functionName = 'disableRuleFile'
+                else:
+                    functionName = 'enableRuleFile'
+
+
+                execPath = "sudo python " + virtualHostUtilities.cyberPanel + "/plogical/modSec.py"
+
+                execPath = execPath + " " + functionName + ' --packName ' + packName + ' --fileName ' + fileName
+
+                output = subprocess.check_output(shlex.split(execPath))
+
+                if output.find("1,None") > -1:
+                    installUtilities.reStartLiteSpeed()
+                    data_ret = {'saveStatus': 1, 'error_message': "None"}
+                    json_data = json.dumps(data_ret)
+                    return HttpResponse(json_data)
+                else:
+                    data_ret = {'saveStatus': 0, 'error_message': output}
+                    json_data = json.dumps(data_ret)
+                    return HttpResponse(json_data)
+
+
+        except BaseException,msg:
+            data_ret = {'saveStatus': 0, 'error_message': str(msg)}
+            json_data = json.dumps(data_ret)
+            return HttpResponse(json_data)
+    except BaseException, msg:
+        data_ret = {'saveStatus': 0, 'error_message': str(msg)}
+        json_data = json.dumps(data_ret)
+        return HttpResponse(json_data)
+