cmCTestSubmitHandler.cxx 32 KB

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