mirror of
https://github.com/SystemRage/py-kms.git
synced 2024-11-10 17:49:37 +08:00
229 lines
12 KiB
Python
229 lines
12 KiB
Python
#!/usr/bin/env python3
|
|
|
|
import re
|
|
import argparse
|
|
import binascii
|
|
import datetime
|
|
import random
|
|
import socket
|
|
import string
|
|
import sys
|
|
import uuid
|
|
import logging
|
|
import os
|
|
import errno
|
|
|
|
import filetimes, rpcBind, rpcRequest
|
|
from dcerpc import MSRPCHeader, MSRPCBindNak, MSRPCRequestHeader, MSRPCRespHeader
|
|
from kmsBase import kmsBase, UUID
|
|
from kmsRequestV4 import kmsRequestV4
|
|
from kmsRequestV5 import kmsRequestV5
|
|
from kmsRequestV6 import kmsRequestV6
|
|
from rpcBase import rpcBase
|
|
from kmsDB2Dict import kmsDB2Dict
|
|
from formatText import shell_message, justify, byterize
|
|
|
|
config = {}
|
|
|
|
def main():
|
|
parser = argparse.ArgumentParser()
|
|
parser.add_argument("ip", action="store", help='The IP address or hostname of the KMS server.', type=str)
|
|
parser.add_argument("port", nargs="?", action="store", default=1688,
|
|
help='The port the KMS service is listening on. The default is \"1688\".', type=int)
|
|
parser.add_argument("-m", "--mode", dest="mode",
|
|
choices=["WindowsVista","Windows7","Windows8","Windows8.1","Windows10",
|
|
"Office2010","Office2013","Office2016","Office2019"], default="Windows8.1",
|
|
help='Use this flag to manually specify a Microsoft product for testing the server. The default is \"Windows81\".', type=str)
|
|
parser.add_argument("-c", "--cmid", dest="cmid", default=None,
|
|
help='Use this flag to manually specify a CMID to use. If no CMID is specified, a random CMID will be generated.', type=str)
|
|
parser.add_argument("-n", "--name", dest="machineName", default=None,
|
|
help='Use this flag to manually specify an ASCII machineName to use. If no machineName is specified,\
|
|
a random machineName will be generated.', type=str)
|
|
parser.add_argument("-v", "--loglevel", dest="loglevel", action="store", default="ERROR", choices=["CRITICAL", "ERROR", "WARNING", "INFO", "DEBUG"],
|
|
help='Use this flag to set a Loglevel. The default is \"ERROR\".', type=str)
|
|
parser.add_argument("-f", "--logfile", dest="logfile", action="store", default=os.path.dirname(os.path.abspath( __file__ )) + "/py3kms_client.log",
|
|
help='Use this flag to set an output Logfile. The default is \"pykms_client.log\".', type=str)
|
|
|
|
config.update(vars(parser.parse_args()))
|
|
|
|
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')
|
|
|
|
checkConfig()
|
|
config['call_id'] = 1
|
|
updateConfig()
|
|
s = socket.socket()
|
|
logging.info("Connecting to %s on port %d..." % (config['ip'], config['port']))
|
|
s.connect((config['ip'], config['port']))
|
|
logging.info("Connection successful !")
|
|
binder = rpcBind.handler(None, config)
|
|
RPC_Bind = str(binder.generateRequest()).encode('latin-1')
|
|
logging.info("Sending RPC bind request...")
|
|
shell_message(nshell = [-1, 1])
|
|
s.send(RPC_Bind)
|
|
try:
|
|
shell_message(nshell = [-4, 7])
|
|
bindResponse = s.recv(1024)
|
|
except socket.error as e:
|
|
if e.errno == errno.ECONNRESET:
|
|
logging.error("Connection reset by peer. Exiting...")
|
|
sys.exit()
|
|
else:
|
|
raise
|
|
if bindResponse == '' or not bindResponse:
|
|
logging.error("No data received ! Exiting...")
|
|
sys.exit()
|
|
packetType = MSRPCHeader(bindResponse)['type']
|
|
if packetType == rpcBase.packetType['bindAck']:
|
|
logging.info("RPC bind acknowledged.")
|
|
shell_message(nshell = 8)
|
|
kmsRequest = createKmsRequest()
|
|
requester = rpcRequest.handler(kmsRequest, config)
|
|
s.send(str(requester.generateRequest()).encode('latin-1'))
|
|
shell_message(nshell = [-1, 12])
|
|
response = s.recv(1024)
|
|
logging.debug("Response: \n%s\n" % justify(binascii.b2a_hex(response).decode('latin-1')))
|
|
shell_message(nshell = [-4, 20])
|
|
parsed = MSRPCRespHeader(response)
|
|
kmsData = readKmsResponse(parsed['pduData'], kmsRequest, config)
|
|
kmsResp = kmsData['response']
|
|
|
|
try:
|
|
hwid = kmsData['hwid']
|
|
logging.info("KMS Host HWID: %s" % binascii.b2a_hex(hwid.encode('latin-1')).upper().decode('utf-8'))
|
|
except KeyError:
|
|
pass
|
|
logging.info("KMS Host ePID: %s" % kmsResp['kmsEpid'].encode('utf-8').decode('utf-16le'))
|
|
logging.info("KMS Host Current Client Count: %s" % kmsResp['currentClientCount'])
|
|
logging.info("KMS VL Activation Interval: %s" % kmsResp['vLActivationInterval'])
|
|
logging.info("KMS VL Renewal Interval: %s" % kmsResp['vLRenewalInterval'])
|
|
shell_message(nshell = 21)
|
|
|
|
elif packetType == rpcBase.packetType['bindNak']:
|
|
logging.info(justify(MSRPCBindNak(bindResponse).dump(print_to_stdout = False)))
|
|
sys.exit()
|
|
else:
|
|
logging.critical("Something went wrong.")
|
|
sys.exit()
|
|
|
|
|
|
def checkConfig():
|
|
if config['cmid'] is not None:
|
|
try:
|
|
uuid.UUID(config['cmid'])
|
|
except ValueError:
|
|
logging.error("Bad CMID. Exiting...")
|
|
sys.exit()
|
|
if config['machineName'] is not None:
|
|
if len(config['machineName']) < 2 or len(config['machineName']) > 63:
|
|
logging.error("Error: machineName must be between 2 and 63 characters in length.")
|
|
sys.exit()
|
|
|
|
def updateConfig():
|
|
kmsdb = kmsDB2Dict()
|
|
|
|
appitems = kmsdb[2]
|
|
for appitem in appitems:
|
|
kmsitems = appitem['KmsItems']
|
|
for kmsitem in kmsitems:
|
|
# Threshold.
|
|
try:
|
|
count = int(kmsitem['NCountPolicy'])
|
|
except KeyError:
|
|
count = 25
|
|
|
|
name = re.sub('\(.*\)', '', kmsitem['DisplayName']).replace('2015', '').replace(' ', '')
|
|
if name == config['mode']:
|
|
skuitems = kmsitem['SkuItems']
|
|
# Select 'Enterprise' for Windows or 'Professional Plus' for Office.
|
|
# (improvement: choice could be also random: skuitem = random.choice(skuitems))
|
|
for skuitem in skuitems:
|
|
if skuitem['DisplayName'].replace(' ','') == name + 'Enterprise' or \
|
|
skuitem['DisplayName'].replace(' ','') == name[:6] + 'ProfessionalPlus' + name[6:]:
|
|
|
|
config['KMSClientSkuID'] = skuitem['Id']
|
|
config['RequiredClientCount'] = count
|
|
config['KMSProtocolMajorVersion'] = int(float(kmsitem['DefaultKmsProtocol']))
|
|
config['KMSProtocolMinorVersion'] = 0
|
|
config['KMSClientLicenseStatus'] = 2
|
|
config['KMSClientAppID'] = appitem['Id']
|
|
config['KMSClientKMSCountedID'] = kmsitem['Id']
|
|
break
|
|
|
|
def createKmsRequestBase():
|
|
requestDict = kmsBase.kmsRequestStruct()
|
|
requestDict['versionMinor'] = config['KMSProtocolMinorVersion']
|
|
requestDict['versionMajor'] = config['KMSProtocolMajorVersion']
|
|
requestDict['isClientVm'] = 0
|
|
requestDict['licenseStatus'] = config['KMSClientLicenseStatus']
|
|
requestDict['graceTime'] = 43200
|
|
requestDict['applicationId'] = UUID(uuid.UUID(config['KMSClientAppID']).bytes_le)
|
|
requestDict['skuId'] = UUID(uuid.UUID(config['KMSClientSkuID']).bytes_le)
|
|
requestDict['kmsCountedId'] = UUID(uuid.UUID(config['KMSClientKMSCountedID']).bytes_le)
|
|
requestDict['clientMachineId'] = UUID(uuid.UUID(config['cmid']).bytes_le if (config['cmid'] is not None) else uuid.uuid4().bytes_le)
|
|
requestDict['previousClientMachineId'] = '\0' * 16 #requestDict['clientMachineId'] # I'm pretty sure this is supposed to be a null UUID.
|
|
requestDict['requiredClientCount'] = config['RequiredClientCount']
|
|
requestDict['requestTime'] = filetimes.dt_to_filetime(datetime.datetime.utcnow())
|
|
requestDict['machineName'] = (config['machineName'] if (config['machineName'] is not None) else
|
|
''.join(random.choice(string.ascii_letters + string.digits) for i in range(random.randint(2,63)))).encode('utf-16le')
|
|
requestDict['mnPad'] = '\0'.encode('utf-16le') * (63 - len(requestDict['machineName'].decode('utf-16le')))
|
|
|
|
# Debug Stuff
|
|
shell_message(nshell = 9)
|
|
requestDict = byterize(requestDict)
|
|
logging.debug("Request Base Dictionary: \n%s\n" % justify(requestDict.dump(print_to_stdout = False)))
|
|
|
|
return requestDict
|
|
|
|
def createKmsRequest():
|
|
# Update the call ID
|
|
config['call_id'] += 1
|
|
|
|
# KMS Protocol Major Version
|
|
if config['KMSProtocolMajorVersion'] == 4:
|
|
handler = kmsRequestV4(None, config)
|
|
elif config['KMSProtocolMajorVersion'] == 5:
|
|
handler = kmsRequestV5(None, config)
|
|
elif config['KMSProtocolMajorVersion'] == 6:
|
|
handler = kmsRequestV6(None, config)
|
|
else:
|
|
return None
|
|
|
|
requestBase = createKmsRequestBase()
|
|
return handler.generateRequest(requestBase)
|
|
|
|
def readKmsResponse(data, request, config):
|
|
if config['KMSProtocolMajorVersion'] == 4:
|
|
logging.info("Received V4 response")
|
|
response = readKmsResponseV4(data, request)
|
|
elif config['KMSProtocolMajorVersion'] == 5:
|
|
logging.info("Received V5 response")
|
|
response = readKmsResponseV5(data)
|
|
elif config['KMSProtocolMajorVersion'] == 6:
|
|
logging.info("Received V6 response")
|
|
response = readKmsResponseV6(data)
|
|
else:
|
|
logging.info("Unhandled response version: %d.%d" % (config['KMSProtocolMajorVersion'], config['KMSProtocolMinorVersion']))
|
|
logging.info("I'm not even sure how this happened...")
|
|
return response
|
|
|
|
def readKmsResponseV4(data, request):
|
|
response = kmsRequestV4.ResponseV4(data)
|
|
hashed = kmsRequestV4(data, config).generateHash(bytearray(str(response['response']).encode('latin-1')))
|
|
if hashed.decode('latin-1') == response['hash']:
|
|
logging.info("Response Hash has expected value !")
|
|
return response
|
|
|
|
def readKmsResponseV5(data):
|
|
response = kmsRequestV5.ResponseV5(data)
|
|
decrypted = kmsRequestV5(data, config).decryptResponse(response)
|
|
return decrypted
|
|
|
|
def readKmsResponseV6(data):
|
|
response = kmsRequestV6.ResponseV5(data)
|
|
decrypted = kmsRequestV6(data, config).decryptResponse(response)
|
|
message = decrypted['message']
|
|
return message
|
|
|
|
if __name__ == "__main__":
|
|
main()
|