Commit 521d90598cd3a85fd085025699d41b17f88fabc1

Authored by Imanol-Mikel Barba Sabariego
1 parent dbab5089

Progress on various modules

.idea/workspace.xml
... ... @@ -2,7 +2,12 @@
2 2 <project version="4">
3 3 <component name="ChangeListManager">
4 4 <list default="true" id="33419617-7da8-422e-9ac6-68e0b317ebf8" name="Default" comment="">
  5 + <change type="NEW" beforePath="" afterPath="$PROJECT_DIR$/modules/info.py" />
  6 + <change type="NEW" beforePath="" afterPath="$PROJECT_DIR$/modules/winreg.py" />
  7 + <change type="MODIFICATION" beforePath="$PROJECT_DIR$/.idea/workspace.xml" afterPath="$PROJECT_DIR$/.idea/workspace.xml" />
  8 + <change type="MODIFICATION" beforePath="$PROJECT_DIR$/digger.py" afterPath="$PROJECT_DIR$/digger.py" />
5 9 <change type="MODIFICATION" beforePath="$PROJECT_DIR$/modules/mft.py" afterPath="$PROJECT_DIR$/modules/mft.py" />
  10 + <change type="MODIFICATION" beforePath="$PROJECT_DIR$/tomb.py" afterPath="$PROJECT_DIR$/tomb.py" />
6 11 </list>
7 12 <ignored path="gravedigger.iws" />
8 13 <ignored path=".idea/workspace.xml" />
... ... @@ -27,21 +32,11 @@
27 32 </component>
28 33 <component name="FileEditorManager">
29 34 <leaf>
30   - <file leaf-file-name="digger.sh" pinned="false" current-in-tab="false">
31   - <entry file="file://$PROJECT_DIR$/scripts/digger.sh">
32   - <provider selected="true" editor-type-id="text-editor">
33   - <state relative-caret-position="133">
34   - <caret line="7" column="26" selection-start-line="7" selection-start-column="26" selection-end-line="7" selection-end-column="26" />
35   - <folding />
36   - </state>
37   - </provider>
38   - </entry>
39   - </file>
40   - <file leaf-file-name="digger.py" pinned="false" current-in-tab="false">
  35 + <file leaf-file-name="digger.py" pinned="false" current-in-tab="true">
41 36 <entry file="file://$PROJECT_DIR$/digger.py">
42 37 <provider selected="true" editor-type-id="text-editor">
43   - <state relative-caret-position="-190">
44   - <caret line="11" column="0" selection-start-line="11" selection-start-column="0" selection-end-line="11" selection-end-column="0" />
  38 + <state relative-caret-position="446">
  39 + <caret line="75" column="13" selection-start-line="75" selection-start-column="13" selection-end-line="75" selection-end-column="13" />
45 40 <folding>
46 41 <element signature="e#25#38#0" expanded="true" />
47 42 </folding>
... ... @@ -62,8 +57,8 @@
62 57 <file leaf-file-name="module.py" pinned="false" current-in-tab="false">
63 58 <entry file="file://$PROJECT_DIR$/module.py">
64 59 <provider selected="true" editor-type-id="text-editor">
65   - <state relative-caret-position="304">
66   - <caret line="16" column="58" selection-start-line="16" selection-start-column="58" selection-end-line="16" selection-end-column="58" />
  60 + <state relative-caret-position="152">
  61 + <caret line="8" column="21" selection-start-line="8" selection-start-column="21" selection-end-line="8" selection-end-column="21" />
67 62 <folding />
68 63 </state>
69 64 </provider>
... ... @@ -72,7 +67,7 @@
72 67 <file leaf-file-name="ntfsvol.py" pinned="false" current-in-tab="false">
73 68 <entry file="file://$PROJECT_DIR$/vars/ntfsvol.py">
74 69 <provider selected="true" editor-type-id="text-editor">
75   - <state relative-caret-position="542">
  70 + <state relative-caret-position="608">
