mirror of
https://github.com/Py-KMS-Organization/py-kms.git
synced 2025-05-17 16:45:15 -04:00
uploaded version 2018_11_15 for python2
This commit is contained in:
Matteo ℱan
GitHub
parent
73f66ea40f
commit
b0a8ff21fb
18 changed files with 1574 additions and 791 deletions
|
|
@ -2,15 +2,14 @@
|
|||
|
||||
import argparse
|
||||
import binascii
|
||||
import hashlib
|
||||
import random
|
||||
import re
|
||||
import sys
|
||||
import socket
|
||||
import SocketServer
|
||||
import struct
|
||||
import uuid
|
||||
import logging
|
||||
import os
|
||||
import errno
|
||||
|
||||
import rpcBind, rpcRequest
|
||||
from dcerpc import MSRPCHeader
|
||||
|
|
@ -19,8 +18,30 @@ from formatText import shell_message
|
|||
|
||||
config = {}
|
||||
|
||||
# Valid language identifiers to be used in the EPID (see "kms.c" in vlmcsd)
|
||||
ValidLcid = [1025, 1026, 1027, 1028, 1029,
|
||||
1030, 1031, 1032, 1033, 1034, 1035, 1036, 1037, 1038, 1039,
|
||||
1040, 1041, 1042, 1043, 1044, 1045, 1046, 1048, 1049,
|
||||
1050, 1051, 1052, 1053, 1054, 1056, 1057, 1058, 1059,
|
||||
1060, 1061, 1062, 1063, 1065, 1066, 1067, 1068, 1069,
|
||||
1071, 1074, 1076, 1077, 1078, 1079,
|
||||
1080, 1081, 1082, 1083, 1086, 1087, 1088, 1089,
|
||||
1091, 1092, 1093, 1094, 1095, 1097, 1098, 1099,
|
||||
1100, 1102, 1103, 1104, 1106, 1110, 1111, 1114, 1125, 1131, 1153,
|
||||
2049, 2052, 2055, 2057, 2058, 2060, 2064, 2067, 2068, 2070, 2074, 2077, 2092, 2107, 2110, 2115, 2155,
|
||||
3073, 3076, 3079, 3081, 3082, 3084, 3098, 3131, 3179,
|
||||
4097, 4100, 4103, 4105, 4106, 4108, 4122, 4155,
|
||||
5121, 5124, 5127, 5129, 5130, 5132, 5146, 5179,
|
||||
6145, 6153, 6154, 6156, 6170, 6203,
|
||||
7169, 7177, 7178, 7194, 7227,
|
||||
8193, 8201, 8202, 8251,
|
||||
9217, 9225, 9226, 9275,
|
||||
10241, 10249, 10250, 11265, 11273, 11274, 12289, 12297, 12298,
|
||||
13313, 13321, 13322, 14337, 14346, 15361, 15370, 16385, 16394, 17418, 18442, 19466, 20490]
|
||||
|
||||
|
||||
def main():
|
||||
parser = argparse.ArgumentParser(description='py2-kms: KMS Server Emulator written in Python2', epilog="version: py2-kms_2018-03-01")
|
||||
parser = argparse.ArgumentParser(description='py2-kms: KMS Server Emulator written in Python2', epilog="version: py2-kms_2018-11-15")
|
||||
parser.add_argument("ip", nargs="?", action="store", default="0.0.0.0",
|
||||
help='The IP address to listen on. The default is \"0.0.0.0\" (all interfaces).', type=str)
|
||||
parser.add_argument("port", nargs="?", action="store", default=1688,
|
||||
|
|
@ -28,10 +49,10 @@ def main():
|
|||
parser.add_argument("-e", "--epid", dest="epid", default=None,
|
||||
help='Use this flag to manually specify an ePID to use. If no ePID is specified, a random ePID will be generated.', type=str)
|
||||
parser.add_argument("-l", "--lcid", dest="lcid", default=1033,
|
||||
help='Use this flag to manually specify an LCID for use with randomly generated ePIDs. If an ePID is manually specified,\
|
||||
this setting is ignored.', type=int)
|
||||
help='Use this flag to manually specify an LCID for use with randomly generated ePIDs. Default is 1033 (en-us)', type=int)
|
||||
parser.add_argument("-c", "--client-count", dest="CurrentClientCount", default=26,
|
||||
help='Use this flag to specify the current client count. Default is 26. A number >25 is required to enable activation.', type=int)
|
||||
help='Use this flag to specify the current client count. Default is 26. A number >=25 is required to enable \
|
||||
activation of client OSes; for server OSes and Office >=5', type=int)
|
||||
parser.add_argument("-a", "--activation-interval", dest="VLActivationInterval", default=120,
|
||||
help='Use this flag to specify the activation interval (in minutes). Default is 120 minutes (2 hours).', type=int)
|
||||
parser.add_argument("-r", "--renewal-interval", dest="VLRenewalInterval", default=1440 * 7,
|
||||
|
|
@ -47,76 +68,95 @@ The default is \"364F463A8863D35F\" or type \"random\" to auto generate the HWID
|
|||
help='Use this flag to set an output Logfile. The default is \"pykms_server.log\".', type=str)
|
||||
|
||||
config.update(vars(parser.parse_args()))
|
||||
#Random HWID
|
||||
if config['hwid'] == "random":
|
||||
randomhwid = uuid.uuid4().hex
|
||||
config['hwid'] = randomhwid[:16]
|
||||
|
||||
# Sanitize HWID
|
||||
try:
|
||||
config['hwid'] = binascii.a2b_hex(re.sub(r'[^0-9a-fA-F]', '', config['hwid'].strip('0x')))
|
||||
if len(binascii.b2a_hex(config['hwid'])) < 16:
|
||||
logging.error("HWID \"%s\" is invalid. Hex string is too short." % binascii.b2a_hex(config['hwid']).upper())
|
||||
return
|
||||
elif len(binascii.b2a_hex(config['hwid'])) > 16:
|
||||
logging.error("HWID \"%s\" is invalid. Hex string is too long." % binascii.b2a_hex(config['hwid']).upper())
|
||||
return
|
||||
except TypeError:
|
||||
logging.error("HWID \"%s\" is invalid. Odd-length hex string." % binascii.b2a_hex(config['hwid']).upper())
|
||||
return
|
||||
|
||||
logging.basicConfig(level=config['loglevel'], format='%(asctime)s %(levelname)-8s %(message)s',
|
||||
datefmt='%a, %d %b %Y %H:%M:%S', filename=config['logfile'], filemode='w')
|
||||
|
||||
# Random HWID.
|
||||
if config['hwid'] == "random":
|
||||
randomhwid = uuid.uuid4().hex
|
||||
config['hwid'] = randomhwid[:16]
|
||||
|
||||
# Sanitize HWID.
|
||||
try:
|
||||
import sqlite3
|
||||
config['dbSupport'] = True
|
||||
config['hwid'] = binascii.a2b_hex(re.sub(r'[^0-9a-fA-F]', '', config['hwid'].strip('0x')))
|
||||
if len(binascii.b2a_hex(config['hwid'])) < 16:
|
||||
logging.error("Error: HWID \"%s\" is invalid. Hex string is too short." % binascii.b2a_hex(config['hwid']).upper())
|
||||
return
|
||||
elif len(binascii.b2a_hex(config['hwid'])) > 16:
|
||||
logging.error("Error: HWID \"%s\" is invalid. Hex string is too long." % binascii.b2a_hex(config['hwid']).upper())
|
||||
return
|
||||
except TypeError:
|
||||
logging.error("Error: HWID \"%s\" is invalid. Odd-length hex string." % binascii.b2a_hex(config['hwid']).upper())
|
||||
return
|
||||
|
||||
# Check LCID.
|
||||
# http://stackoverflow.com/questions/3425294/how-to-detect-the-os-default-language-in-python
|
||||
if not config['lcid'] or (config['lcid'] not in ValidLcid):
|
||||
if hasattr(sys, 'implementation') and sys.implementation.name == 'cpython':
|
||||
config['lcid'] = 1033
|
||||
elif os.name == 'nt':
|
||||
import ctypes
|
||||
|
||||
config['lcid'] = ctypes.windll.kernel32.GetUserDefaultUILanguage() # TODO: or GetSystemDefaultUILanguage?
|
||||
else:
|
||||
import locale
|
||||
|
||||
try:
|
||||
config['lcid'] = next(k for k, v in locale.windows_locale.items() if v == locale.getdefaultlocale()[0])
|
||||
except StopIteration:
|
||||
config['lcid'] = 1033
|
||||
|
||||
try:
|
||||
import sqlite3
|
||||
except:
|
||||
logging.warning("Module \"sqlite3\" is not installed, database support disabled.")
|
||||
config['dbSupport'] = False
|
||||
else:
|
||||
config['dbSupport'] = True
|
||||
|
||||
server = SocketServer.TCPServer((config['ip'], config['port']), kmsServer)
|
||||
server.timeout = 5
|
||||
logging.info("TCP server listening at %s on port %d." % (config['ip'], config['port']))
|
||||
logging.info("HWID: %s" % binascii.b2a_hex(config['hwid']).upper())
|
||||
server.serve_forever()
|
||||
|
||||
|
||||
class kmsServer(SocketServer.BaseRequestHandler):
|
||||
def setup(self):
|
||||
self.connection = self.request
|
||||
logging.info("Connection accepted: %s:%d" % (self.client_address[0], self.client_address[1]))
|
||||
|
||||
def handle(self):
|
||||
while True:
|
||||
# self.request is the TCP socket connected to the client
|
||||
try:
|
||||
self.data = self.connection.recv(1024)
|
||||
data = self.request.recv(1024)
|
||||
except socket.error, e:
|
||||
if e[0] == 104:
|
||||
if e.errno == errno.ECONNRESET:
|
||||
logging.error("Connection reset by peer.")
|
||||
break
|
||||
else:
|
||||
raise
|
||||
if self.data == '' or not self.data:
|
||||
if not data:
|
||||
logging.warning("No data received !")
|
||||
break
|
||||
# self.data = bytearray(self.data.strip())
|
||||
# logging.debug(binascii.b2a_hex(str(self.data)))
|
||||
packetType = MSRPCHeader(self.data)['type']
|
||||
# data = bytearray(self.data.strip())
|
||||
# logging.debug(binascii.b2a_hex(str(data)))
|
||||
packetType = MSRPCHeader(data)['type']
|
||||
if packetType == rpcBase.packetType['bindReq']:
|
||||
logging.info("RPC bind request received.")
|
||||
shell_message(nshell = [-2, 2])
|
||||
handler = rpcBind.handler(self.data, config)
|
||||
handler = rpcBind.handler(data, config)
|
||||
elif packetType == rpcBase.packetType['request']:
|
||||
logging.info("Received activation request.")
|
||||
shell_message(nshell = [-2, 13])
|
||||
handler = rpcRequest.handler(self.data, config)
|
||||
handler = rpcRequest.handler(data, config)
|
||||
else:
|
||||
logging.error("Invalid RPC request type ", packetType)
|
||||
logging.error("Error: Invalid RPC request type ", packetType)
|
||||
break
|
||||
|
||||
handler.populate()
|
||||
res = str(handler.getResponse())
|
||||
self.connection.send(res)
|
||||
res = str(handler.populate())
|
||||
self.request.send(res)
|
||||
|
||||
if packetType == rpcBase.packetType['bindReq']:
|
||||
logging.info("RPC bind acknowledged.")
|
||||
|
|
@ -127,7 +167,7 @@ class kmsServer(SocketServer.BaseRequestHandler):
|
|||
break
|
||||
|
||||
def finish(self):
|
||||
self.connection.close()
|
||||
self.request.close()
|
||||
logging.info("Connection closed: %s:%d" % (self.client_address[0], self.client_address[1]))
|
||||
|
||||
if __name__ == "__main__":
|
||||
|
|
|
|||
Loading…
Add table
Add a link
Reference in a new issue