156 lines
4.2 KiB
C++
156 lines
4.2 KiB
C++
#include "HGUpgrade.h"
|
|
#include "base/HGDef.h"
|
|
#include "base/HGInc.h"
|
|
#include "base/HGUtility.h"
|
|
#include "base/HGMd5.h"
|
|
#include "curl/curl.h"
|
|
#include <iostream>
|
|
#include <sstream>
|
|
#include <HGString.h>
|
|
|
|
static std::string GetCurrVersion()
|
|
{
|
|
std::string version = "0.0.0.0";
|
|
|
|
#if defined(HG_CMP_MSC)
|
|
HKEY hKey = NULL;
|
|
RegOpenKeyExA(HKEY_LOCAL_MACHINE, "SOFTWARE\\HuaGoScan", 0, KEY_QUERY_VALUE, &hKey);
|
|
if (NULL != hKey)
|
|
{
|
|
CHAR szData[MAX_PATH] = { 0 };
|
|
DWORD cbData = MAX_PATH;
|
|
if (ERROR_SUCCESS == RegQueryValueExA(hKey, "AppVersion", NULL, NULL, (LPBYTE)szData, &cbData))
|
|
{
|
|
version = szData;
|
|
}
|
|
|
|
RegCloseKey(hKey);
|
|
}
|
|
#else
|
|
// TODO
|
|
#endif
|
|
|
|
return version;
|
|
}
|
|
|
|
static size_t write_data(void *ptr, size_t size, size_t nmemb, void *stream)
|
|
{
|
|
std::string data((const char*) ptr, (size_t) size * nmemb);
|
|
|
|
*((std::stringstream*) stream) << data << std::endl;
|
|
|
|
return size * nmemb;
|
|
}
|
|
|
|
// type:1表示安装 2表示卸载 src:来源 desc: 说明
|
|
static bool PostInfo(int type, const std::string &src, const std::string &desc)
|
|
{
|
|
curl_global_init(CURL_GLOBAL_ALL);
|
|
|
|
bool ret = false;
|
|
CURL* curl = curl_easy_init();
|
|
if (curl)
|
|
{
|
|
std::stringstream out;
|
|
|
|
std::string url = "http://up.appqy.com/api/recode";
|
|
curl_easy_setopt(curl, CURLOPT_CONNECTTIMEOUT, 2);
|
|
curl_easy_setopt(curl, CURLOPT_URL, url.c_str());
|
|
|
|
HGByte mac[256];
|
|
HGUInt macCount = 0;
|
|
HGBase_GetMACAddress(mac, 256, &macCount);
|
|
char macStr[64] = {0};
|
|
sprintf(macStr, "%02x-%02x-%02x-%02x-%02x-%02x", mac[0], mac[1], mac[2], mac[3], mac[4], mac[5]);
|
|
|
|
HGByte md5[16] = {0};
|
|
HGBase_MakeMd5(mac, macCount * 6, md5);
|
|
char md5Str[64] = {0};
|
|
char *pstr = md5Str;
|
|
for (int i = 0; i < 16; ++i)
|
|
{
|
|
sprintf(pstr, "%02x", md5[i]);
|
|
pstr += 2;
|
|
}
|
|
|
|
std::string version = GetCurrVersion();
|
|
|
|
char json[1024];
|
|
sprintf(json, "{\"type\":%d, \"mac\":\"%s\", \"localid\":\"%s\", \"v\":\"%s\", \"ref\":\"%s\", \"desc\":\"%s\"}",
|
|
type, macStr, md5Str, version.c_str(), src.c_str(), desc.c_str());
|
|
|
|
struct curl_slist* headers = nullptr;
|
|
headers = curl_slist_append(headers, "Content-Type:application/json;charset=UTF-8");
|
|
curl_easy_setopt(curl, CURLOPT_HTTPHEADER, headers);
|
|
curl_easy_setopt(curl, CURLOPT_POST, 1);
|
|
curl_easy_setopt(curl, CURLOPT_POSTFIELDS, json);
|
|
|
|
curl_easy_setopt(curl, CURLOPT_WRITEFUNCTION, write_data);
|
|
curl_easy_setopt(curl, CURLOPT_WRITEDATA, &out);
|
|
|
|
/* Perform the request, res will get the return code */
|
|
CURLcode res = curl_easy_perform(curl);
|
|
/* Check for errors */
|
|
if (res != CURLE_OK)
|
|
fprintf(stderr, "curl_easy_perform() failed: %s", curl_easy_strerror(res));
|
|
else
|
|
ret = true;
|
|
|
|
curl_slist_free_all(headers);
|
|
|
|
std::string str_json = out.str(); // 返回值
|
|
printf("%s\n", str_json.c_str());
|
|
|
|
/* always cleanup */
|
|
curl_easy_cleanup(curl);
|
|
}
|
|
|
|
curl_global_cleanup();
|
|
return ret;
|
|
}
|
|
|
|
bool PostInstallInfo(const std::string &src, const std::string &desc)
|
|
{
|
|
return PostInfo(1, src, desc);
|
|
}
|
|
|
|
bool PostUninstallInfo(const std::string &src, const std::string &desc)
|
|
{
|
|
return PostInfo(2, src, desc);
|
|
}
|
|
|
|
static bool Setup(const std::string& pkgPath)
|
|
{
|
|
bool ret = false;
|
|
#if defined(HG_CMP_MSC)
|
|
|
|
PROCESS_INFORMATION ProcessInfo;
|
|
STARTUPINFOA StartupInfo;
|
|
ZeroMemory(&StartupInfo, sizeof(StartupInfo));
|
|
StartupInfo.cb = sizeof(StartupInfo);
|
|
|
|
char command[256];
|
|
sprintf(command, "%s %s", Utf8ToStdString(pkgPath).c_str(), "/verysilent");
|
|
if (CreateProcessA(NULL, command, NULL, NULL, FALSE, 0, NULL, NULL, &StartupInfo, &ProcessInfo))
|
|
{
|
|
WaitForSingleObject(ProcessInfo.hProcess, INFINITE);
|
|
CloseHandle(ProcessInfo.hThread);
|
|
CloseHandle(ProcessInfo.hProcess);
|
|
ret = true;
|
|
}
|
|
|
|
#else
|
|
// TODO
|
|
#endif
|
|
|
|
return ret;
|
|
}
|
|
|
|
bool Upgrade(const std::string& pkgPath, const std::string &src, const std::string &desc)
|
|
{
|
|
bool ret = Setup(pkgPath);
|
|
if (ret)
|
|
PostInfo(1, src, desc);
|
|
return ret;
|
|
}
|