76 71 <caret line="32" column="29" selection-start-line="32" selection-start-column="29" selection-end-line="32" selection-end-column="29" />
77 72 <folding>
78 73 <element signature="e#0#31#0" expanded="true" />
... ... @@ -91,11 +86,11 @@
91 86 </provider>
92 87 </entry>
93 88 </file>
94   - <file leaf-file-name="mft.py" pinned="false" current-in-tab="true">
  89 + <file leaf-file-name="mft.py" pinned="false" current-in-tab="false">
95 90 <entry file="file://$PROJECT_DIR$/modules/mft.py">
96 91 <provider selected="true" editor-type-id="text-editor">
97   - <state relative-caret-position="190">
98   - <caret line="10" column="23" selection-start-line="10" selection-start-column="23" selection-end-line="10" selection-end-column="23" />
  92 + <state relative-caret-position="0">
  93 + <caret line="0" column="0" selection-start-line="0" selection-start-column="0" selection-end-line="25" selection-end-column="0" />
99 94 <folding>
100 95 <element signature="e#0#25#0" expanded="true" />
101 96 </folding>
... ... @@ -103,31 +98,45 @@
103 98 </provider>
104 99 </entry>
105 100 </file>
106   - <file leaf-file-name="tomb.py" pinned="false" current-in-tab="false">
107   - <entry file="file://$PROJECT_DIR$/tomb.py">
  101 + <file leaf-file-name="winreg.py" pinned="false" current-in-tab="false">
  102 + <entry file="file://$PROJECT_DIR$/modules/winreg.py">
108 103 <provider selected="true" editor-type-id="text-editor">
109   - <state relative-caret-position="190">
110   - <caret line="10" column="25" selection-start-line="10" selection-start-column="25" selection-end-line="10" selection-end-column="25" />
111   - <folding />
  104 + <state relative-caret-position="323">
  105 + <caret line="17" column="12" selection-start-line="17" selection-start-column="12" selection-end-line="18" selection-end-column="10" />
  106 + <folding>
  107 + <element signature="e#0#25#0" expanded="true" />
  108 + </folding>
112 109 </state>
113 110 </provider>
114 111 </entry>
115 112 </file>
116   - <file leaf-file-name="modulevar.py" pinned="false" current-in-tab="false">
117   - <entry file="file://$PROJECT_DIR$/modulevar.py">
  113 + <file leaf-file-name="info.py" pinned="false" current-in-tab="false">
  114 + <entry file="file://$PROJECT_DIR$/modules/info.py">
  115 + <provider selected="true" editor-type-id="text-editor">
  116 + <state relative-caret-position="266">
  117 + <caret line="14" column="22" selection-start-line="14" selection-start-column="22" selection-end-line="14" selection-end-column="22" />
  118 + <folding>
  119 + <element signature="e#0#25#0" expanded="true" />
  120 + </folding>
  121 + </state>
  122 + </provider>
  123 + </entry>
  124 + </file>
  125 + <file leaf-file-name="tomb.py" pinned="false" current-in-tab="false">
  126 + <entry file="file://$PROJECT_DIR$/tomb.py">
118 127 <provider selected="true" editor-type-id="text-editor">
119 128 <state relative-caret-position="114">
120   - <caret line="6" column="21" selection-start-line="6" selection-start-column="21" selection-end-line="6" selection-end-column="21" />
  129 + <caret line="6" column="27" selection-start-line="6" selection-start-column="27" selection-end-line="6" selection-end-column="27" />
121 130 <folding />
122 131 </state>
123 132 </provider>
124 133 </entry>
125 134 </file>
126   - <file leaf-file-name="__init__.py" pinned="false" current-in-tab="false">
127   - <entry file="file://$PROJECT_DIR$/modules/__init__.py">
  135 + <file leaf-file-name="modulevar.py" pinned="false" current-in-tab="false">
  136 + <entry file="file://$PROJECT_DIR$/modulevar.py">
