cmCTestSubmitHandler.cxx 31 KB

123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125126127128129130131132133134135136137138139140141142143144145146147148149150151152153154155156157158159160161162163164165166167168169170171172173174175176177178179180181182183184185186187188189190191192193194195196197198199200201202203204205206207208209210211212213214215216217218219220221222223224225226227228229230231232233234235236237238239240241242243244245246247248249250251252253254255256257258259260261262263264265266267268269270271272273274275276277278279280281282283284285286287288289290291292293294295296297298299300301302303304305306307308309310311312313314315316317318319320321322323324325326327328329330331332333334335336337338339340341342343344345346347348349350351352353354355356357358359360361362363364365366367368369370371372373374375376377378379380381382383384385386387388389390391392393394395396397398399400401402403404405406407408409410411412413414415416417418419420421422423424425426427428429430431432433434435436437438439440441442443444445446447448449450451452453454455456457458459460461462463464465466467468469470471472473474475476477478479480481482483484485486487488489490491492493494495496497498499500501502503504505506507508509510511512513514515516517518519520521522523524525526527528529530531532533534535536537538539540541542543544545546547548549550551552553554555556557558559560561562563564565566567568569570571572573574575576577578579580581582583584585586587588589590591592593594595596597598599600601602603604605606607608609610611612613614615616617618619620621622623624625626627628629630631632633634635636637638639640641642643644645646647648649650651652653654655656657658659660661662663664665666667668669670671672673674675676677678679680681682683684685686687688689690691692693694695696697698699700701702703704705706707708709710711712713714715716717718719720721722723724725726727728729730731732733734735736737738739740741742743744745746747748749750751752753754755756757758759760761762763764765766767768769770771772773774775776777778779780781782783784785786787788789790791792793794795796797798799800801802803804805806807808809810811812813814815816817818819820821822823824825826827828829830831832833834835836837838839840841842843844845846847848849850851852853854855856857858859860861862863864865866867868869870871872873874875876877878879880881882883884885886887888889890891892893894895896897898899900901902903904905906
  1. /* Distributed under the OSI-approved BSD 3-Clause License. See accompanying
  2. file Copyright.txt or https://cmake.org/licensing for details. */
  3. #include "cmCTestSubmitHandler.h"
  4. #include "cm_curl.h"
  5. #include "cm_jsoncpp_reader.h"
  6. #include "cm_jsoncpp_value.h"
  7. #include <chrono>
  8. #include <sstream>
  9. #include <stdio.h>
  10. #include <stdlib.h>
  11. #include "cmAlgorithms.h"
  12. #include "cmCTest.h"
  13. #include "cmCTestCurl.h"
  14. #include "cmCTestScriptHandler.h"
  15. #include "cmCryptoHash.h"
  16. #include "cmCurl.h"
  17. #include "cmDuration.h"
  18. #include "cmGeneratedFileStream.h"
  19. #include "cmState.h"
  20. #include "cmSystemTools.h"
  21. #include "cmXMLParser.h"
  22. #include "cmake.h"
  23. #define SUBMIT_TIMEOUT_IN_SECONDS_DEFAULT 120
  24. typedef std::vector<char> cmCTestSubmitHandlerVectorOfChar;
  25. class cmCTestSubmitHandler::ResponseParser : public cmXMLParser
  26. {
  27. public:
  28. enum StatusType
  29. {
  30. STATUS_OK,
  31. STATUS_WARNING,
  32. STATUS_ERROR
  33. };
  34. StatusType Status = STATUS_OK;
  35. std::string Filename;
  36. std::string MD5;
  37. std::string Message;
  38. std::string BuildID;
  39. private:
  40. std::vector<char> CurrentValue;
  41. std::string GetCurrentValue()
  42. {
  43. std::string val;
  44. if (!this->CurrentValue.empty()) {
  45. val.assign(&this->CurrentValue[0], this->CurrentValue.size());
  46. }
  47. return val;
  48. }
  49. void StartElement(const std::string& /*name*/,
  50. const char** /*atts*/) override
  51. {
  52. this->CurrentValue.clear();
  53. }
  54. void CharacterDataHandler(const char* data, int length) override
  55. {
  56. this->CurrentValue.insert(this->CurrentValue.end(), data, data + length);
  57. }
  58. void EndElement(const std::string& name) override
  59. {
  60. if (name == "status") {
  61. std::string status = cmSystemTools::UpperCase(this->GetCurrentValue());
  62. if (status == "OK" || status == "SUCCESS") {
  63. this->Status = STATUS_OK;
  64. } else if (status == "WARNING") {
  65. this->Status = STATUS_WARNING;
  66. } else {
  67. this->Status = STATUS_ERROR;
  68. }
  69. } else if (name == "filename") {
  70. this->Filename = this->GetCurrentValue();
  71. } else if (name == "md5") {
  72. this->MD5 = this->GetCurrentValue();
  73. } else if (name == "message") {
  74. this->Message = this->GetCurrentValue();
  75. } else if (name == "buildId") {
  76. this->BuildID = this->GetCurrentValue();
  77. }
  78. }
  79. };
  80. static size_t cmCTestSubmitHandlerWriteMemoryCallback(void* ptr, size_t size,
  81. size_t nmemb, void* data)
  82. {
  83. int realsize = static_cast<int>(size * nmemb);
  84. cmCTestSubmitHandlerVectorOfChar* vec =
  85. static_cast<cmCTestSubmitHandlerVectorOfChar*>(data);
  86. const char* chPtr = static_cast<char*>(ptr);
  87. vec->insert(vec->end(), chPtr, chPtr + realsize);
  88. return realsize;
  89. }
  90. static size_t cmCTestSubmitHandlerCurlDebugCallback(CURL* /*unused*/,
  91. curl_infotype /*unused*/,
  92. char* chPtr, size_t size,
  93. void* data)
  94. {
  95. cmCTestSubmitHandlerVectorOfChar* vec =
  96. static_cast<cmCTestSubmitHandlerVectorOfChar*>(data);
  97. vec->insert(vec->end(), chPtr, chPtr + size);
  98. return size;
  99. }
  100. cmCTestSubmitHandler::cmCTestSubmitHandler()
  101. {
  102. this->Initialize();
  103. }
  104. void cmCTestSubmitHandler::Initialize()
  105. {
  106. // We submit all available parts by default.
  107. for (cmCTest::Part p = cmCTest::PartStart; p != cmCTest::PartCount;
  108. p = cmCTest::Part(p + 1)) {
  109. this->SubmitPart[p] = true;
  110. }
  111. this->HasWarnings = false;
  112. this->HasErrors = false;
  113. this->Superclass::Initialize();
  114. this->HTTPProxy.clear();
  115. this->HTTPProxyType = 0;
  116. this->HTTPProxyAuth.clear();
  117. this->LogFile = nullptr;
  118. this->Files.clear();
  119. }
  120. bool cmCTestSubmitHandler::SubmitUsingHTTP(
  121. const std::string& localprefix, const std::vector<std::string>& files,
  122. const std::string& remoteprefix, const std::string& url)
  123. {
  124. CURL* curl;
  125. CURLcode res;
  126. FILE* ftpfile;
  127. char error_buffer[1024];
  128. // Set Content-Type to satisfy fussy modsecurity rules.
  129. struct curl_slist* headers =
  130. ::curl_slist_append(nullptr, "Content-Type: text/xml");
  131. // Add any additional headers that the user specified.
  132. for (std::string const& h : this->HttpHeaders) {
  133. cmCTestOptionalLog(this->CTest, DEBUG,
  134. " Add HTTP Header: \"" << h << "\"" << std::endl,
  135. this->Quiet);
  136. headers = ::curl_slist_append(headers, h.c_str());
  137. }
  138. /* In windows, this will init the winsock stuff */
  139. ::curl_global_init(CURL_GLOBAL_ALL);
  140. std::string curlopt(this->CTest->GetCTestConfiguration("CurlOptions"));
  141. std::vector<std::string> args;
  142. cmSystemTools::ExpandListArgument(curlopt, args);
  143. bool verifyPeerOff = false;
  144. bool verifyHostOff = false;
  145. for (std::string const& arg : args) {
  146. if (arg == "CURLOPT_SSL_VERIFYPEER_OFF") {
  147. verifyPeerOff = true;
  148. }
  149. if (arg == "CURLOPT_SSL_VERIFYHOST_OFF") {
  150. verifyHostOff = true;
  151. }
  152. }
  153. for (std::string const& file : files) {
  154. /* get a curl handle */
  155. curl = curl_easy_init();
  156. if (curl) {
  157. cmCurlSetCAInfo(curl);
  158. if (verifyPeerOff) {
  159. cmCTestOptionalLog(this->CTest, HANDLER_VERBOSE_OUTPUT,
  160. " Set CURLOPT_SSL_VERIFYPEER to off\n",
  161. this->Quiet);
  162. curl_easy_setopt(curl, CURLOPT_SSL_VERIFYPEER, 0);
  163. }
  164. if (verifyHostOff) {
  165. cmCTestOptionalLog(this->CTest, HANDLER_VERBOSE_OUTPUT,
  166. " Set CURLOPT_SSL_VERIFYHOST to off\n",
  167. this->Quiet);
  168. curl_easy_setopt(curl, CURLOPT_SSL_VERIFYHOST, 0);
  169. }
  170. // Using proxy
  171. if (this->HTTPProxyType > 0) {
  172. curl_easy_setopt(curl, CURLOPT_PROXY, this->HTTPProxy.c_str());
  173. switch (this->HTTPProxyType) {
  174. case 2:
  175. curl_easy_setopt(curl, CURLOPT_PROXYTYPE, CURLPROXY_SOCKS4);
  176. break;
  177. case 3:
  178. curl_easy_setopt(curl, CURLOPT_PROXYTYPE, CURLPROXY_SOCKS5);
  179. break;
  180. default:
  181. curl_easy_setopt(curl, CURLOPT_PROXYTYPE, CURLPROXY_HTTP);
  182. if (!this->HTTPProxyAuth.empty()) {
  183. curl_easy_setopt(curl, CURLOPT_PROXYUSERPWD,
  184. this->HTTPProxyAuth.c_str());
  185. }
  186. }
  187. }
  188. if (this->CTest->ShouldUseHTTP10()) {
  189. curl_easy_setopt(curl, CURLOPT_HTTP_VERSION, CURL_HTTP_VERSION_1_0);
  190. }
  191. // enable HTTP ERROR parsing
  192. curl_easy_setopt(curl, CURLOPT_FAILONERROR, 1);
  193. /* enable uploading */
  194. curl_easy_setopt(curl, CURLOPT_UPLOAD, 1);
  195. // if there is little to no activity for too long stop submitting
  196. ::curl_easy_setopt(curl, CURLOPT_LOW_SPEED_LIMIT, 1);
  197. ::curl_easy_setopt(curl, CURLOPT_LOW_SPEED_TIME,
  198. SUBMIT_TIMEOUT_IN_SECONDS_DEFAULT);
  199. /* HTTP PUT please */
  200. ::curl_easy_setopt(curl, CURLOPT_PUT, 1);
  201. ::curl_easy_setopt(curl, CURLOPT_VERBOSE, 1);
  202. ::curl_easy_setopt(curl, CURLOPT_HTTPHEADER, headers);
  203. std::string local_file = file;
  204. bool initialize_cdash_buildid = false;
  205. if (!cmSystemTools::FileExists(local_file)) {
  206. local_file = localprefix + "/" + file;
  207. // If this file exists within the local Testing directory we assume
  208. // that it will be associated with the current build in CDash.
  209. initialize_cdash_buildid = true;
  210. }
  211. std::string remote_file =
  212. remoteprefix + cmSystemTools::GetFilenameName(file);
  213. *this->LogFile << "\tUpload file: " << local_file << " to "
  214. << remote_file << std::endl;
  215. std::string ofile = cmSystemTools::EncodeURL(remote_file);
  216. std::string upload_as = url +
  217. ((url.find('?') == std::string::npos) ? '?' : '&') +
  218. "FileName=" + ofile;
  219. if (initialize_cdash_buildid) {
  220. // Provide extra arguments to CDash so that it can initialize and
  221. // return a buildid.
  222. cmCTestCurl ctest_curl(this->CTest);
  223. upload_as += "&build=";
  224. upload_as +=
  225. ctest_curl.Escape(this->CTest->GetCTestConfiguration("BuildName"));
  226. upload_as += "&site=";
  227. upload_as +=
  228. ctest_curl.Escape(this->CTest->GetCTestConfiguration("Site"));
  229. upload_as += "&stamp=";
  230. upload_as += ctest_curl.Escape(this->CTest->GetCurrentTag());
  231. upload_as += "-";
  232. upload_as += ctest_curl.Escape(this->CTest->GetTestModelString());
  233. cmCTestScriptHandler* ch = this->CTest->GetScriptHandler();
  234. cmake* cm = ch->GetCMake();
  235. if (cm) {
  236. const char* subproject =
  237. cm->GetState()->GetGlobalProperty("SubProject");
  238. if (subproject) {
  239. upload_as += "&subproject=";
  240. upload_as += ctest_curl.Escape(subproject);
  241. }
  242. }
  243. }
  244. upload_as += "&MD5=";
  245. if (cmSystemTools::IsOn(this->GetOption("InternalTest"))) {
  246. upload_as += "bad_md5sum";
  247. } else {
  248. upload_as +=
  249. cmSystemTools::ComputeFileHash(local_file, cmCryptoHash::AlgoMD5);
  250. }
  251. // Generate Done.xml right before it is submitted.
  252. // The reason for this is two-fold:
  253. // 1) It must be generated after some other part has been submitted
  254. // so we have a buildId to refer to in its contents.
  255. // 2) By generating Done.xml here its timestamp will be as late as
  256. // possible. This gives us a more accurate record of how long the
  257. // entire build took to complete.
  258. if (file == "Done.xml") {
  259. this->CTest->GenerateDoneFile();
  260. }
  261. if (!cmSystemTools::FileExists(local_file)) {
  262. cmCTestLog(this->CTest, ERROR_MESSAGE,
  263. " Cannot find file: " << local_file << std::endl);
  264. ::curl_easy_cleanup(curl);
  265. ::curl_slist_free_all(headers);
  266. ::curl_global_cleanup();
  267. return false;
  268. }
  269. unsigned long filelen = cmSystemTools::FileLength(local_file);
  270. ftpfile = cmsys::SystemTools::Fopen(local_file, "rb");
  271. cmCTestOptionalLog(this->CTest, HANDLER_VERBOSE_OUTPUT,
  272. " Upload file: " << local_file << " to "
  273. << upload_as << " Size: "
  274. << filelen << std::endl,
  275. this->Quiet);
  276. // specify target
  277. ::curl_easy_setopt(curl, CURLOPT_URL, upload_as.c_str());
  278. // CURLAUTH_BASIC is default, and here we allow additional methods,
  279. // including more secure ones
  280. ::curl_easy_setopt(curl, CURLOPT_HTTPAUTH, CURLAUTH_ANY);
  281. // now specify which file to upload
  282. ::curl_easy_setopt(curl, CURLOPT_INFILE, ftpfile);
  283. // and give the size of the upload (optional)
  284. ::curl_easy_setopt(curl, CURLOPT_INFILESIZE, static_cast<long>(filelen));
  285. // and give curl the buffer for errors
  286. ::curl_easy_setopt(curl, CURLOPT_ERRORBUFFER, &error_buffer);
  287. // specify handler for output
  288. ::curl_easy_setopt(curl, CURLOPT_WRITEFUNCTION,
  289. cmCTestSubmitHandlerWriteMemoryCallback);
  290. ::curl_easy_setopt(curl, CURLOPT_DEBUGFUNCTION,
  291. cmCTestSubmitHandlerCurlDebugCallback);
  292. /* we pass our 'chunk' struct to the callback function */
  293. cmCTestSubmitHandlerVectorOfChar chunk;
  294. cmCTestSubmitHandlerVectorOfChar chunkDebug;
  295. ::curl_easy_setopt(curl, CURLOPT_FILE, &chunk);
  296. ::curl_easy_setopt(curl, CURLOPT_DEBUGDATA, &chunkDebug);
  297. // Now run off and do what you've been told!
  298. res = ::curl_easy_perform(curl);
  299. if (!chunk.empty()) {
  300. cmCTestOptionalLog(this->CTest, DEBUG,
  301. "CURL output: ["
  302. << cmCTestLogWrite(chunk.data(), chunk.size())
  303. << "]" << std::endl,
  304. this->Quiet);
  305. this->ParseResponse(chunk);
  306. }
  307. if (!chunkDebug.empty()) {
  308. cmCTestOptionalLog(
  309. this->CTest, DEBUG,
  310. "CURL debug output: ["
  311. << cmCTestLogWrite(chunkDebug.data(), chunkDebug.size()) << "]"
  312. << std::endl,
  313. this->Quiet);
  314. }
  315. // If curl failed for any reason, or checksum fails, wait and retry
  316. //
  317. if (res != CURLE_OK || this->HasErrors) {
  318. std::string retryDelay = this->GetOption("RetryDelay") == nullptr
  319. ? ""
  320. : this->GetOption("RetryDelay");
  321. std::string retryCount = this->GetOption("RetryCount") == nullptr
  322. ? ""
  323. : this->GetOption("RetryCount");
  324. auto delay = cmDuration(
  325. retryDelay.empty()
  326. ? atoi(this->CTest->GetCTestConfiguration("CTestSubmitRetryDelay")
  327. .c_str())
  328. : atoi(retryDelay.c_str()));
  329. int count = retryCount.empty()
  330. ? atoi(this->CTest->GetCTestConfiguration("CTestSubmitRetryCount")
  331. .c_str())
  332. : atoi(retryCount.c_str());
  333. for (int i = 0; i < count; i++) {
  334. cmCTestOptionalLog(this->CTest, HANDLER_OUTPUT,
  335. " Submit failed, waiting " << delay.count()
  336. << " seconds...\n",
  337. this->Quiet);
  338. auto stop = std::chrono::steady_clock::now() + delay;
  339. while (std::chrono::steady_clock::now() < stop) {
  340. cmSystemTools::Delay(100);
  341. }
  342. cmCTestOptionalLog(this->CTest, HANDLER_OUTPUT,
  343. " Retry submission: Attempt "
  344. << (i + 1) << " of " << count << std::endl,
  345. this->Quiet);
  346. ::fclose(ftpfile);
  347. ftpfile = cmsys::SystemTools::Fopen(local_file, "rb");
  348. ::curl_easy_setopt(curl, CURLOPT_INFILE, ftpfile);
  349. chunk.clear();
  350. chunkDebug.clear();
  351. this->HasErrors = false;
  352. res = ::curl_easy_perform(curl);
  353. if (!chunk.empty()) {
  354. cmCTestOptionalLog(this->CTest, DEBUG,
  355. "CURL output: ["
  356. << cmCTestLogWrite(chunk.data(), chunk.size())
  357. << "]" << std::endl,
  358. this->Quiet);
  359. this->ParseResponse(chunk);
  360. }
  361. if (res == CURLE_OK && !this->HasErrors) {
  362. break;
  363. }
  364. }
  365. }
  366. fclose(ftpfile);
  367. if (res) {
  368. cmCTestLog(this->CTest, ERROR_MESSAGE,
  369. " Error when uploading file: " << local_file
  370. << std::endl);
  371. cmCTestLog(this->CTest, ERROR_MESSAGE,
  372. " Error message was: " << error_buffer << std::endl);
  373. *this->LogFile << " Error when uploading file: " << local_file
  374. << std::endl
  375. << " Error message was: " << error_buffer
  376. << std::endl;
  377. // avoid deref of begin for zero size array
  378. if (!chunk.empty()) {
  379. *this->LogFile << " Curl output was: "
  380. << cmCTestLogWrite(chunk.data(), chunk.size())
  381. << std::endl;
  382. cmCTestLog(this->CTest, ERROR_MESSAGE,
  383. "CURL output: ["
  384. << cmCTestLogWrite(chunk.data(), chunk.size()) << "]"
  385. << std::endl);
  386. }
  387. ::curl_easy_cleanup(curl);
  388. ::curl_slist_free_all(headers);
  389. ::curl_global_cleanup();
  390. return false;
  391. }
  392. // always cleanup
  393. ::curl_easy_cleanup(curl);
  394. cmCTestOptionalLog(this->CTest, HANDLER_OUTPUT,
  395. " Uploaded: " + local_file << std::endl,
  396. this->Quiet);
  397. }
  398. }
  399. ::curl_slist_free_all(headers);
  400. ::curl_global_cleanup();
  401. return true;
  402. }
  403. void cmCTestSubmitHandler::ParseResponse(
  404. cmCTestSubmitHandlerVectorOfChar chunk)
  405. {
  406. std::string output;
  407. output.append(chunk.begin(), chunk.end());
  408. if (output.find("<cdash") != std::string::npos) {
  409. ResponseParser parser;
  410. parser.Parse(output.c_str());
  411. if (parser.Status != ResponseParser::STATUS_OK) {
  412. this->HasErrors = true;
  413. cmCTestLog(this->CTest, HANDLER_OUTPUT,
  414. " Submission failed: " << parser.Message << std::endl);
  415. return;
  416. }
  417. this->CTest->SetBuildID(parser.BuildID);
  418. }
  419. output = cmSystemTools::UpperCase(output);
  420. if (output.find("WARNING") != std::string::npos) {
  421. this->HasWarnings = true;
  422. }
  423. if (output.find("ERROR") != std::string::npos) {
  424. this->HasErrors = true;
  425. }
  426. if (this->HasWarnings || this->HasErrors) {
  427. cmCTestLog(this->CTest, HANDLER_OUTPUT,
  428. " Server Response:\n"
  429. << cmCTestLogWrite(chunk.data(), chunk.size()) << "\n");
  430. }
  431. }
  432. int cmCTestSubmitHandler::HandleCDashUploadFile(std::string const& file,
  433. std::string const& typeString)
  434. {
  435. if (file.empty()) {
  436. cmCTestLog(this->CTest, ERROR_MESSAGE, "Upload file not specified\n");
  437. return -1;
  438. }
  439. if (!cmSystemTools::FileExists(file)) {
  440. cmCTestLog(this->CTest, ERROR_MESSAGE,
  441. "Upload file not found: '" << file << "'\n");
  442. return -1;
  443. }
  444. cmCTestCurl curl(this->CTest);
  445. curl.SetQuiet(this->Quiet);
  446. std::string curlopt(this->CTest->GetCTestConfiguration("CurlOptions"));
  447. std::vector<std::string> args;
  448. cmSystemTools::ExpandListArgument(curlopt, args);
  449. curl.SetCurlOptions(args);
  450. curl.SetTimeOutSeconds(SUBMIT_TIMEOUT_IN_SECONDS_DEFAULT);
  451. curl.SetHttpHeaders(this->HttpHeaders);
  452. std::string url = this->CTest->GetSubmitURL();
  453. std::string fields;
  454. std::string::size_type pos = url.find('?');
  455. if (pos != std::string::npos) {
  456. fields = url.substr(pos + 1);
  457. url = url.substr(0, pos);
  458. }
  459. if (!cmHasLiteralPrefix(url, "http://") &&
  460. !cmHasLiteralPrefix(url, "https://")) {
  461. cmCTestLog(this->CTest, ERROR_MESSAGE,
  462. "Only http and https are supported for CDASH_UPLOAD\n");
  463. return -1;
  464. }
  465. bool internalTest = cmSystemTools::IsOn(this->GetOption("InternalTest"));
  466. // Get RETRY_COUNT and RETRY_DELAY values if they were set.
  467. std::string retryDelayString = this->GetOption("RetryDelay") == nullptr
  468. ? ""
  469. : this->GetOption("RetryDelay");
  470. std::string retryCountString = this->GetOption("RetryCount") == nullptr
  471. ? ""
  472. : this->GetOption("RetryCount");
  473. auto retryDelay = std::chrono::seconds(0);
  474. if (!retryDelayString.empty()) {
  475. unsigned long retryDelayValue = 0;
  476. if (!cmSystemTools::StringToULong(retryDelayString.c_str(),
  477. &retryDelayValue)) {
  478. cmCTestLog(this->CTest, WARNING,
  479. "Invalid value for 'RETRY_DELAY' : " << retryDelayString
  480. << std::endl);
  481. } else {
  482. retryDelay = std::chrono::seconds(retryDelayValue);
  483. }
  484. }
  485. unsigned long retryCount = 0;
  486. if (!retryCountString.empty()) {
  487. if (!cmSystemTools::StringToULong(retryCountString.c_str(), &retryCount)) {
  488. cmCTestLog(this->CTest, WARNING,
  489. "Invalid value for 'RETRY_DELAY' : " << retryCountString
  490. << std::endl);
  491. }
  492. }
  493. std::string md5sum =
  494. cmSystemTools::ComputeFileHash(file, cmCryptoHash::AlgoMD5);
  495. // 1. request the buildid and check to see if the file
  496. // has already been uploaded
  497. // TODO I added support for subproject. You would need to add
  498. // a "&subproject=subprojectname" to the first POST.
  499. cmCTestScriptHandler* ch = this->CTest->GetScriptHandler();
  500. cmake* cm = ch->GetCMake();
  501. const char* subproject = cm->GetState()->GetGlobalProperty("SubProject");
  502. // TODO: Encode values for a URL instead of trusting caller.
  503. std::ostringstream str;
  504. if (subproject) {
  505. str << "subproject=" << curl.Escape(subproject) << "&";
  506. }
  507. auto timeNow =
  508. std::chrono::system_clock::to_time_t(std::chrono::system_clock::now());
  509. str << "stamp=" << curl.Escape(this->CTest->GetCurrentTag()) << "-"
  510. << curl.Escape(this->CTest->GetTestModelString()) << "&"
  511. << "model=" << curl.Escape(this->CTest->GetTestModelString()) << "&"
  512. << "build="
  513. << curl.Escape(this->CTest->GetCTestConfiguration("BuildName")) << "&"
  514. << "site=" << curl.Escape(this->CTest->GetCTestConfiguration("Site"))
  515. << "&"
  516. << "track=" << curl.Escape(this->CTest->GetTestModelString()) << "&"
  517. << "starttime=" << timeNow << "&"
  518. << "endtime=" << timeNow << "&"
  519. << "datafilesmd5[0]=" << md5sum << "&"
  520. << "type=" << curl.Escape(typeString);
  521. if (!fields.empty()) {
  522. fields += '&';
  523. }
  524. fields += str.str();
  525. cmCTestOptionalLog(this->CTest, DEBUG,
  526. "fields: " << fields << "\nurl:" << url
  527. << "\nfile: " << file << "\n",
  528. this->Quiet);
  529. std::string response;
  530. bool requestSucceeded = curl.HttpRequest(url, fields, response);
  531. if (!internalTest && !requestSucceeded) {
  532. // If request failed, wait and retry.
  533. for (unsigned long i = 0; i < retryCount; i++) {
  534. cmCTestOptionalLog(this->CTest, HANDLER_OUTPUT,
  535. " Request failed, waiting " << retryDelay.count()
  536. << " seconds...\n",
  537. this->Quiet);
  538. auto stop = std::chrono::steady_clock::now() + retryDelay;
  539. while (std::chrono::steady_clock::now() < stop) {
  540. cmSystemTools::Delay(100);
  541. }
  542. cmCTestOptionalLog(this->CTest, HANDLER_OUTPUT,
  543. " Retry request: Attempt "
  544. << (i + 1) << " of " << retryCount << std::endl,
  545. this->Quiet);
  546. requestSucceeded = curl.HttpRequest(url, fields, response);
  547. if (requestSucceeded) {
  548. break;
  549. }
  550. }
  551. }
  552. if (!internalTest && !requestSucceeded) {
  553. cmCTestLog(this->CTest, ERROR_MESSAGE,
  554. "Error in HttpRequest\n"
  555. << response);
  556. return -1;
  557. }
  558. cmCTestOptionalLog(this->CTest, HANDLER_VERBOSE_OUTPUT,
  559. "Request upload response: [" << response << "]\n",
  560. this->Quiet);
  561. Json::Value json;
  562. Json::Reader reader;
  563. if (!internalTest && !reader.parse(response, json)) {
  564. cmCTestLog(this->CTest, ERROR_MESSAGE,
  565. "error parsing json string ["
  566. << response << "]\n"
  567. << reader.getFormattedErrorMessages() << "\n");
  568. return -1;
  569. }
  570. if (!internalTest && json["status"].asInt() != 0) {
  571. cmCTestLog(this->CTest, ERROR_MESSAGE,
  572. "Bad status returned from CDash: " << json["status"].asInt());
  573. return -1;
  574. }
  575. if (!internalTest) {
  576. if (json["datafilesmd5"].isArray()) {
  577. int datares = json["datafilesmd5"][0].asInt();
  578. if (datares == 1) {
  579. cmCTestOptionalLog(this->CTest, HANDLER_VERBOSE_OUTPUT,
  580. "File already exists on CDash, skip upload "
  581. << file << "\n",
  582. this->Quiet);
  583. return 0;
  584. }
  585. } else {
  586. cmCTestLog(this->CTest, ERROR_MESSAGE,
  587. "bad datafilesmd5 value in response " << response << "\n");
  588. return -1;
  589. }
  590. }
  591. std::string upload_as = cmSystemTools::GetFilenameName(file);
  592. std::ostringstream fstr;
  593. fstr << "type=" << curl.Escape(typeString) << "&"
  594. << "md5=" << md5sum << "&"
  595. << "filename=" << curl.Escape(upload_as) << "&"
  596. << "buildid=" << json["buildid"].asString();
  597. bool uploadSucceeded = false;
  598. if (!internalTest) {
  599. uploadSucceeded = curl.UploadFile(file, url, fstr.str(), response);
  600. }
  601. if (!uploadSucceeded) {
  602. // If upload failed, wait and retry.
  603. for (unsigned long i = 0; i < retryCount; i++) {
  604. cmCTestOptionalLog(this->CTest, HANDLER_OUTPUT,
  605. " Upload failed, waiting " << retryDelay.count()
  606. << " seconds...\n",
  607. this->Quiet);
  608. auto stop = std::chrono::steady_clock::now() + retryDelay;
  609. while (std::chrono::steady_clock::now() < stop) {
  610. cmSystemTools::Delay(100);
  611. }
  612. cmCTestOptionalLog(this->CTest, HANDLER_OUTPUT,
  613. " Retry upload: Attempt "
  614. << (i + 1) << " of " << retryCount << std::endl,
  615. this->Quiet);
  616. if (!internalTest) {
  617. uploadSucceeded = curl.UploadFile(file, url, fstr.str(), response);
  618. }
  619. if (uploadSucceeded) {
  620. break;
  621. }
  622. }
  623. }
  624. if (!uploadSucceeded) {
  625. cmCTestLog(this->CTest, ERROR_MESSAGE,
  626. "error uploading to CDash. " << file << " " << url << " "
  627. << fstr.str());
  628. return -1;
  629. }
  630. if (!reader.parse(response, json)) {
  631. cmCTestLog(this->CTest, ERROR_MESSAGE,
  632. "error parsing json string ["
  633. << response << "]\n"
  634. << reader.getFormattedErrorMessages() << "\n");
  635. return -1;
  636. }
  637. cmCTestOptionalLog(this->CTest, HANDLER_VERBOSE_OUTPUT,
  638. "Upload file response: [" << response << "]\n",
  639. this->Quiet);
  640. return 0;
  641. }
  642. int cmCTestSubmitHandler::ProcessHandler()
  643. {
  644. const char* cdashUploadFile = this->GetOption("CDashUploadFile");
  645. const char* cdashUploadType = this->GetOption("CDashUploadType");
  646. if (cdashUploadFile && cdashUploadType) {
  647. return this->HandleCDashUploadFile(cdashUploadFile, cdashUploadType);
  648. }
  649. const std::string& buildDirectory =
  650. this->CTest->GetCTestConfiguration("BuildDirectory");
  651. if (buildDirectory.empty()) {
  652. cmCTestLog(this->CTest, ERROR_MESSAGE,
  653. "Cannot find BuildDirectory key in the DartConfiguration.tcl"
  654. << std::endl);
  655. return -1;
  656. }
  657. if (getenv("HTTP_PROXY")) {
  658. this->HTTPProxyType = 1;
  659. this->HTTPProxy = getenv("HTTP_PROXY");
  660. if (getenv("HTTP_PROXY_PORT")) {
  661. this->HTTPProxy += ":";
  662. this->HTTPProxy += getenv("HTTP_PROXY_PORT");
  663. }
  664. if (getenv("HTTP_PROXY_TYPE")) {
  665. std::string type = getenv("HTTP_PROXY_TYPE");
  666. // HTTP/SOCKS4/SOCKS5
  667. if (type == "HTTP") {
  668. this->HTTPProxyType = 1;
  669. } else if (type == "SOCKS4") {
  670. this->HTTPProxyType = 2;
  671. } else if (type == "SOCKS5") {
  672. this->HTTPProxyType = 3;
  673. }
  674. }
  675. if (getenv("HTTP_PROXY_USER")) {
  676. this->HTTPProxyAuth = getenv("HTTP_PROXY_USER");
  677. }
  678. if (getenv("HTTP_PROXY_PASSWD")) {
  679. this->HTTPProxyAuth += ":";
  680. this->HTTPProxyAuth += getenv("HTTP_PROXY_PASSWD");
  681. }
  682. }
  683. if (!this->HTTPProxy.empty()) {
  684. cmCTestOptionalLog(this->CTest, HANDLER_OUTPUT,
  685. " Use HTTP Proxy: " << this->HTTPProxy << std::endl,
  686. this->Quiet);
  687. }
  688. cmGeneratedFileStream ofs;
  689. this->StartLogFile("Submit", ofs);
  690. std::vector<std::string> files;
  691. std::string prefix = this->GetSubmitResultsPrefix();
  692. if (!this->Files.empty()) {
  693. // Submit the explicitly selected files:
  694. //
  695. files.insert(files.end(), this->Files.begin(), this->Files.end());
  696. }
  697. // Add to the list of files to submit from any selected, existing parts:
  698. //
  699. // TODO:
  700. // Check if test is enabled
  701. this->CTest->AddIfExists(cmCTest::PartUpdate, "Update.xml");
  702. this->CTest->AddIfExists(cmCTest::PartConfigure, "Configure.xml");
  703. this->CTest->AddIfExists(cmCTest::PartBuild, "Build.xml");
  704. this->CTest->AddIfExists(cmCTest::PartTest, "Test.xml");
  705. if (this->CTest->AddIfExists(cmCTest::PartCoverage, "Coverage.xml")) {
  706. std::vector<std::string> gfiles;
  707. std::string gpath =
  708. buildDirectory + "/Testing/" + this->CTest->GetCurrentTag();
  709. std::string::size_type glen = gpath.size() + 1;
  710. gpath = gpath + "/CoverageLog*";
  711. cmCTestOptionalLog(this->CTest, DEBUG,
  712. "Globbing for: " << gpath << std::endl, this->Quiet);
  713. if (cmSystemTools::SimpleGlob(gpath, gfiles, 1)) {
  714. for (std::string& gfile : gfiles) {
  715. gfile = gfile.substr(glen);
  716. cmCTestOptionalLog(this->CTest, DEBUG,
  717. "Glob file: " << gfile << std::endl, this->Quiet);
  718. this->CTest->AddSubmitFile(cmCTest::PartCoverage, gfile.c_str());
  719. }
  720. } else {
  721. cmCTestLog(this->CTest, ERROR_MESSAGE, "Problem globbing" << std::endl);
  722. }
  723. }
  724. this->CTest->AddIfExists(cmCTest::PartMemCheck, "DynamicAnalysis.xml");
  725. this->CTest->AddIfExists(cmCTest::PartMemCheck, "Purify.xml");
  726. this->CTest->AddIfExists(cmCTest::PartNotes, "Notes.xml");
  727. this->CTest->AddIfExists(cmCTest::PartUpload, "Upload.xml");
  728. // Query parts for files to submit.
  729. for (cmCTest::Part p = cmCTest::PartStart; p != cmCTest::PartCount;
  730. p = cmCTest::Part(p + 1)) {
  731. // Skip parts we are not submitting.
  732. if (!this->SubmitPart[p]) {
  733. continue;
  734. }
  735. // Submit files from this part.
  736. std::vector<std::string> const& pfiles = this->CTest->GetSubmitFiles(p);
  737. files.insert(files.end(), pfiles.begin(), pfiles.end());
  738. }
  739. // Make sure files are unique, but preserve order.
  740. {
  741. // This endPos intermediate is needed to work around non-conformant C++11
  742. // standard libraries that have erase(iterator,iterator) instead of
  743. // erase(const_iterator,const_iterator).
  744. size_t endPos = cmRemoveDuplicates(files) - files.cbegin();
  745. files.erase(files.begin() + endPos, files.end());
  746. }
  747. // Submit Done.xml last
  748. if (this->SubmitPart[cmCTest::PartDone]) {
  749. files.emplace_back("Done.xml");
  750. }
  751. if (ofs) {
  752. ofs << "Upload files:" << std::endl;
  753. int cnt = 0;
  754. for (std::string const& file : files) {
  755. ofs << cnt << "\t" << file << std::endl;
  756. cnt++;
  757. }
  758. }
  759. cmCTestOptionalLog(this->CTest, HANDLER_OUTPUT, "Submit files\n",
  760. this->Quiet);
  761. const char* specificTrack = this->CTest->GetSpecificTrack();
  762. if (specificTrack) {
  763. cmCTestOptionalLog(this->CTest, HANDLER_OUTPUT,
  764. " Send to track: " << specificTrack << std::endl,
  765. this->Quiet);
  766. }
  767. this->SetLogFile(&ofs);
  768. std::string url = this->CTest->GetSubmitURL();
  769. cmCTestOptionalLog(this->CTest, HANDLER_OUTPUT,
  770. " SubmitURL: " << url << '\n', this->Quiet);
  771. if (!this->SubmitUsingHTTP(buildDirectory + "/Testing/" +
  772. this->CTest->GetCurrentTag(),
  773. files, prefix, url)) {
  774. cmCTestLog(this->CTest, ERROR_MESSAGE,
  775. " Problems when submitting via HTTP\n");
  776. ofs << " Problems when submitting via HTTP\n";
  777. return -1;
  778. }
  779. if (this->HasErrors) {
  780. cmCTestLog(this->CTest, HANDLER_OUTPUT,
  781. " Errors occurred during submission.\n");
  782. ofs << " Errors occurred during submission.\n";
  783. } else {
  784. cmCTestOptionalLog(this->CTest, HANDLER_OUTPUT,
  785. " Submission successful"
  786. << (this->HasWarnings ? ", with warnings." : "")
  787. << std::endl,
  788. this->Quiet);
  789. ofs << " Submission successful"
  790. << (this->HasWarnings ? ", with warnings." : "") << std::endl;
  791. }
  792. return 0;
  793. }
  794. std::string cmCTestSubmitHandler::GetSubmitResultsPrefix()
  795. {
  796. std::string buildname =
  797. cmCTest::SafeBuildIdField(this->CTest->GetCTestConfiguration("BuildName"));
  798. std::string name = this->CTest->GetCTestConfiguration("Site") + "___" +
  799. buildname + "___" + this->CTest->GetCurrentTag() + "-" +
  800. this->CTest->GetTestModelString() + "___XML___";
  801. return name;
  802. }
  803. void cmCTestSubmitHandler::SelectParts(std::set<cmCTest::Part> const& parts)
  804. {
  805. // Check whether each part is selected.
  806. for (cmCTest::Part p = cmCTest::PartStart; p != cmCTest::PartCount;
  807. p = cmCTest::Part(p + 1)) {
  808. this->SubmitPart[p] =
  809. (std::set<cmCTest::Part>::const_iterator(parts.find(p)) != parts.end());
  810. }
  811. }
  812. void cmCTestSubmitHandler::SelectFiles(cmCTest::SetOfStrings const& files)
  813. {
  814. this->Files.insert(files.begin(), files.end());
  815. }