fix at OpCodeFile file format

mainProject/SFAnalytics/migrations/0004_stringlist.py

@@ -0,0 +1,22 @@
+# -*- coding: utf-8 -*-
+# Generated by Django 1.11 on 2017-06-11 14:34
+from __future__ import unicode_literals
+
+from django.db import migrations, models
+
+
+class Migration(migrations.Migration):
+
+    dependencies = [
+        ('SFAnalytics', '0003_auto_20170609_0148'),
+    ]
+
+    operations = [
+        migrations.CreateModel(
+            name='StringList',
+            fields=[
+                ('id', models.AutoField(auto_created=True, primary_key=True, serialize=False, verbose_name='ID')),
+                ('Strings', models.CharField(max_length=10)),
+            ],
+        ),
+    ]

mainProject/SFAnalytics/migrations/0005_auto_20170611_1512.py

@@ -0,0 +1,34 @@
+# -*- coding: utf-8 -*-
+# Generated by Django 1.11 on 2017-06-11 18:12
+from __future__ import unicode_literals
+
+from django.db import migrations, models
+
+
+class Migration(migrations.Migration):
+
+    dependencies = [
+        ('SFAnalytics', '0004_stringlist'),
+    ]
+
+    operations = [
+        migrations.RemoveField(
+            model_name='stringlist',
+            name='Strings',
+        ),
+        migrations.AddField(
+            model_name='stringlist',
+            name='string',
+            field=models.CharField(default='', max_length=250),
+        ),
+        migrations.AlterField(
+            model_name='dlllist',
+            name='name',
+            field=models.CharField(default='', max_length=100),
+        ),
+        migrations.AlterField(
+            model_name='opcodelist',
+            name='instruction',
+            field=models.CharField(default='', max_length=10),
+        ),
+    ]

mainProject/SFAnalytics/models.py

@@ -1,13 +1,13 @@
 from django.db import models
 
 class DllList(models.Model):
-    name = models.CharField(max_length=100)
+    name = models.CharField(default="", max_length=100)
 
 class OpCodeList(models.Model):
-    instruction = models.CharField(max_length=10)
+    instruction = models.CharField(default="", max_length=10)
 
 class StringList(models.Model):
-    Strings = models.CharField(max_length=10)
+    string = models.CharField(default="", max_length=250)
 
 class EntryFile(models.Model):
     hashMd5 = models.CharField(max_length=32)

mainProject/SFAnalytics/tools.py

@@ -19,6 +19,8 @@ def saveFile(data, path):
 #improve this to get file locations and path
 class OpCodeFile(object):
     def __init__(self, path):
+        #Chaging to windows path format
+        path = re.sub(r'/', '\\\\', path)
         #subprocess.run executes a cmd command : ['.\objdump.exe', '-d','7zFM.exe']
         out = subprocess.run(['C:\\Users\\lucas\\Documents\\TCC\\MalwareAnalysis\\Tools\\objdump.exe', '-d', path], stdout=subprocess.PIPE, shell=True).stdout.decode().splitlines()
         self.opcodes = []

mainProject/SFAnalytics/views.py

@@ -25,10 +25,10 @@ def uploadDataBaseView(request):
         if form.is_valid():
             for m in request.FILES.getlist('malwareUpload'):
                 HandleFile.saveFile(m, malwaresPath)
-                #addLearn(m, malwaresPath, True)
+                addLearn(m, malwaresPath, True)
             for s in request.FILES.getlist('softwareUpload'):
                 HandleFile.saveFile(s, softwaresPath)
-                #addLearn(s, softwaresPath, False)
+                addLearn(s, softwaresPath, False)
             return HttpResponseRedirect('/SFAnalytics/uploadDataBase/')
     else:
         form = UploadDataBaseForm()
@@ -44,3 +44,30 @@ def addLearn(data, path, malwareLabel):
     entry = EntryFile(hashMd5=fileHashs.getHashMd5(), hashSha256=fileHashs.getHashSha256(), malware=malwareLabel,
                       dataSectionEntropy=pe.getDataSectionEntropy(), textSectionEntropy=pe.getTextSectionEntropy())
     entry.save()
+    # Adding strings to string list
+    for s in strings.getStrings():
+        #Don't accept string lengths higher than 250 bytes
+        if len(s) < 251:
+            query = StringList.objects.all().filter(string__exact=s)
+            if query == None:
+                stringItem = StringList(string=s)
+                stringItem.save()
+
+    # Adding opcodes to OpCodeList
+    for op in opcodes.getOpcodes():
+        #Don't accept Operation Codes with length higher than 10 bytes
+        #X86 don't have any, this is here just to avoid software crash
+        if len(op) < 11:
+            query = OpCodeList.objects.all().filter(instruction__exact=op)
+            if query == None:
+                opItem = OpCodeList(instruction=op)
+                opItem.save()
+
+    # Adding dlls to DllList
+    for dll in pe.getDlls():
+        #Don't accept dll name lengths higher than 100 bytes
+        if len(dll) < 101:
+            query = OpCodeList.objects.all().filter(instruction__name=dll)
+            if query == None:
+                dllItem = DllList(name=dll)
+                dllItem.save()