128 137 <provider selected="true" editor-type-id="text-editor">
129   - <state relative-caret-position="0">
130   - <caret line="0" column="0" selection-start-line="0" selection-start-column="0" selection-end-line="0" selection-end-column="0" />
  138 + <state relative-caret-position="114">
  139 + <caret line="6" column="21" selection-start-line="6" selection-start-column="21" selection-end-line="6" selection-end-column="21" />
131 140 <folding />
132 141 </state>
133 142 </provider>
... ... @@ -152,7 +161,6 @@
152 161 <option value="$PROJECT_DIR$/dig.service" />
153 162 <option value="$PROJECT_DIR$/modules/module.py" />
154 163 <option value="$PROJECT_DIR$/modulevar.py" />
155   - <option value="$PROJECT_DIR$/digger.py" />
156 164 <option value="$PROJECT_DIR$/module.py" />
157 165 <option value="$PROJECT_DIR$/scripts/digger.sh" />
158 166 <option value="$PROJECT_DIR$/runcmd.py" />
... ... @@ -160,6 +168,9 @@
160 168 <option value="$PROJECT_DIR$/scripts/list_ntfs.sh" />
161 169 <option value="$PROJECT_DIR$/tomb.py" />
162 170 <option value="$PROJECT_DIR$/modules/mft.py" />
  171 + <option value="$PROJECT_DIR$/modules/info.py" />
  172 + <option value="$PROJECT_DIR$/modules/winreg.py" />
  173 + <option value="$PROJECT_DIR$/digger.py" />
163 174 </list>
164 175 </option>
165 176 </component>
... ... @@ -170,10 +181,10 @@
170 181 <sorting>DEFINITION_ORDER</sorting>
171 182 </component>
172 183 <component name="ProjectFrameBounds">
173   - <option name="x" value="1915" />
174   - <option name="y" value="31" />
  184 + <option name="x" value="2113" />
  185 + <option name="y" value="215" />
175 186 <option name="width" value="1930" />
176   - <option name="height" value="1174" />
  187 + <option name="height" value="1173" />
177 188 </component>
178 189 <component name="ProjectLevelVcsManager" settingsEditedManually="false">
179 190 <OptionsSetting value="true" id="Add" />
... ... @@ -244,20 +255,6 @@
244 255 <option name="myItemType" value="com.intellij.ide.projectView.impl.nodes.PsiDirectoryNode" />
245 256 </PATH_ELEMENT>
246 257 <PATH_ELEMENT>
247   - <option name="myItemId" value="scripts" />
248   - <option name="myItemType" value="com.intellij.ide.projectView.impl.nodes.PsiDirectoryNode" />
249   - </PATH_ELEMENT>
250   - </PATH>
251   - <PATH>
252   - <PATH_ELEMENT>
253   - <option name="myItemId" value="gravedigger" />
254   - <option name="myItemType" value="com.intellij.ide.projectView.impl.nodes.ProjectViewProjectNode" />
255   - </PATH_ELEMENT>
256   - <PATH_ELEMENT>
257   - <option name="myItemId" value="gravedigger" />
258   - <option name="myItemType" value="com.intellij.ide.projectView.impl.nodes.PsiDirectoryNode" />
259   - </PATH_ELEMENT>
260   - <PATH_ELEMENT>
261 258 <option name="myItemId" value="modules" />
262 259 <option name="myItemType" value="com.intellij.ide.projectView.impl.nodes.PsiDirectoryNode" />
263 260 </PATH_ELEMENT>
... ... @@ -508,6 +505,9 @@
508 505 <component name="ShelveChangesManager" show_recycled="false">
509 506 <option name="remove_strategy" value="false" />
510 507 </component>
  508 + <component name="SvnConfiguration">
  509 + <configuration />
  510 + </component>
511 511 <component name="TaskManager">
512 512 <task active="true" id="Default" summary="Default task">
513 513 <changelist id="33419617-7da8-422e-9ac6-68e0b317ebf8" name="Default" comment="" />
... ... @@ -519,7 +519,7 @@
519 519 <servers />
520 520 </component>
521 521 <component name="ToolWindowManager">
522   - <frame x="1915" y="31" width="1930" height="1174" extended-state="6" />
  522 + <frame x="2113" y="215" width="1930" height="1173" extended-state="0" />
