Implement update API (no UI yet)

2 files changed, 91 insertions, 0 deletions

diff --git a/binaryninjaapi.h b/binaryninjaapi.h
index 27e2c565..46d675f2 100644
--- a/binaryninjaapi.h
+++ b/binaryninjaapi.h
@@ -1355,6 +1355,14 @@ namespace BinaryNinja
 		std::string latestVersion;
 
 		static std::vector<UpdateChannel> GetList();
+
+		bool AreUpdatesAvailable();
+
+		BNUpdateResult UpdateToVersion(const std::string& version);
+		BNUpdateResult UpdateToVersion(const std::string& version,
+		                               const std::function<void(uint64_t progress, uint64_t total)>& progress);
+		BNUpdateResult UpdateToLatestVersion();
+		BNUpdateResult UpdateToLatestVersion(const std::function<void(uint64_t progress, uint64_t total)>& progress);
 	};
 
 	struct UpdateVersion
diff --git a/update.cpp b/update.cpp
index 545305b0..90e3477b 100644
--- a/update.cpp
+++ b/update.cpp
@@ -4,6 +4,21 @@ using namespace BinaryNinja;
 using namespace std;
 
 
+namespace BinaryNinja
+{
+	struct UpdateProgress
+	{
+		function<void(uint64_t progress, uint64_t total)> func;
+
+		static void UpdateCallback(void* ctxt, uint64_t progress, uint64_t total)
+		{
+			UpdateProgress* self = (UpdateProgress*)ctxt;
+			self->func(progress, total);
+		}
+	};
+}
+
+
 vector<UpdateChannel> UpdateChannel::GetList()
 {
 	size_t count;
@@ -32,6 +47,74 @@ vector<UpdateChannel> UpdateChannel::GetList()
 }
 
 
+bool UpdateChannel::AreUpdatesAvailable()
+{
+	char* errors;
+	bool result = BNAreUpdatesAvailable(name.c_str(), &errors);
+
+	if (errors)
+	{
+		string errorStr = errors;
+		BNFreeString(errors);
+		throw UpdateException(errorStr);
+	}
+
+	return result;
+}
+
+
+BNUpdateResult UpdateChannel::UpdateToVersion(const string& version)
+{
+	return UpdateToVersion(version, [](uint64_t, uint64_t){});
+}
+
+
+BNUpdateResult UpdateChannel::UpdateToVersion(const string& version,
+                                              const function<void(uint64_t progress, uint64_t total)>& progress)
+{
+	UpdateProgress up;
+	up.func = progress;
+
+	char* errors;
+	BNUpdateResult result = BNUpdateToVersion(name.c_str(), version.c_str(), &errors,
+	                                          UpdateProgress::UpdateCallback, &up);
+
+	if (errors)
+	{
+		string errorStr = errors;
+		BNFreeString(errors);
+		throw UpdateException(errorStr);
+	}
+
+	return result;
+}
+
+
+BNUpdateResult UpdateChannel::UpdateToLatestVersion()
+{
+	return UpdateToLatestVersion([](uint64_t, uint64_t){});
+}
+
+
+BNUpdateResult UpdateChannel::UpdateToLatestVersion(const function<void(uint64_t progress, uint64_t total)>& progress)
+{
+	UpdateProgress up;
+	up.func = progress;
+
+	char* errors;
+	BNUpdateResult result = BNUpdateToLatestVersion(name.c_str(), &errors, UpdateProgress::UpdateCallback, &up);
+
+	if (errors)
+	{
+		string errorStr = errors;
+		BNFreeString(errors);
+		throw UpdateException(errorStr);
+	}
+
+	return result;
+}
+
+
 vector<UpdateVersion> UpdateVersion::GetChannelVersions(const string& channel)
 {
 	size_t count;