Added WAKE UP! payload (#30)

payloads/library/general/Wake-On-Lan/wol.py

@@ -0,0 +1,89 @@
+import socket
+import struct
+import os
+import sys
+from subprocess import Popen, PIPE
+import fcntl
+
+#### OPTIONS
+interface = str(sys.argv[1])
+SINGLETARGET = int(sys.argv[2])
+       ##########  if SINGLETARGET is 1  WAKETARGET is used
+WAKETARGET = str(sys.argv[3])
+       ########## if SINGLETARGET is 0  ranges are used
+startrange = int(sys.argv[4])
+endrange = int(sys.argv[5])
+####
+
+def wake_on_lan(host, broad):
+    if host == '00:00:00:00:00:00':
+      return False
+    try:
+      macaddress = host
+    except:
+      return False
+    if len(macaddress) == 12:
+        pass
+    elif len(macaddress) == 12 + 5:
+        sep = macaddress[2]
+        macaddress = macaddress.replace(sep, '')
+    else:
+        raise ValueError('Incorrect MAC address format')
+    data = ''.join(['FFFFFFFFFFFF', macaddress * 20])
+    send_data = b''
+    for i in range(0, len(data), 2):
+        send_data = b''.join([send_data,
+                             struct.pack('B', int(data[i: i + 2], 16))])
+    sock = socket.socket(socket.AF_INET, socket.SOCK_DGRAM)
+    sock.setsockopt(socket.SOL_SOCKET, socket.SO_BROADCAST, 1)
+    sock.sendto(send_data, (broad,9))
+    print('sent to '+host)
+    return True
+
+def get_mac(IP):
+    try:
+        Popen(["ping", "-c1", IP], stdout = PIPE)
+        pid = Popen(["cat", "/proc/net/arp"], stdout = PIPE )
+        mac = str(pid.communicate()[0]).split()
+        mac = mac[int(mac.index(IP)+3)]
+    except:
+        pass
+    return mac
+
+def get_ip_address(ifname):
+    s = socket.socket(socket.AF_INET, socket.SOCK_DGRAM)
+    return socket.inet_ntoa(fcntl.ioctl(
+        s.fileno(),
+        0x8915,
+        struct.pack('256s', ifname)
+    )[20:24])
+
+ip = str(get_ip_address(interface))
+submask = socket.inet_ntoa(struct.pack(">L", (1<<32) - (1<<32>>24)))
+addr = ip.split('.')
+cidr = int(sum([bin(int(x)).count('1') for x in submask.split('.')]))
+mask = submask.split('.')
+net = []
+for i in range(4):
+        net.append(int(addr[i]) & int(mask[i]))
+for i in range(int(32 - cidr)):
+        net[3 - i/8] = net[3 - i/8] + (1 << (i % 8))
+
+if SINGLETARGET == 0:
+    ip = ip.split('.')
+    exclude = str(ip[3])
+    del ip[3]
+    ip.append('x')
+    ip = ".".join(map(str, ip))
+    for num in range(startrange, endrange):
+        if str(num) != exclude:
+            wakeip = ip.replace('x', str(num))
+            try:
+                wake_on_lan(get_mac(str(wakeip)), str(".".join(map(str, net))))
+            except:
+                pass
+else:
+    try:
+        wake_on_lan(get_mac(str(WAKETARGET)), str(".".join(map(str, net))))
+    except:
+        pass