523 523 <editor active="true" />
524 524 <layout>
525 525 <window_info id="Project" active="false" anchor="left" auto_hide="false" internal_type="DOCKED" type="DOCKED" visible="true" show_stripe_button="true" weight="0.25" sideWeight="0.5" order="0" side_tool="false" content_ui="combo" />
... ... @@ -528,7 +528,7 @@
528 528 <window_info id="Database" active="false" anchor="right" auto_hide="false" internal_type="DOCKED" type="DOCKED" visible="false" show_stripe_button="true" weight="0.33" sideWeight="0.5" order="-1" side_tool="false" content_ui="tabs" />
529 529 <window_info id="Python Console" active="false" anchor="bottom" auto_hide="false" internal_type="DOCKED" type="DOCKED" visible="false" show_stripe_button="true" weight="0.33" sideWeight="0.5" order="-1" side_tool="false" content_ui="tabs" />
530 530 <window_info id="Version Control" active="false" anchor="bottom" auto_hide="false" internal_type="DOCKED" type="DOCKED" visible="false" show_stripe_button="true" weight="0.33" sideWeight="0.5" order="-1" side_tool="false" content_ui="tabs" />
531   - <window_info id="Run" active="false" anchor="bottom" auto_hide="false" internal_type="DOCKED" type="DOCKED" visible="true" show_stripe_button="true" weight="0.32992566" sideWeight="0.5" order="2" side_tool="false" content_ui="tabs" />
  531 + <window_info id="Run" active="false" anchor="bottom" auto_hide="false" internal_type="DOCKED" type="DOCKED" visible="true" show_stripe_button="true" weight="0.33023256" sideWeight="0.5" order="2" side_tool="false" content_ui="tabs" />
532 532 <window_info id="Structure" active="false" anchor="left" auto_hide="false" internal_type="DOCKED" type="DOCKED" visible="false" show_stripe_button="true" weight="0.25" sideWeight="0.5" order="1" side_tool="false" content_ui="tabs" />
533 533 <window_info id="Terminal" active="false" anchor="bottom" auto_hide="false" internal_type="DOCKED" type="DOCKED" visible="false" show_stripe_button="true" weight="0.33" sideWeight="0.5" order="-1" side_tool="false" content_ui="tabs" />
534 534 <window_info id="Favorites" active="false" anchor="left" auto_hide="false" internal_type="DOCKED" type="DOCKED" visible="false" show_stripe_button="true" weight="0.33" sideWeight="0.5" order="-1" side_tool="true" content_ui="tabs" />
... ... @@ -542,6 +542,14 @@
542 542 <window_info id="Ant Build" active="false" anchor="right" auto_hide="false" internal_type="DOCKED" type="DOCKED" visible="false" show_stripe_button="true" weight="0.25" sideWeight="0.5" order="1" side_tool="false" content_ui="tabs" />
543 543 </layout>
544 544 </component>
  545 + <component name="Vcs.Log.UiProperties">
  546 + <option name="RECENTLY_FILTERED_USER_GROUPS">
  547 + <collection />
  548 + </option>
  549 + <option name="RECENTLY_FILTERED_BRANCH_GROUPS">
  550 + <collection />
  551 + </option>
  552 + </component>
545 553 <component name="VcsContentAnnotationSettings">
546 554 <option name="myLimit" value="2678400000" />
547 555 </component>
... ... @@ -566,14 +574,6 @@
566 574 </state>
567 575 </provider>
568 576 </entry>
569   - <entry file="file://$PROJECT_DIR$/modulevar.py">
570   - <provider selected="true" editor-type-id="text-editor">
571   - <state relative-caret-position="114">
572   - <caret line="6" column="21" selection-start-line="6" selection-start-column="21" selection-end-line="6" selection-end-column="21" />
573   - <folding />
574   - </state>
575   - </provider>
576   - </entry>
577 577 <entry file="file://$PROJECT_DIR$/scripts/digger.sh">
578 578 <provider selected="true" editor-type-id="text-editor">
579 579 <state relative-caret-position="133">
... ... @@ -582,25 +582,25 @@
582 582 </state>
583 583 </provider>
584 584 </entry>
585   - <entry file="file://$PROJECT_DIR$/scripts/list_ntfs.sh">
  585 + <entry file="file://$PROJECT_DIR$/runcmd.py">
