Add new license APIs.

Author: fuzyll

binaryninjaapi.cpp

@@ -272,6 +272,36 @@ string BinaryNinja::GetSerialNumber()
 }
 
 
+string BinaryNinja::GetLicenseUserId()
+{
+	char* str = BNGetLicenseUserId();
+	string result = str;
+	BNFreeString(str);
+	return result;
+}
+
+
+string BinaryNinja::GetLicenseAddonsJson()
+{
+	char* str = BNGetLicenseAddonsJson();
+	string result = str;
+	BNFreeString(str);
+	return result;
+}
+
+
+vector<string> BinaryNinja::GetLicenseAddons()
+{
+	size_t count = 0;
+	char** addons = BNGetLicenseAddons(&count);
+	vector<string> result;
+	for (size_t i = 0; i < count; i++)
+		result.emplace_back(addons[i]);
+	BNFreeStringList(addons, count);
+	return result;
+}
+
+
 int BinaryNinja::GetLicenseCount()
 {
 	return BNGetLicenseCount();

binaryninjaapi.h

@@ -2027,6 +2027,9 @@ namespace BinaryNinja {
 	std::string GetProduct();
 	std::string GetProductType();
 	std::string GetSerialNumber();
+	std::string GetLicenseUserId();
+	std::string GetLicenseAddonsJson();
+	std::vector<std::string> GetLicenseAddons();
 	int GetLicenseCount();
 	bool IsUIEnabled();
 	uint32_t GetBuildId();

binaryninjacore.h

@@ -4072,6 +4072,9 @@ extern "C"
 	BINARYNINJACOREAPI char* BNGetLicensedUserEmail(void);
 	BINARYNINJACOREAPI char* BNGetProduct(void);
 	BINARYNINJACOREAPI char* BNGetProductType(void);
+	BINARYNINJACOREAPI char* BNGetLicenseUserId(void);
+	BINARYNINJACOREAPI char* BNGetLicenseAddonsJson(void);
+	BINARYNINJACOREAPI char** BNGetLicenseAddons(size_t* count);
 	BINARYNINJACOREAPI int BNGetLicenseCount(void);
 	BINARYNINJACOREAPI bool BNIsUIEnabled(void);
 	BINARYNINJACOREAPI void BNSetLicense(const char* licenseData);

python/__init__.py

@@ -21,9 +21,10 @@
 import atexit
 import sys
 import ctypes
+import json
 from time import gmtime, struct_time
 import os
-from typing import Mapping, Optional
+from typing import Any, List, Mapping, Optional
 import functools
 
 # Binary Ninja components
@@ -373,6 +374,36 @@ def core_license_count() -> int:
 	return core.BNGetLicenseCount()
 
 
+def core_license_addons() -> List[str]:
+	'''License addons from the license file'''
+	_init_plugins()
+	count = ctypes.c_ulonglong()
+	addons = core.BNGetLicenseAddons(ctypes.byref(count))
+	if not addons:
+		return []
+	result = [core.pyNativeStr(addons[i]) for i in range(count.value)]
+	core.BNFreeStringList(addons, count.value)
+	return result
+
+
+def core_license_addons_json() -> str:
+	'''Structured license addons JSON from the license file'''
+	_init_plugins()
+	return core.BNGetLicenseAddonsJson()
+
+
+def core_license_addon_data() -> List[Mapping[str, Any]]:
+	'''Structured license addons from the license file'''
+	return json.loads(core_license_addons_json())
+
+
+def core_license_user_id() -> Optional[int]:
+	'''License user id from the license file'''
+	_init_plugins()
+	uid = core.BNGetLicenseUserId()
+	return int(uid) if uid else None
+
+
 def core_ui_enabled() -> bool:
 	'''Indicates that a UI exists and the UI has invoked BNInitUI'''
 	return core.BNIsUIEnabled()

rust/src/lib.rs

@@ -540,6 +540,34 @@ pub fn license_count() -> i32 {
     unsafe { BNGetLicenseCount() }
 }
 
+pub fn license_addons() -> Vec<String> {
+    let mut count = 0;
+    let addons = unsafe { BNGetLicenseAddons(&mut count) };
+    if addons.is_null() {
+        return Vec::new();
+    }
+
+    let result = unsafe { std::slice::from_raw_parts(addons, count) }
+        .iter()
+        .map(|addon| unsafe { CStr::from_ptr(*addon).to_string_lossy().into_owned() })
+        .collect();
+    unsafe { BNFreeStringList(addons, count) };
+    result
+}
+
+pub fn license_addons_json() -> String {
+    unsafe { BnString::into_string(BNGetLicenseAddonsJson()) }
+}
+
+pub fn license_user_id() -> Option<String> {
+    let uid = unsafe { BnString::into_string(BNGetLicenseUserId()) };
+    if uid.is_empty() {
+        None
+    } else {
+        Some(uid)
+    }
+}
+
 /// Set the license that will be used once the core initializes. You can reset the license by passing `None`.
 ///
 /// If not set, the normal license retrieval will occur: