test_runner.c 28 KB

123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125126127128129130131132133134135136137138139140141142143144145146147148149150151152153154155156157158159160161162163164165166167168169170171172173174175176177178179180181182183184185186187188189190191192193194195196197198199200201202203204205206207208209210211212213214215216217218219220221222223224225226227228229230231232233234235236237238239240241242243244245246247248249250251252253254255256257258259260261262263264265266267268269270271272273274275276277278279280281282283284285286287288289290291292293294295296297298299300301302303304305306307308309310311312313314315316317318319320321322323324325326327328329330331332333334335336337338339340341342343344345346347348349350351352353354355356357358359360361362363364365366367368369370371372373374375376377378379380381382383384385386387388389390391392393394395396397398399400401402403404405406407408409410411412413414415416417418419420421422423424425426427428429430431432433434435436437438439440441442443444445446447448449450451452453454455456457458459460461462463464465466467468469470471472473474475476477478479480481482483484485486487488489490491492493494495496497498499500501502503504505506507508509510511512513514515516517518519520521522523524525526527528529530531532533534535536537538539540541542543544545546547548549550551552553554555556557558559560561562563564565566567568569570571572573574575576577578579580581582583584585586587588589590591592593594595596597598599600601602603604605606607608609610611612613614615616617618619620621622623624625626627628629630631632633634635636637638639640641642643644645646647648649650651652653654655656657658659660661662663664665666667668669670671672673674675676677678679680681682683684685686687688689690691692693694695696697698699700701702703704705706707708709710711712713714715716717718719720721722723724725726727728729730731732733734735736737738739740741742743744745746747748749750751752753754755756757758759760761762763764765766767768769770771772773774775776777778779780781782783784785786787788789790791792793794795796797798799800801802803804805806807808809810811812813814815816817818819820821822823824825826827828829830831832833834835836837838839840841842843844845846847848849850851852853854855856857858859860861862863864865866867868869870871872873874875876
  1. #include "runners/test_runner.h"
  2. #include "bd/lfs_testbd.h"
  3. #include <getopt.h>
  4. #include <sys/types.h>
  5. #include <errno.h>
  6. // test geometries
  7. struct test_geometry {
  8. const char *name;
  9. test_define_t defines[TEST_GEOMETRY_DEFINE_COUNT];
  10. };
  11. const struct test_geometry test_geometries[TEST_GEOMETRY_COUNT]
  12. = TEST_GEOMETRIES;
  13. // test define lookup and management
  14. #define TEST_DEFINE_LAYERS 4
  15. const test_define_t *test_defines[TEST_DEFINE_LAYERS] = {
  16. NULL,
  17. NULL,
  18. NULL,
  19. (const test_define_t[TEST_DEFAULT_COUNT])TEST_DEFAULTS,
  20. };
  21. const uint8_t *test_predefine_maps[TEST_DEFINE_LAYERS] = {
  22. NULL,
  23. NULL,
  24. (const uint8_t[TEST_PREDEFINE_COUNT])TEST_GEOMETRY_DEFINE_MAP,
  25. (const uint8_t[TEST_PREDEFINE_COUNT])TEST_DEFAULT_MAP,
  26. };
  27. const uint8_t *test_define_maps[TEST_DEFINE_LAYERS] = {
  28. NULL,
  29. NULL,
  30. NULL,
  31. NULL,
  32. };
  33. uint8_t test_override_predefine_map[TEST_PREDEFINE_COUNT];
  34. uint8_t test_override_define_map[256];
  35. uint8_t test_case_predefine_map[TEST_PREDEFINE_COUNT];
  36. const char *const *test_override_names;
  37. size_t test_override_count;
  38. const char *const test_predefine_names[TEST_PREDEFINE_COUNT]
  39. = TEST_PREDEFINE_NAMES;
  40. const char *const *test_define_names;
  41. size_t test_define_count;
  42. test_define_t test_predefine(size_t define) {
  43. for (int i = 0; i < TEST_DEFINE_LAYERS; i++) {
  44. if (test_defines[i]
  45. && test_predefine_maps[i]
  46. && test_predefine_maps[i][define] != 0xff) {
  47. return test_defines[i][test_predefine_maps[i][define]];
  48. }
  49. }
  50. fprintf(stderr, "error: undefined predefine %s\n",
  51. test_predefine_names[define]);
  52. assert(false);
  53. exit(-1);
  54. }
  55. test_define_t test_define(size_t define) {
  56. for (int i = 0; i < TEST_DEFINE_LAYERS; i++) {
  57. if (test_defines[i]
  58. && test_define_maps[i]
  59. && test_define_maps[i][define] != 0xff) {
  60. return test_defines[i][test_define_maps[i][define]];
  61. }
  62. }
  63. fprintf(stderr, "error: undefined define %s\n",
  64. test_define_names[define]);
  65. assert(false);
  66. exit(-1);
  67. }
  68. static void test_define_geometry(const struct test_geometry *geometry) {
  69. test_defines[2] = geometry->defines;
  70. }
  71. static void test_define_overrides(
  72. const char *const *override_names,
  73. const test_define_t *override_defines,
  74. size_t override_count) {
  75. test_defines[0] = override_defines;
  76. test_override_names = override_names;
  77. test_override_count = override_count;
  78. // map any predefines
  79. memset(test_override_predefine_map, 0xff, TEST_PREDEFINE_COUNT);
  80. for (size_t i = 0; i < override_count; i++) {
  81. for (size_t j = 0; j < TEST_PREDEFINE_COUNT; j++) {
  82. if (strcmp(override_names[i], test_predefine_names[j]) == 0) {
  83. test_override_predefine_map[j] = i;
  84. }
  85. }
  86. }
  87. test_predefine_maps[0] = test_override_predefine_map;
  88. }
  89. static void test_define_suite(const struct test_suite *suite) {
  90. test_define_names = suite->define_names;
  91. test_define_count = suite->define_count;
  92. // map any defines
  93. memset(test_override_define_map, 0xff, suite->define_count);
  94. for (size_t i = 0; i < test_override_count; i++) {
  95. for (size_t j = 0; j < suite->define_count; j++) {
  96. if (strcmp(test_override_names[i], suite->define_names[j]) == 0) {
  97. test_override_define_map[j] = i;
  98. }
  99. }
  100. }
  101. test_define_maps[0] = test_override_define_map;
  102. }
  103. static void test_define_case(
  104. const struct test_suite *suite,
  105. const struct test_case *case_) {
  106. (void)suite;
  107. // case_->define_map is already correct, but we need to do
  108. // some fixup for the predefine map
  109. test_define_maps[1] = case_->define_map;
  110. memset(test_case_predefine_map, 0xff, TEST_PREDEFINE_COUNT);
  111. for (size_t i = 0; i < test_define_count; i++) {
  112. for (size_t j = 0; j < TEST_PREDEFINE_COUNT; j++) {
  113. if (strcmp(test_define_names[i], test_predefine_names[j]) == 0) {
  114. test_case_predefine_map[j] = case_->define_map[i];
  115. }
  116. }
  117. }
  118. test_predefine_maps[1] = test_case_predefine_map;
  119. }
  120. static void test_define_perm(
  121. const struct test_suite *suite,
  122. const struct test_case *case_,
  123. size_t perm) {
  124. (void)suite;
  125. if (case_->defines) {
  126. test_defines[1] = case_->defines[perm];
  127. } else {
  128. test_defines[1] = NULL;
  129. }
  130. }
  131. // other miscellany
  132. static const char *test_suite = NULL;
  133. static const char *test_case = NULL;
  134. static size_t test_perm = -1;
  135. static const char *test_geometry = NULL;
  136. static test_types_t test_types = 0;
  137. static size_t test_skip = 0;
  138. static size_t test_count = -1;
  139. static size_t test_every = 1;
  140. static const char *test_persist = NULL;
  141. FILE *test_trace = NULL;
  142. // note, these skips are different than filtered tests
  143. static bool test_suite_skip(const struct test_suite *suite) {
  144. return (test_suite && strcmp(suite->name, test_suite) != 0)
  145. || (test_types && (suite->types & test_types) == 0);
  146. }
  147. static bool test_case_skip(const struct test_case *case_) {
  148. return (test_case && strcmp(case_->name, test_case) != 0)
  149. || (test_types && (case_->types & test_types) == 0);
  150. }
  151. static bool test_perm_skip(size_t perm) {
  152. size_t geom_perm = perm % TEST_GEOMETRY_COUNT;
  153. return (test_perm != (size_t)-1 && perm != test_perm)
  154. || (test_geometry && (strcmp(
  155. test_geometries[geom_perm].name,
  156. test_geometry) != 0));
  157. }
  158. static bool test_step_skip(size_t step) {
  159. return !(step >= test_skip
  160. && (step-test_skip) < test_count
  161. && (step-test_skip) % test_every == 0);
  162. }
  163. static void test_case_permcount(
  164. const struct test_suite *suite,
  165. const struct test_case *case_,
  166. size_t *perms,
  167. size_t *filtered) {
  168. size_t perms_ = 0;
  169. size_t filtered_ = 0;
  170. for (size_t perm = 0;
  171. perm < TEST_GEOMETRY_COUNT
  172. * case_->permutations;
  173. perm++) {
  174. if (test_perm_skip(perm)) {
  175. continue;
  176. }
  177. perms_ += 1;
  178. // setup defines
  179. size_t case_perm = perm / TEST_GEOMETRY_COUNT;
  180. size_t geom_perm = perm % TEST_GEOMETRY_COUNT;
  181. test_define_perm(suite, case_, case_perm);
  182. test_define_geometry(&test_geometries[geom_perm]);
  183. if (case_->filter) {
  184. if (!case_->filter(case_perm)) {
  185. continue;
  186. }
  187. }
  188. filtered_ += 1;
  189. }
  190. *perms += perms_;
  191. *filtered += filtered_;
  192. }
  193. // operations we can do
  194. static void summary(void) {
  195. printf("%-36s %7s %7s %7s %11s\n",
  196. "", "types", "suites", "cases", "perms");
  197. size_t cases = 0;
  198. test_types_t types = 0;
  199. size_t perms = 0;
  200. size_t filtered = 0;
  201. for (size_t i = 0; i < test_suite_count; i++) {
  202. if (test_suite_skip(test_suites[i])) {
  203. continue;
  204. }
  205. test_define_suite(test_suites[i]);
  206. for (size_t j = 0; j < test_suites[i]->case_count; j++) {
  207. if (test_case_skip(test_suites[i]->cases[j])) {
  208. continue;
  209. }
  210. test_define_case(test_suites[i], test_suites[i]->cases[j]);
  211. test_case_permcount(test_suites[i], test_suites[i]->cases[j],
  212. &perms, &filtered);
  213. }
  214. cases += test_suites[i]->case_count;
  215. types |= test_suites[i]->types;
  216. }
  217. char perm_buf[64];
  218. sprintf(perm_buf, "%zu/%zu", filtered, perms);
  219. char type_buf[64];
  220. sprintf(type_buf, "%s%s%s",
  221. (types & TEST_NORMAL) ? "n" : "",
  222. (types & TEST_REENTRANT) ? "r" : "",
  223. (types & TEST_VALGRIND) ? "V" : "");
  224. printf("%-36s %7s %7zu %7zu %11s\n",
  225. "TOTAL",
  226. type_buf,
  227. test_suite_count,
  228. cases,
  229. perm_buf);
  230. }
  231. static void list_suites(void) {
  232. printf("%-36s %7s %7s %11s\n", "suite", "types", "cases", "perms");
  233. for (size_t i = 0; i < test_suite_count; i++) {
  234. if (test_suite_skip(test_suites[i])) {
  235. continue;
  236. }
  237. test_define_suite(test_suites[i]);
  238. size_t perms = 0;
  239. size_t filtered = 0;
  240. for (size_t j = 0; j < test_suites[i]->case_count; j++) {
  241. if (test_case_skip(test_suites[i]->cases[j])) {
  242. continue;
  243. }
  244. test_define_case(test_suites[i], test_suites[i]->cases[j]);
  245. test_case_permcount(test_suites[i], test_suites[i]->cases[j],
  246. &perms, &filtered);
  247. }
  248. char perm_buf[64];
  249. sprintf(perm_buf, "%zu/%zu", filtered, perms);
  250. char type_buf[64];
  251. sprintf(type_buf, "%s%s%s",
  252. (test_suites[i]->types & TEST_NORMAL) ? "n" : "",
  253. (test_suites[i]->types & TEST_REENTRANT) ? "r" : "",
  254. (test_suites[i]->types & TEST_VALGRIND) ? "V" : "");
  255. printf("%-36s %7s %7zu %11s\n",
  256. test_suites[i]->id,
  257. type_buf,
  258. test_suites[i]->case_count,
  259. perm_buf);
  260. }
  261. }
  262. static void list_cases(void) {
  263. printf("%-36s %7s %11s\n", "case", "types", "perms");
  264. for (size_t i = 0; i < test_suite_count; i++) {
  265. if (test_suite_skip(test_suites[i])) {
  266. continue;
  267. }
  268. test_define_suite(test_suites[i]);
  269. for (size_t j = 0; j < test_suites[i]->case_count; j++) {
  270. if (test_case_skip(test_suites[i]->cases[j])) {
  271. continue;
  272. }
  273. test_define_case(test_suites[i], test_suites[i]->cases[j]);
  274. size_t perms = 0;
  275. size_t filtered = 0;
  276. test_case_permcount(test_suites[i], test_suites[i]->cases[j],
  277. &perms, &filtered);
  278. test_types_t types = test_suites[i]->cases[j]->types;
  279. char perm_buf[64];
  280. sprintf(perm_buf, "%zu/%zu", filtered, perms);
  281. char type_buf[64];
  282. sprintf(type_buf, "%s%s%s",
  283. (types & TEST_NORMAL) ? "n" : "",
  284. (types & TEST_REENTRANT) ? "r" : "",
  285. (types & TEST_VALGRIND) ? "V" : "");
  286. printf("%-36s %7s %11s\n",
  287. test_suites[i]->cases[j]->id,
  288. type_buf,
  289. perm_buf);
  290. }
  291. }
  292. }
  293. static void list_paths(void) {
  294. for (size_t i = 0; i < test_suite_count; i++) {
  295. if (test_suite_skip(test_suites[i])) {
  296. continue;
  297. }
  298. for (size_t j = 0; j < test_suites[i]->case_count; j++) {
  299. if (test_case_skip(test_suites[i]->cases[j])) {
  300. continue;
  301. }
  302. printf("%-36s %-36s\n",
  303. test_suites[i]->cases[j]->id,
  304. test_suites[i]->cases[j]->path);
  305. }
  306. }
  307. }
  308. static void list_defines(void) {
  309. for (size_t i = 0; i < test_suite_count; i++) {
  310. if (test_suite_skip(test_suites[i])) {
  311. continue;
  312. }
  313. test_define_suite(test_suites[i]);
  314. for (size_t j = 0; j < test_suites[i]->case_count; j++) {
  315. if (test_case_skip(test_suites[i]->cases[j])) {
  316. continue;
  317. }
  318. test_define_case(test_suites[i], test_suites[i]->cases[j]);
  319. for (size_t perm = 0;
  320. perm < TEST_GEOMETRY_COUNT
  321. * test_suites[i]->cases[j]->permutations;
  322. perm++) {
  323. if (test_perm_skip(perm)) {
  324. continue;
  325. }
  326. // setup defines
  327. size_t case_perm = perm / TEST_GEOMETRY_COUNT;
  328. size_t geom_perm = perm % TEST_GEOMETRY_COUNT;
  329. test_define_perm(test_suites[i],
  330. test_suites[i]->cases[j], case_perm);
  331. test_define_geometry(&test_geometries[geom_perm]);
  332. // print the case
  333. char id_buf[256];
  334. sprintf(id_buf, "%s#%zu", test_suites[i]->cases[j]->id, perm);
  335. printf("%-36s ", id_buf);
  336. // special case for the current geometry
  337. printf("GEOMETRY=%s ", test_geometries[geom_perm].name);
  338. // print each define
  339. for (size_t k = 0; k < test_suites[i]->define_count; k++) {
  340. if (test_suites[i]->cases[j]->define_map
  341. && test_suites[i]->cases[j]->define_map[k]
  342. != 0xff) {
  343. printf("%s=%jd ",
  344. test_suites[i]->define_names[k],
  345. test_define(k));
  346. }
  347. }
  348. printf("\n");
  349. }
  350. }
  351. }
  352. }
  353. static void list_geometries(void) {
  354. for (size_t i = 0; i < TEST_GEOMETRY_COUNT; i++) {
  355. if (test_geometry && strcmp(
  356. test_geometries[i].name,
  357. test_geometry) != 0) {
  358. continue;
  359. }
  360. test_define_geometry(&test_geometries[i]);
  361. printf("%-36s ", test_geometries[i].name);
  362. // print each define
  363. for (size_t k = 0; k < TEST_PREDEFINE_COUNT; k++) {
  364. if (test_predefine_maps[2][k] != 0xff) {
  365. printf("%s=%jd ",
  366. test_predefine_names[k],
  367. test_predefine(k));
  368. }
  369. }
  370. printf("\n");
  371. }
  372. }
  373. static void list_defaults(void) {
  374. printf("%-36s ", "defaults");
  375. // print each define
  376. for (size_t k = 0; k < TEST_PREDEFINE_COUNT; k++) {
  377. if (test_predefine_maps[3][k] != 0xff) {
  378. printf("%s=%jd ",
  379. test_predefine_names[k],
  380. test_predefine(k));
  381. }
  382. }
  383. printf("\n");
  384. }
  385. static void run(void) {
  386. size_t step = 0;
  387. for (size_t i = 0; i < test_suite_count; i++) {
  388. if (test_suite_skip(test_suites[i])) {
  389. continue;
  390. }
  391. test_define_suite(test_suites[i]);
  392. for (size_t j = 0; j < test_suites[i]->case_count; j++) {
  393. if (test_case_skip(test_suites[i]->cases[j])) {
  394. continue;
  395. }
  396. test_define_case(test_suites[i], test_suites[i]->cases[j]);
  397. for (size_t perm = 0;
  398. perm < TEST_GEOMETRY_COUNT
  399. * test_suites[i]->cases[j]->permutations;
  400. perm++) {
  401. if (test_perm_skip(perm)) {
  402. continue;
  403. }
  404. if (test_step_skip(step)) {
  405. step += 1;
  406. continue;
  407. }
  408. step += 1;
  409. // setup defines
  410. size_t case_perm = perm / TEST_GEOMETRY_COUNT;
  411. size_t geom_perm = perm % TEST_GEOMETRY_COUNT;
  412. test_define_perm(test_suites[i],
  413. test_suites[i]->cases[j], case_perm);
  414. test_define_geometry(&test_geometries[geom_perm]);
  415. // filter?
  416. if (test_suites[i]->cases[j]->filter) {
  417. if (!test_suites[i]->cases[j]->filter(case_perm)) {
  418. printf("skipped %s#%zu\n",
  419. test_suites[i]->cases[j]->id,
  420. perm);
  421. continue;
  422. }
  423. }
  424. // create block device and configuration
  425. lfs_testbd_t bd;
  426. struct lfs_config cfg = {
  427. .context = &bd,
  428. .read = lfs_testbd_read,
  429. .prog = lfs_testbd_prog,
  430. .erase = lfs_testbd_erase,
  431. .sync = lfs_testbd_sync,
  432. .read_size = READ_SIZE,
  433. .prog_size = PROG_SIZE,
  434. .block_size = BLOCK_SIZE,
  435. .block_count = BLOCK_COUNT,
  436. .block_cycles = BLOCK_CYCLES,
  437. .cache_size = CACHE_SIZE,
  438. .lookahead_size = LOOKAHEAD_SIZE,
  439. };
  440. struct lfs_testbd_config bdcfg = {
  441. .erase_value = ERASE_VALUE,
  442. .erase_cycles = ERASE_CYCLES,
  443. .badblock_behavior = BADBLOCK_BEHAVIOR,
  444. .power_cycles = 0,
  445. };
  446. int err = lfs_testbd_createcfg(&cfg, test_persist, &bdcfg);
  447. if (err) {
  448. fprintf(stderr, "error: "
  449. "could not create block device: %d\n", err);
  450. exit(-1);
  451. }
  452. // run the test
  453. printf("running %s#%zu\n", test_suites[i]->cases[j]->id, perm);
  454. test_suites[i]->cases[j]->run(&cfg, case_perm);
  455. printf("finished %s#%zu\n", test_suites[i]->cases[j]->id, perm);
  456. // cleanup
  457. err = lfs_testbd_destroy(&cfg);
  458. if (err) {
  459. fprintf(stderr, "error: "
  460. "could not destroy block device: %d\n", err);
  461. exit(-1);
  462. }
  463. }
  464. }
  465. }
  466. }
  467. // option handling
  468. enum opt_flags {
  469. OPT_HELP = 'h',
  470. OPT_SUMMARY = 'Y',
  471. OPT_LIST_SUITES = 'l',
  472. OPT_LIST_CASES = 'L',
  473. OPT_LIST_PATHS = 1,
  474. OPT_LIST_DEFINES = 2,
  475. OPT_LIST_GEOMETRIES = 3,
  476. OPT_LIST_DEFAULTS = 4,
  477. OPT_DEFINE = 'D',
  478. OPT_GEOMETRY = 'G',
  479. OPT_NORMAL = 'n',
  480. OPT_REENTRANT = 'r',
  481. OPT_VALGRIND = 'V',
  482. OPT_SKIP = 5,
  483. OPT_COUNT = 6,
  484. OPT_EVERY = 7,
  485. OPT_PERSIST = 'p',
  486. OPT_TRACE = 't',
  487. };
  488. const char *short_opts = "hYlLD:G:nrVp:t:";
  489. const struct option long_opts[] = {
  490. {"help", no_argument, NULL, OPT_HELP},
  491. {"summary", no_argument, NULL, OPT_SUMMARY},
  492. {"list-suites", no_argument, NULL, OPT_LIST_SUITES},
  493. {"list-cases", no_argument, NULL, OPT_LIST_CASES},
  494. {"list-paths", no_argument, NULL, OPT_LIST_PATHS},
  495. {"list-defines", no_argument, NULL, OPT_LIST_DEFINES},
  496. {"list-geometries", no_argument, NULL, OPT_LIST_GEOMETRIES},
  497. {"list-defaults", no_argument, NULL, OPT_LIST_DEFAULTS},
  498. {"define", required_argument, NULL, OPT_DEFINE},
  499. {"geometry", required_argument, NULL, OPT_GEOMETRY},
  500. {"normal", no_argument, NULL, OPT_NORMAL},
  501. {"reentrant", no_argument, NULL, OPT_REENTRANT},
  502. {"valgrind", no_argument, NULL, OPT_VALGRIND},
  503. {"skip", required_argument, NULL, OPT_SKIP},
  504. {"count", required_argument, NULL, OPT_COUNT},
  505. {"every", required_argument, NULL, OPT_EVERY},
  506. {"persist", required_argument, NULL, OPT_PERSIST},
  507. {"trace", required_argument, NULL, OPT_TRACE},
  508. {NULL, 0, NULL, 0},
  509. };
  510. const char *const help_text[] = {
  511. "Show this help message.",
  512. "Show quick summary.",
  513. "List test suites.",
  514. "List test cases.",
  515. "List the path for each test case.",
  516. "List the defines for each test permutation.",
  517. "List the disk geometries used for testing.",
  518. "List the default defines in this test-runner.",
  519. "Override a test define.",
  520. "Filter by geometry.",
  521. "Filter for normal tests. Can be combined.",
  522. "Filter for reentrant tests. Can be combined.",
  523. "Filter for Valgrind tests. Can be combined.",
  524. "Skip the first n tests.",
  525. "Stop after n tests.",
  526. "Only run every n tests, calculated after --skip and --stop.",
  527. "Persist the disk to this file.",
  528. "Redirect trace output to this file.",
  529. };
  530. int main(int argc, char **argv) {
  531. void (*op)(void) = run;
  532. static const char **override_names = NULL;
  533. static test_define_t *override_defines = NULL;
  534. static size_t override_count = 0;
  535. static size_t override_cap = 0;
  536. // parse options
  537. while (true) {
  538. int c = getopt_long(argc, argv, short_opts, long_opts, NULL);
  539. switch (c) {
  540. // generate help message
  541. case OPT_HELP: {
  542. printf("usage: %s [options] [test_id]\n", argv[0]);
  543. printf("\n");
  544. printf("options:\n");
  545. size_t i = 0;
  546. while (long_opts[i].name) {
  547. size_t indent;
  548. if (long_opts[i].has_arg == no_argument) {
  549. if (long_opts[i].val >= '0' && long_opts[i].val < 'z') {
  550. indent = printf(" -%c, --%s ",
  551. long_opts[i].val,
  552. long_opts[i].name);
  553. } else {
  554. indent = printf(" --%s ",
  555. long_opts[i].name);
  556. }
  557. } else {
  558. if (long_opts[i].val >= '0' && long_opts[i].val < 'z') {
  559. indent = printf(" -%c %s, --%s %s ",
  560. long_opts[i].val,
  561. long_opts[i].name,
  562. long_opts[i].name,
  563. long_opts[i].name);
  564. } else {
  565. indent = printf(" --%s %s ",
  566. long_opts[i].name,
  567. long_opts[i].name);
  568. }
  569. }
  570. // a quick, hacky, byte-level method for text wrapping
  571. size_t len = strlen(help_text[i]);
  572. size_t j = 0;
  573. if (indent < 24) {
  574. printf("%*s %.80s\n",
  575. (int)(24-1-indent),
  576. "",
  577. &help_text[i][j]);
  578. j += 80;
  579. } else {
  580. printf("\n");
  581. }
  582. while (j < len) {
  583. printf("%24s%.80s\n", "", &help_text[i][j]);
  584. j += 80;
  585. }
  586. i += 1;
  587. }
  588. printf("\n");
  589. exit(0);
  590. }
  591. // summary/list flags
  592. case OPT_SUMMARY:
  593. op = summary;
  594. break;
  595. case OPT_LIST_SUITES:
  596. op = list_suites;
  597. break;
  598. case OPT_LIST_CASES:
  599. op = list_cases;
  600. break;
  601. case OPT_LIST_PATHS:
  602. op = list_paths;
  603. break;
  604. case OPT_LIST_DEFINES:
  605. op = list_defines;
  606. break;
  607. case OPT_LIST_GEOMETRIES:
  608. op = list_geometries;
  609. break;
  610. case OPT_LIST_DEFAULTS:
  611. op = list_defaults;
  612. break;
  613. // configuration
  614. case OPT_DEFINE: {
  615. // special case for -DGEOMETRY=<name>, we treat this the same
  616. // as --geometry=<name>
  617. if (strncmp(optarg, "GEOMETRY=", strlen("GEOMETRY=")) == 0) {
  618. test_geometry = &optarg[strlen("GEOMETRY=")];
  619. break;
  620. }
  621. // realloc if necessary
  622. override_count += 1;
  623. if (override_count > override_cap) {
  624. override_cap = (2*override_cap > 4) ? 2*override_cap : 4;
  625. override_names = realloc(override_names, override_cap
  626. * sizeof(const char *));
  627. override_defines = realloc(override_defines, override_cap
  628. * sizeof(test_define_t));
  629. }
  630. // parse into string key/test_define_t value, cannibalizing the
  631. // arg in the process
  632. char *sep = strchr(optarg, '=');
  633. char *parsed = NULL;
  634. if (!sep) {
  635. goto invalid_define;
  636. }
  637. override_defines[override_count-1]
  638. = strtoumax(sep+1, &parsed, 0);
  639. if (parsed == sep+1) {
  640. goto invalid_define;
  641. }
  642. override_names[override_count-1] = optarg;
  643. *sep = '\0';
  644. break;
  645. invalid_define:
  646. fprintf(stderr, "error: invalid define: %s\n", optarg);
  647. exit(-1);
  648. }
  649. case OPT_GEOMETRY:
  650. test_geometry = optarg;
  651. break;
  652. case OPT_NORMAL:
  653. test_types |= TEST_NORMAL;
  654. break;
  655. case OPT_REENTRANT:
  656. test_types |= TEST_REENTRANT;
  657. break;
  658. case OPT_VALGRIND:
  659. test_types |= TEST_VALGRIND;
  660. break;
  661. case OPT_SKIP: {
  662. char *parsed = NULL;
  663. test_skip = strtoumax(optarg, &parsed, 0);
  664. if (parsed == optarg) {
  665. fprintf(stderr, "error: invalid skip: %s\n", optarg);
  666. exit(-1);
  667. }
  668. break;
  669. }
  670. case OPT_COUNT: {
  671. char *parsed = NULL;
  672. test_count = strtoumax(optarg, &parsed, 0);
  673. if (parsed == optarg) {
  674. fprintf(stderr, "error: invalid count: %s\n", optarg);
  675. exit(-1);
  676. }
  677. break;
  678. }
  679. case OPT_EVERY: {
  680. char *parsed = NULL;
  681. test_every = strtoumax(optarg, &parsed, 0);
  682. if (parsed == optarg) {
  683. fprintf(stderr, "error: invalid every: %s\n", optarg);
  684. exit(-1);
  685. }
  686. break;
  687. }
  688. case OPT_PERSIST:
  689. test_persist = optarg;
  690. break;
  691. case OPT_TRACE:
  692. if (strcmp(optarg, "-") == 0) {
  693. test_trace = stdout;
  694. } else {
  695. test_trace = fopen(optarg, "w");
  696. if (!test_trace) {
  697. fprintf(stderr, "error: could not open for trace: %d\n",
  698. -errno);
  699. exit(-1);
  700. }
  701. }
  702. break;
  703. // done parsing
  704. case -1:
  705. goto getopt_done;
  706. // unknown arg, getopt prints a message for us
  707. default:
  708. exit(-1);
  709. }
  710. }
  711. getopt_done: ;
  712. // parse test identifier, if any, cannibalizing the arg in the process
  713. if (argc > optind) {
  714. if (argc - optind > 1) {
  715. fprintf(stderr, "error: more than one test identifier\n");
  716. exit(-1);
  717. }
  718. // parse suite
  719. char *suite = argv[optind];
  720. char *case_ = strchr(suite, '#');
  721. if (case_) {
  722. *case_ = '\0';
  723. case_ += 1;
  724. // parse case
  725. char *perm = strchr(case_, '#');
  726. if (perm) {
  727. *perm = '\0';
  728. perm += 1;
  729. char *parsed = NULL;
  730. test_perm = strtoumax(perm, &parsed, 10);
  731. if (parsed == perm) {
  732. fprintf(stderr, "error: could not parse test identifier\n");
  733. exit(-1);
  734. }
  735. }
  736. test_case = case_;
  737. }
  738. // remove optional path and .toml suffix
  739. char *slash = strrchr(suite, '/');
  740. if (slash) {
  741. suite = slash+1;
  742. }
  743. size_t suite_len = strlen(suite);
  744. if (suite_len > 5 && strcmp(&suite[suite_len-5], ".toml") == 0) {
  745. suite[suite_len-5] = '\0';
  746. }
  747. test_suite = suite;
  748. }
  749. // register overrides
  750. test_define_overrides(override_names, override_defines, override_count);
  751. // do the thing
  752. op();
  753. // cleanup (need to be done for valgrind testing)
  754. free(override_names);
  755. free(override_defines);
  756. }