586 586 <provider selected="true" editor-type-id="text-editor">
587   - <state relative-caret-position="38">
588   - <caret line="2" column="83" selection-start-line="2" selection-start-column="83" selection-end-line="2" selection-end-column="83" />
  587 + <state relative-caret-position="76">
  588 + <caret line="4" column="22" selection-start-line="4" selection-start-column="22" selection-end-line="4" selection-end-column="22" />
589 589 <folding />
590 590 </state>
591 591 </provider>
592 592 </entry>
593   - <entry file="file://$PROJECT_DIR$/runcmd.py">
  593 + <entry file="file://$PROJECT_DIR$/module.py">
594 594 <provider selected="true" editor-type-id="text-editor">
595   - <state relative-caret-position="76">
596   - <caret line="4" column="22" selection-start-line="4" selection-start-column="22" selection-end-line="4" selection-end-column="22" />
  595 + <state relative-caret-position="152">
  596 + <caret line="8" column="21" selection-start-line="8" selection-start-column="21" selection-end-line="8" selection-end-column="21" />
597 597 <folding />
598 598 </state>
599 599 </provider>
600 600 </entry>
601 601 <entry file="file://$PROJECT_DIR$/vars/ntfsvol.py">
602 602 <provider selected="true" editor-type-id="text-editor">
603   - <state relative-caret-position="542">
  603 + <state relative-caret-position="608">
604 604 <caret line="32" column="29" selection-start-line="32" selection-start-column="29" selection-end-line="32" selection-end-column="29" />
605 605 <folding>
606 606 <element signature="e#0#31#0" expanded="true" />
... ... @@ -608,41 +608,69 @@
608 608 </state>
609 609 </provider>
610 610 </entry>
611   - <entry file="file://$PROJECT_DIR$/module.py">
  611 + <entry file="file://$PROJECT_DIR$/scripts/list_ntfs.sh">
612 612 <provider selected="true" editor-type-id="text-editor">
613   - <state relative-caret-position="304">
614   - <caret line="16" column="58" selection-start-line="16" selection-start-column="58" selection-end-line="16" selection-end-column="58" />
  613 + <state relative-caret-position="38">
  614 + <caret line="2" column="83" selection-start-line="2" selection-start-column="83" selection-end-line="2" selection-end-column="83" />
615 615 <folding />
616 616 </state>
617 617 </provider>
618 618 </entry>
619   - <entry file="file://$PROJECT_DIR$/digger.py">
  619 + <entry file="file://$PROJECT_DIR$/modules/info.py">
620 620 <provider selected="true" editor-type-id="text-editor">
621   - <state relative-caret-position="-190">
622   - <caret line="11" column="0" selection-start-line="11" selection-start-column="0" selection-end-line="11" selection-end-column="0" />
  621 + <state relative-caret-position="266">
  622 + <caret line="14" column="22" selection-start-line="14" selection-start-column="22" selection-end-line="14" selection-end-column="22" />
623 623 <folding>
624   - <element signature="e#25#38#0" expanded="true" />
  624 + <element signature="e#0#25#0" expanded="true" />
625 625 </folding>
626 626 </state>
627 627 </provider>
628 628 </entry>
629 629 <entry file="file://$PROJECT_DIR$/tomb.py">
630 630 <provider selected="true" editor-type-id="text-editor">
631   - <state relative-caret-position="190">
632   - <caret line="10" column="25" selection-start-line="10" selection-start-column="25" selection-end-line="10" selection-end-column="25" />
  631 + <state relative-caret-position="114">
  632 + <caret line="6" column="27" selection-start-line="6" selection-start-column="27" selection-end-line="6" selection-end-column="27" />
  633 + <folding />
  634 + </state>
  635 + </provider>
  636 + </entry>
  637 + <entry file="file://$PROJECT_DIR$/modulevar.py">
  638 + <provider selected="true" editor-type-id="text-editor">
  639 + <state relative-caret-position="114">
  640 + <caret line="6" column="21" selection-start-line="6" selection-start-column="21" selection-end-line="6" selection-end-column="21" />
