88 lines
2.3 KiB
Python
88 lines
2.3 KiB
Python
|
#!/usr/bin/env python3
|
||
|
|
|
||
|
|
from Crypto.Hash import SHA, SHA256
|
||
|
|
from Crypto.Signature import PKCS1_v1_5
|
||
|
|
from Crypto.PublicKey import RSA
|
||
|
|
from shutil import copyfile
|
||
|
|
import json
|
||
|
|
import codecs
|
||
|
|
|
||
|
|
|
||
|
|
|
||
|
|
def gen_keys():
|
||
|
|
privKey = RSA.generate(1024)
|
||
|
|
publKey = privKey.publickey().exportKey('PEM')
|
||
|
|
|
||
|
|
f = open("/var/www/onlyoffice/license_key.pub", "w+")
|
||
|
|
f.write(publKey.decode('utf-8'))
|
||
|
|
f.close()
|
||
|
|
|
||
|
|
return publKey, privKey
|
||
|
|
|
||
|
|
|
||
|
|
|
||
|
|
def write_license(publKey, privKey):
|
||
|
|
license = {
|
||
|
|
"branding": False,
|
||
|
|
"connections": 9999,
|
||
|
|
"customization": False,
|
||
|
|
"end_date": "2099-01-01T23:59:59.000Z",
|
||
|
|
"light": "False",
|
||
|
|
"mode": "",
|
||
|
|
"portal_count": "0",
|
||
|
|
"process": 2,
|
||
|
|
"ssbranding": False,
|
||
|
|
"test": "False",
|
||
|
|
"trial": "False",
|
||
|
|
"user_quota": "0",
|
||
|
|
"users_count": 9999,
|
||
|
|
"users_expire": 99999,
|
||
|
|
"whiteLabel": False,
|
||
|
|
"customer_id": "customerID",
|
||
|
|
"start_date": "2020-01-01T00:00:00.000Z",
|
||
|
|
"users": [],
|
||
|
|
"version": 2
|
||
|
|
}
|
||
|
|
|
||
|
|
jsonData = codecs.encode(json.dumps(license, separators=(',', ':')), encoding='utf-8')
|
||
|
|
|
||
|
|
digest = SHA.new(jsonData)
|
||
|
|
signer = PKCS1_v1_5.new(privKey)
|
||
|
|
signature = signer.sign(digest)
|
||
|
|
finalSignature = signature.hex()
|
||
|
|
|
||
|
|
license['signature'] = finalSignature
|
||
|
|
|
||
|
|
f = open("/var/www/onlyoffice/license.lic", "w+")
|
||
|
|
f.write(json.dumps(license))
|
||
|
|
f.close
|
||
|
|
|
||
|
|
|
||
|
|
|
||
|
|
def patch_files():
|
||
|
|
basePath = "/var/www/onlyoffice/documentserver/server/"
|
||
|
|
files = ["DocService/docservice", "FileConverter/converter"]
|
||
|
|
|
||
|
|
for file in files:
|
||
|
|
f = open(basePath + file, 'rb')
|
||
|
|
data = f.read()
|
||
|
|
f.close()
|
||
|
|
|
||
|
|
replacedData = data.replace(b"-----BEGIN PUBLIC KEY-----\nMIGfMA0GCSqGSIb3DQEBAQUAA4GNADCBiQKBgQDRhGF7X4A0ZVlEg594WmODVVUI\niiPQs04aLmvfg8SborHss5gQXu0aIdUT6nb5rTh5hD2yfpF2WIW6M8z0WxRhwicg\nXwi80H1aLPf6lEPPLvN29EhQNjBpkFkAJUbS8uuhJEeKw0cE49g80eBBF4BCqSL6\nPFQbP9/rByxdxEoAIQIDAQAB\n-----END PUBLIC KEY-----", bytes(publKey))
|
||
|
|
|
||
|
|
f = open(basePath + file, 'wb')
|
||
|
|
f.write(replacedData)
|
||
|
|
f.close()
|
||
|
|
|
||
|
|
|
||
|
|
|
||
|
|
|
||
|
|
print("Generating and exporting key pair...")
|
||
|
|
publKey, privKey = gen_keys()
|
||
|
|
|
||
|
|
print("Writing license file...")
|
||
|
|
write_license(publKey, privKey)
|
||
|
|
|
||
|
|
print("Patching document server and converter...")
|
||
|
|
patch_files()
|