CZhipuziHttpClient.cpp 3.6 KB

123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125126127128129130131132133134135136137138139140141142143144145146147148149150151152153154155156157158159160161
  1. #include "../pch/pch.h"
  2. #include "CZhipuziHttpClient.h"
  3. #include "rapidjson/document.h"
  4. #include "rapidjson/prettywriter.h"
  5. #include "rapidjson/writer.h"
  6. #include "rapidjson/stringbuffer.h"
  7. using namespace rapidjson;
  8. #include "../helper/MD5.h"
  9. CZhipuziHttpClient CZhipuziHttpClient::m_client;
  10. CZhipuziHttpClient::CZhipuziHttpClient()
  11. {
  12. }
  13. CZhipuziHttpClient::~CZhipuziHttpClient()
  14. {
  15. }
  16. void CZhipuziHttpClient::Init(std::string username, std::string password)
  17. {
  18. m_client.m_username = username;
  19. m_client.m_password = password;
  20. }
  21. bool CZhipuziHttpClient::Login(std::string& errmsg)
  22. {
  23. std::map<string, string> params;
  24. std::string response;
  25. bool ret = m_client.Request("/user/login", params, response);
  26. if (!ret)
  27. {
  28. //网络请求出错
  29. LOG_INFO("network failed!");
  30. errmsg = "network failed!";
  31. return false;
  32. }
  33. LOG_INFO("response:" << response.c_str());
  34. rapidjson::Document document;
  35. document.Parse(response.c_str());
  36. if (!document.IsObject())
  37. {
  38. LOG_INFO("message 非法!");
  39. errmsg = "message 非法!";
  40. return false;
  41. }
  42. rapidjson::Value& v_errcode = document["errcode"];
  43. int errcode = v_errcode.GetInt();
  44. if (errcode > 0)
  45. {
  46. LOG_INFO("login failed! message:" << document["errmsg"].GetString());
  47. errmsg = std::string(document["errmsg"].GetString());
  48. return false;
  49. }
  50. LOG_INFO("login success!");
  51. return true;
  52. }
  53. bool CZhipuziHttpClient::Request(std::string url, std::map<string, string> params, std::string& response)
  54. {
  55. std::string timestamp = to_string(time(NULL));
  56. std::string nonce = "123456";
  57. std::string machinecode = "e6bc5694877b2aec";
  58. //先添加默认参数,用于计算签名
  59. params["username"] = m_username;
  60. params["timestamp"] = timestamp;
  61. params["nonce"] = nonce;
  62. params["machinecode"] = machinecode;
  63. params["url"] = m_url + url;
  64. //计算签名
  65. std::string postString;
  66. for (std::map<string, string>::iterator it = params.begin(); it != params.end(); )
  67. {
  68. postString += it->first + "=" + it->second;
  69. it++;
  70. if (it != params.end())
  71. {
  72. postString += "&";
  73. }
  74. }
  75. //LOG_INFO("postString:" << postString.c_str());
  76. std::string key = md5(m_password);
  77. transform(key.begin(), key.end(), key.begin(), ::toupper);
  78. //用于计算签名的临时变量
  79. std::string tmp = postString + key;
  80. string sign = md5(tmp);
  81. transform(sign.begin(), sign.end(), sign.begin(), ::toupper);
  82. //LOG_INFO("sign:" << sign.c_str());
  83. //加上签名,去掉url,计算post
  84. params["sign"] = sign;
  85. params.erase("url");
  86. postString = "";
  87. for (std::map<string, string>::iterator it = params.begin(); it != params.end(); )
  88. {
  89. postString += it->first + "=" + it->second;
  90. it++;
  91. if (it != params.end())
  92. {
  93. postString += "&";
  94. }
  95. }
  96. //LOG_INFO("postString:" << postString.c_str());
  97. int ret = m_httpClient.Posts(m_url + url, postString, response, NULL);
  98. //LOG_INFO("response:" << response.c_str());
  99. if (!ret)
  100. {
  101. //ret为0表示没有出错
  102. return true;
  103. }
  104. return false;
  105. /*
  106. rapidjson::Document doc;
  107. doc.SetObject();
  108. rapidjson::Document::AllocatorType &allocator = doc.GetAllocator(); //获取分配器
  109. doc.AddMember(rapidjson::StringRef("useranme"), rapidjson::StringRef(m_username.c_str()), allocator);
  110. doc.AddMember(rapidjson::StringRef("timestamp"), rapidjson::StringRef(timestamp.c_str()), allocator);
  111. doc.AddMember(rapidjson::StringRef("nonce"), rapidjson::StringRef(nonce.c_str()), allocator);
  112. doc.AddMember(rapidjson::StringRef("machinecode"), rapidjson::StringRef(machinecode.c_str()), allocator);
  113. rapidjson::StringBuffer buffer;
  114. rapidjson::Writer<rapidjson::StringBuffer> writer(buffer);
  115. doc.Accept(writer);
  116. std::string json = std::string(buffer.GetString());
  117. LOG_INFO("json:" << json.c_str());*/
  118. }