633 641 <folding />
634 642 </state>
635 643 </provider>
636 644 </entry>
637 645 <entry file="file://$PROJECT_DIR$/modules/mft.py">
638 646 <provider selected="true" editor-type-id="text-editor">
639   - <state relative-caret-position="190">
640   - <caret line="10" column="23" selection-start-line="10" selection-start-column="23" selection-end-line="10" selection-end-column="23" />
  647 + <state relative-caret-position="0">
  648 + <caret line="0" column="0" selection-start-line="0" selection-start-column="0" selection-end-line="25" selection-end-column="0" />
641 649 <folding>
642 650 <element signature="e#0#25#0" expanded="true" />
643 651 </folding>
644 652 </state>
645 653 </provider>
646 654 </entry>
  655 + <entry file="file://$PROJECT_DIR$/modules/winreg.py">
  656 + <provider selected="true" editor-type-id="text-editor">
  657 + <state relative-caret-position="323">
  658 + <caret line="17" column="12" selection-start-line="17" selection-start-column="12" selection-end-line="18" selection-end-column="10" />
  659 + <folding>
  660 + <element signature="e#0#25#0" expanded="true" />
  661 + </folding>
  662 + </state>
  663 + </provider>
  664 + </entry>
  665 + <entry file="file://$PROJECT_DIR$/digger.py">
  666 + <provider selected="true" editor-type-id="text-editor">
  667 + <state relative-caret-position="446">
  668 + <caret line="75" column="13" selection-start-line="75" selection-start-column="13" selection-end-line="75" selection-end-column="13" />
  669 + <folding>
  670 + <element signature="e#25#38#0" expanded="true" />
  671 + </folding>
  672 + </state>
  673 + </provider>
  674 + </entry>
647 675 </component>
648 676 </project>
649 677 \ No newline at end of file
... ...
__pycache__/tomb.cpython-34.pyc
No preview for this file type
digger.py
... ... @@ -6,6 +6,7 @@ import glob
6 6 import os
7 7 import sys
8 8 import subprocess as sp
  9 +import tomb
9 10  
10 11 sys.path.append('modules')
11 12 sys.path.append('vars')
... ... @@ -27,8 +28,6 @@ def getModules():
27 28 sys.modules.pop(modname)
28 29 except Exception as e:
29 30 print("Exception raised while importing " + modname)
30   -
31   - choiceList.append(("all","Execute all modules",False))
32 31 return choiceList
33 32  
34 33 def prepareModule(moduleName):
... ... @@ -41,29 +40,43 @@ d = Dialog(dialog=&quot;dialog&quot;,autowidgetsize=True)
41 40 d.set_background_title("Gravedigger")
42 41  
43 42 moduleList = getModules()
44   -code, tags = d.checklist("Select modules to execute",
45   - choices=moduleList,
46   - title="Module selection")
  43 +code,value = d.inputbox("Input computer's name")
47 44 if code == d.OK:
48   - runlist = []
49   - try:
50   - if "All" in tags:
51   - for module in moduleList:
52   - runlist.append(prepareModule(module))
53   - else:
54   - for tag in tags:
55   - runlist.append(prepareModule(tag))
  45 + tomb.__MACHINE_NAME__ = value
  46 + d.set_background_title("Gravedigger - " + value)
  47 + code, tags = d.checklist("Select modules to execute",
  48 + choices=moduleList + [("all","Execute all modules",False)],
  49 + title="Module selection")
  50 + if code == d.OK:
  51 + runlist = []
  52 + try:
  53 + if "all" in tags:
  54 + for module in moduleList:
  55 + runlist.append(prepareModule(module[0]))
  56 + else:
  57 + for tag in tags:
  58 + runlist.append(prepareModule(tag))
56 59  
57   - sp.call('clear', shell=True)
  60 + sp.call('clear', shell=True)
58 61  
59   - for module in runlist:
60   - print("Running " + module.name + "...")
61   - try:
62   - module.run()
63   - except Exception as e:
64   - print("Exception raised while running " + module.name + ": " + str(e))
  62 + for module in runlist:
  63 + print("Running " + module.name + "...")
  64 + try:
  65 + module.run()
  66 + except Exception as e:
  67 + print("Exception raised while running " + module.name + ": " + str(e))
65 68  
66   - except Exception as e:
67   - print("Exception raised while preparing module: " + str(e))
  69 + code, tag = d.menu("All modules finished execution",choices=[("Poweroff","Shutdown the computer"),
  70 + ("Reboot","Reboot the computer"),
  71 + ("Restart","Run Gravedigger again")])
  72 + if(code == d.OK):
  73 + if tag == "Poweroff":
  74 + sp.call("poweroff",shell=True)
  75 + elif tag == "Reboot":
  76 + sp.call("reboot", shell=True)
  77 +
  78 +
  79 + except Exception as e:
  80 + print("Exception raised while preparing module: " + str(e))
68 81  
69 82  
... ...
modules/__pycache__/info.cpython-34.pyc 0 → 100644
No preview for this file type
modules/__pycache__/mft.cpython-34.pyc
No preview for this file type
modules/__pycache__/winreg.cpython-34.pyc 0 → 100644
No preview for this file type
modules/info.py 0 → 100644
  1 +from module import Module
  2 +from runcmd import runProcess
  3 +import tomb
  4 +import os
  5 +
  6 +def getInstance():
  7 + return INFOModule()
  8 +
  9 +class INFOModule(Module):
  10 +
  11 + def __init__(self):
  12 + self.name = "info"
  13 + self.description = "Extracts information about this machine"
  14 + self.requiredVars = []
  15 + self.vars = {}
  16 +
  17 + def run(self):
  18 + path = tomb.getPath() + self.name + "/"
  19 + os.mkdir(path)
  20 + #getinfo
... ...
modules/mft.py
... ... @@ -15,10 +15,11 @@ class MFTModule(Module):
15 15 self.vars = {}
16 16  
17 17 def run(self):
18   - os.mkdir(tomb.getPath() + "mft/")
  18 + path = tomb.getPath() + self.name + "/"
  19 + os.mkdir(path)
19 20 for vol in self.vars['ntfsvol'].value:
20 21 result = runProcess(["icat","/dev/" + vol,"0"])
21   - mftbin = open(tomb.getPath() + "mft/" + vol + ".bin",'wb')
  22 + mftbin = open(path + vol + ".bin",'wb')
22 23 mftbin.write(result[0])
23 24 mftbin.close()
24 25  
... ...
modules/winreg.py 0 → 100644
  1 +from module import Module
  2 +from runcmd import runProcess
  3 +import tomb
  4 +import os
  5 +
  6 +def getInstance():
  7 + return RegistryModule()
  8 +
  9 +class RegistryModule(Module):
  10 +
  11 + def __init__(self):
  12 + self.name = "winreg"
  13 + self.description = "Extracts Windows Registry files"
  14 + self.requiredVars = ["ntfsvol"]
  15 + self.vars = {}
  16 +
  17 + def run(self):
  18 + path = tomb.getPath() + self.name + "/"
  19 + os.mkdir(path)
  20 + for vol in self.vars['ntfsvol'].value:
  21 + pass
  22 + #runstuff
  23 +
... ...
1 1 import os
2 2  
3 3 __ROOT_PATH__ = "/bones/"
  4 +__MACHINE_NAME__ = "JOHN_DOE"
4 5  
5 6 def __getTomb__():
6   - return "test"
  7 + return __MACHINE_NAME__
7 8  
8 9 def getPath():
9 10 path = __ROOT_PATH__ + __getTomb__() + "/"
... ...