bench_runner.c 66 KB

12345678910111213141516171819202122232425262728293031323334353637383940414243444546474849505152535455565758596061626364656667686970717273747576777879808182838485868788899091929394959697989910010110210310410510610710810911011111211311411511611711811912012112212312412512612712812913013113213313413513613713813914014114214314414514614714814915015115215315415515615715815916016116216316416516616716816917017117217317417517617717817918018118218318418518618718818919019119219319419519619719819920020120220320420520620720820921021121221321421521621721821922022122222322422522622722822923023123223323423523623723823924024124224324424524624724824925025125225325425525625725825926026126226326426526626726826927027127227327427527627727827928028128228328428528628728828929029129229329429529629729829930030130230330430530630730830931031131231331431531631731831932032132232332432532632732832933033133233333433533633733833934034134234334434534634734834935035135235335435535635735835936036136236336436536636736836937037137237337437537637737837938038138238338438538638738838939039139239339439539639739839940040140240340440540640740840941041141241341441541641741841942042142242342442542642742842943043143243343443543643743843944044144244344444544644744844945045145245345445545645745845946046146246346446546646746846947047147247347447547647747847948048148248348448548648748848949049149249349449549649749849950050150250350450550650750850951051151251351451551651751851952052152252352452552652752852953053153253353453553653753853954054154254354454554654754854955055155255355455555655755855956056156256356456556656756856957057157257357457557657757857958058158258358458558658758858959059159259359459559659759859960060160260360460560660760860961061161261361461561661761861962062162262362462562662762862963063163263363463563663763863964064164264364464564664764864965065165265365465565665765865966066166266366466566666766866967067167267367467567667767867968068168268368468568668768868969069169269369469569669769869970070170270370470570670770870971071171271371471571671771871972072172272372472572672772872973073173273373473573673773873974074174274374474574674774874975075175275375475575675775875976076176276376476576676776876977077177277377477577677777877978078178278378478578678778878979079179279379479579679779879980080180280380480580680780880981081181281381481581681781881982082182282382482582682782882983083183283383483583683783883984084184284384484584684784884985085185285385485585685785885986086186286386486586686786886987087187287387487587687787887988088188288388488588688788888989089189289389489589689789889990090190290390490590690790890991091191291391491591691791891992092192292392492592692792892993093193293393493593693793893994094194294394494594694794894995095195295395495595695795895996096196296396496596696796896997097197297397497597697797897998098198298398498598698798898999099199299399499599699799899910001001100210031004100510061007100810091010101110121013101410151016101710181019102010211022102310241025102610271028102910301031103210331034103510361037103810391040104110421043104410451046104710481049105010511052105310541055105610571058105910601061106210631064106510661067106810691070107110721073107410751076107710781079108010811082108310841085108610871088108910901091109210931094109510961097109810991100110111021103110411051106110711081109111011111112111311141115111611171118111911201121112211231124112511261127112811291130113111321133113411351136113711381139114011411142114311441145114611471148114911501151115211531154115511561157115811591160116111621163116411651166116711681169117011711172117311741175117611771178117911801181118211831184118511861187118811891190119111921193119411951196119711981199120012011202120312041205120612071208120912101211121212131214121512161217121812191220122112221223122412251226122712281229123012311232123312341235123612371238123912401241124212431244124512461247124812491250125112521253125412551256125712581259126012611262126312641265126612671268126912701271127212731274127512761277127812791280128112821283128412851286128712881289129012911292129312941295129612971298129913001301130213031304130513061307130813091310131113121313131413151316131713181319132013211322132313241325132613271328132913301331133213331334133513361337133813391340134113421343134413451346134713481349135013511352135313541355135613571358135913601361136213631364136513661367136813691370137113721373137413751376137713781379138013811382138313841385138613871388138913901391139213931394139513961397139813991400140114021403140414051406140714081409141014111412141314141415141614171418141914201421142214231424142514261427142814291430143114321433143414351436143714381439144014411442144314441445144614471448144914501451145214531454145514561457145814591460146114621463146414651466146714681469147014711472147314741475147614771478147914801481148214831484148514861487148814891490149114921493149414951496149714981499150015011502150315041505150615071508150915101511151215131514151515161517151815191520152115221523152415251526152715281529153015311532153315341535153615371538153915401541154215431544154515461547154815491550155115521553155415551556155715581559156015611562156315641565156615671568156915701571157215731574157515761577157815791580158115821583158415851586158715881589159015911592159315941595159615971598159916001601160216031604160516061607160816091610161116121613161416151616161716181619162016211622162316241625162616271628162916301631163216331634163516361637163816391640164116421643164416451646164716481649165016511652165316541655165616571658165916601661166216631664166516661667166816691670167116721673167416751676167716781679168016811682168316841685168616871688168916901691169216931694169516961697169816991700170117021703170417051706170717081709171017111712171317141715171617171718171917201721172217231724172517261727172817291730173117321733173417351736173717381739174017411742174317441745174617471748174917501751175217531754175517561757175817591760176117621763176417651766176717681769177017711772177317741775177617771778177917801781178217831784178517861787178817891790179117921793179417951796179717981799180018011802180318041805180618071808180918101811181218131814181518161817181818191820182118221823182418251826182718281829183018311832183318341835183618371838183918401841184218431844184518461847184818491850185118521853185418551856185718581859186018611862186318641865186618671868186918701871187218731874187518761877187818791880188118821883188418851886188718881889189018911892189318941895189618971898189919001901190219031904190519061907190819091910191119121913191419151916191719181919192019211922192319241925192619271928192919301931193219331934193519361937193819391940194119421943194419451946194719481949195019511952195319541955195619571958195919601961196219631964196519661967196819691970197119721973197419751976197719781979198019811982198319841985198619871988198919901991199219931994199519961997199819992000200120022003200420052006200720082009201020112012201320142015201620172018201920202021202220232024202520262027202820292030203120322033203420352036203720382039204020412042204320442045204620472048204920502051
  1. /*
  2. * Runner for littlefs benchmarks
  3. *
  4. * Copyright (c) 2022, The littlefs authors.
  5. * SPDX-License-Identifier: BSD-3-Clause
  6. */
  7. #ifndef _POSIX_C_SOURCE
  8. #define _POSIX_C_SOURCE 199309L
  9. #endif
  10. #include "runners/bench_runner.h"
  11. #include "bd/lfs_emubd.h"
  12. #include <getopt.h>
  13. #include <sys/types.h>
  14. #include <errno.h>
  15. #include <setjmp.h>
  16. #include <fcntl.h>
  17. #include <stdarg.h>
  18. #include <stdio.h>
  19. #include <unistd.h>
  20. #include <execinfo.h>
  21. #include <time.h>
  22. // some helpers
  23. // append to an array with amortized doubling
  24. void *mappend(void **p,
  25. size_t size,
  26. size_t *count,
  27. size_t *capacity) {
  28. uint8_t *p_ = *p;
  29. size_t count_ = *count;
  30. size_t capacity_ = *capacity;
  31. count_ += 1;
  32. if (count_ > capacity_) {
  33. capacity_ = (2*capacity_ < 4) ? 4 : 2*capacity_;
  34. p_ = realloc(p_, capacity_*size);
  35. if (!p_) {
  36. return NULL;
  37. }
  38. }
  39. *p = p_;
  40. *count = count_;
  41. *capacity = capacity_;
  42. return &p_[(count_-1)*size];
  43. }
  44. // a quick self-terminating text-safe varint scheme
  45. static void leb16_print(uintmax_t x) {
  46. // allow 'w' to indicate negative numbers
  47. if ((intmax_t)x < 0) {
  48. printf("w");
  49. x = -x;
  50. }
  51. while (true) {
  52. char nibble = (x & 0xf) | (x > 0xf ? 0x10 : 0);
  53. printf("%c", (nibble < 10) ? '0'+nibble : 'a'+nibble-10);
  54. if (x <= 0xf) {
  55. break;
  56. }
  57. x >>= 4;
  58. }
  59. }
  60. static uintmax_t leb16_parse(const char *s, char **tail) {
  61. bool neg = false;
  62. uintmax_t x = 0;
  63. if (tail) {
  64. *tail = (char*)s;
  65. }
  66. if (s[0] == 'w') {
  67. neg = true;
  68. s = s+1;
  69. }
  70. size_t i = 0;
  71. while (true) {
  72. uintmax_t nibble = s[i];
  73. if (nibble >= '0' && nibble <= '9') {
  74. nibble = nibble - '0';
  75. } else if (nibble >= 'a' && nibble <= 'v') {
  76. nibble = nibble - 'a' + 10;
  77. } else {
  78. // invalid?
  79. return 0;
  80. }
  81. x |= (nibble & 0xf) << (4*i);
  82. i += 1;
  83. if (!(nibble & 0x10)) {
  84. s = s + i;
  85. break;
  86. }
  87. }
  88. if (tail) {
  89. *tail = (char*)s;
  90. }
  91. return neg ? -x : x;
  92. }
  93. // bench_runner types
  94. typedef struct bench_geometry {
  95. const char *name;
  96. bench_define_t defines[BENCH_GEOMETRY_DEFINE_COUNT];
  97. } bench_geometry_t;
  98. typedef struct bench_id {
  99. const char *name;
  100. const bench_define_t *defines;
  101. size_t define_count;
  102. } bench_id_t;
  103. // bench suites are linked into a custom ld section
  104. extern struct bench_suite __start__bench_suites;
  105. extern struct bench_suite __stop__bench_suites;
  106. const struct bench_suite *bench_suites = &__start__bench_suites;
  107. #define BENCH_SUITE_COUNT \
  108. ((size_t)(&__stop__bench_suites - &__start__bench_suites))
  109. // bench define management
  110. typedef struct bench_define_map {
  111. const bench_define_t *defines;
  112. size_t count;
  113. } bench_define_map_t;
  114. typedef struct bench_define_names {
  115. const char *const *names;
  116. size_t count;
  117. } bench_define_names_t;
  118. intmax_t bench_define_lit(void *data) {
  119. return (intptr_t)data;
  120. }
  121. #define BENCH_CONST(x) {bench_define_lit, (void*)(uintptr_t)(x)}
  122. #define BENCH_LIT(x) ((bench_define_t)BENCH_CONST(x))
  123. #define BENCH_DEF(k, v) \
  124. intmax_t bench_define_##k(void *data) { \
  125. (void)data; \
  126. return v; \
  127. }
  128. BENCH_IMPLICIT_DEFINES
  129. #undef BENCH_DEF
  130. #define BENCH_DEFINE_MAP_OVERRIDE 0
  131. #define BENCH_DEFINE_MAP_EXPLICIT 1
  132. #define BENCH_DEFINE_MAP_PERMUTATION 2
  133. #define BENCH_DEFINE_MAP_GEOMETRY 3
  134. #define BENCH_DEFINE_MAP_IMPLICIT 4
  135. #define BENCH_DEFINE_MAP_COUNT 5
  136. bench_define_map_t bench_define_maps[BENCH_DEFINE_MAP_COUNT] = {
  137. [BENCH_DEFINE_MAP_IMPLICIT] = {
  138. (const bench_define_t[BENCH_IMPLICIT_DEFINE_COUNT]) {
  139. #define BENCH_DEF(k, v) \
  140. [k##_i] = {bench_define_##k, NULL},
  141. BENCH_IMPLICIT_DEFINES
  142. #undef BENCH_DEF
  143. },
  144. BENCH_IMPLICIT_DEFINE_COUNT,
  145. },
  146. };
  147. #define BENCH_DEFINE_NAMES_SUITE 0
  148. #define BENCH_DEFINE_NAMES_IMPLICIT 1
  149. #define BENCH_DEFINE_NAMES_COUNT 2
  150. bench_define_names_t bench_define_names[BENCH_DEFINE_NAMES_COUNT] = {
  151. [BENCH_DEFINE_NAMES_IMPLICIT] = {
  152. (const char *const[BENCH_IMPLICIT_DEFINE_COUNT]){
  153. #define BENCH_DEF(k, v) \
  154. [k##_i] = #k,
  155. BENCH_IMPLICIT_DEFINES
  156. #undef BENCH_DEF
  157. },
  158. BENCH_IMPLICIT_DEFINE_COUNT,
  159. },
  160. };
  161. intmax_t *bench_define_cache;
  162. size_t bench_define_cache_count;
  163. unsigned *bench_define_cache_mask;
  164. const char *bench_define_name(size_t define) {
  165. // lookup in our bench names
  166. for (size_t i = 0; i < BENCH_DEFINE_NAMES_COUNT; i++) {
  167. if (define < bench_define_names[i].count
  168. && bench_define_names[i].names
  169. && bench_define_names[i].names[define]) {
  170. return bench_define_names[i].names[define];
  171. }
  172. }
  173. return NULL;
  174. }
  175. bool bench_define_ispermutation(size_t define) {
  176. // is this define specific to the permutation?
  177. for (size_t i = 0; i < BENCH_DEFINE_MAP_IMPLICIT; i++) {
  178. if (define < bench_define_maps[i].count
  179. && bench_define_maps[i].defines[define].cb) {
  180. return true;
  181. }
  182. }
  183. return false;
  184. }
  185. intmax_t bench_define(size_t define) {
  186. // is the define in our cache?
  187. if (define < bench_define_cache_count
  188. && (bench_define_cache_mask[define/(8*sizeof(unsigned))]
  189. & (1 << (define%(8*sizeof(unsigned)))))) {
  190. return bench_define_cache[define];
  191. }
  192. // lookup in our bench defines
  193. for (size_t i = 0; i < BENCH_DEFINE_MAP_COUNT; i++) {
  194. if (define < bench_define_maps[i].count
  195. && bench_define_maps[i].defines[define].cb) {
  196. intmax_t v = bench_define_maps[i].defines[define].cb(
  197. bench_define_maps[i].defines[define].data);
  198. // insert into cache!
  199. bench_define_cache[define] = v;
  200. bench_define_cache_mask[define / (8*sizeof(unsigned))]
  201. |= 1 << (define%(8*sizeof(unsigned)));
  202. return v;
  203. }
  204. }
  205. return 0;
  206. // not found?
  207. const char *name = bench_define_name(define);
  208. fprintf(stderr, "error: undefined define %s (%zd)\n",
  209. name ? name : "(unknown)",
  210. define);
  211. assert(false);
  212. exit(-1);
  213. }
  214. void bench_define_flush(void) {
  215. // clear cache between permutations
  216. memset(bench_define_cache_mask, 0,
  217. sizeof(unsigned)*(
  218. (bench_define_cache_count+(8*sizeof(unsigned))-1)
  219. / (8*sizeof(unsigned))));
  220. }
  221. // geometry updates
  222. const bench_geometry_t *bench_geometry = NULL;
  223. void bench_define_geometry(const bench_geometry_t *geometry) {
  224. bench_define_maps[BENCH_DEFINE_MAP_GEOMETRY] = (bench_define_map_t){
  225. geometry->defines, BENCH_GEOMETRY_DEFINE_COUNT};
  226. }
  227. // override updates
  228. typedef struct bench_override {
  229. const char *name;
  230. const intmax_t *defines;
  231. size_t permutations;
  232. } bench_override_t;
  233. const bench_override_t *bench_overrides = NULL;
  234. size_t bench_override_count = 0;
  235. bench_define_t *bench_override_defines = NULL;
  236. size_t bench_override_define_count = 0;
  237. size_t bench_override_define_permutations = 1;
  238. size_t bench_override_define_capacity = 0;
  239. // suite/perm updates
  240. void bench_define_suite(const struct bench_suite *suite) {
  241. bench_define_names[BENCH_DEFINE_NAMES_SUITE] = (bench_define_names_t){
  242. suite->define_names, suite->define_count};
  243. // make sure our cache is large enough
  244. if (lfs_max(suite->define_count, BENCH_IMPLICIT_DEFINE_COUNT)
  245. > bench_define_cache_count) {
  246. // align to power of two to avoid any superlinear growth
  247. size_t ncount = 1 << lfs_npw2(
  248. lfs_max(suite->define_count, BENCH_IMPLICIT_DEFINE_COUNT));
  249. bench_define_cache = realloc(bench_define_cache, ncount*sizeof(intmax_t));
  250. bench_define_cache_mask = realloc(bench_define_cache_mask,
  251. sizeof(unsigned)*(
  252. (ncount+(8*sizeof(unsigned))-1)
  253. / (8*sizeof(unsigned))));
  254. bench_define_cache_count = ncount;
  255. }
  256. // map any overrides
  257. if (bench_override_count > 0) {
  258. // first figure out the total size of override permutations
  259. size_t count = 0;
  260. size_t permutations = 1;
  261. for (size_t i = 0; i < bench_override_count; i++) {
  262. for (size_t d = 0;
  263. d < lfs_max(
  264. suite->define_count,
  265. BENCH_IMPLICIT_DEFINE_COUNT);
  266. d++) {
  267. // define name match?
  268. const char *name = bench_define_name(d);
  269. if (name && strcmp(name, bench_overrides[i].name) == 0) {
  270. count = lfs_max(count, d+1);
  271. permutations *= bench_overrides[i].permutations;
  272. break;
  273. }
  274. }
  275. }
  276. bench_override_define_count = count;
  277. bench_override_define_permutations = permutations;
  278. // make sure our override arrays are big enough
  279. if (count * permutations > bench_override_define_capacity) {
  280. // align to power of two to avoid any superlinear growth
  281. size_t ncapacity = 1 << lfs_npw2(count * permutations);
  282. bench_override_defines = realloc(
  283. bench_override_defines,
  284. sizeof(bench_define_t)*ncapacity);
  285. bench_override_define_capacity = ncapacity;
  286. }
  287. // zero unoverridden defines
  288. memset(bench_override_defines, 0,
  289. sizeof(bench_define_t) * count * permutations);
  290. // compute permutations
  291. size_t p = 1;
  292. for (size_t i = 0; i < bench_override_count; i++) {
  293. for (size_t d = 0;
  294. d < lfs_max(
  295. suite->define_count,
  296. BENCH_IMPLICIT_DEFINE_COUNT);
  297. d++) {
  298. // define name match?
  299. const char *name = bench_define_name(d);
  300. if (name && strcmp(name, bench_overrides[i].name) == 0) {
  301. // scatter the define permutations based on already
  302. // seen permutations
  303. for (size_t j = 0; j < permutations; j++) {
  304. bench_override_defines[j*count + d] = BENCH_LIT(
  305. bench_overrides[i].defines[(j/p)
  306. % bench_overrides[i].permutations]);
  307. }
  308. // keep track of how many permutations we've seen so far
  309. p *= bench_overrides[i].permutations;
  310. break;
  311. }
  312. }
  313. }
  314. }
  315. }
  316. void bench_define_perm(
  317. const struct bench_suite *suite,
  318. const struct bench_case *case_,
  319. size_t perm) {
  320. if (case_->defines) {
  321. bench_define_maps[BENCH_DEFINE_MAP_PERMUTATION] = (bench_define_map_t){
  322. case_->defines + perm*suite->define_count,
  323. suite->define_count};
  324. } else {
  325. bench_define_maps[BENCH_DEFINE_MAP_PERMUTATION] = (bench_define_map_t){
  326. NULL, 0};
  327. }
  328. }
  329. void bench_define_override(size_t perm) {
  330. bench_define_maps[BENCH_DEFINE_MAP_OVERRIDE] = (bench_define_map_t){
  331. bench_override_defines + perm*bench_override_define_count,
  332. bench_override_define_count};
  333. }
  334. void bench_define_explicit(
  335. const bench_define_t *defines,
  336. size_t define_count) {
  337. bench_define_maps[BENCH_DEFINE_MAP_EXPLICIT] = (bench_define_map_t){
  338. defines, define_count};
  339. }
  340. void bench_define_cleanup(void) {
  341. // bench define management can allocate a few things
  342. free(bench_define_cache);
  343. free(bench_define_cache_mask);
  344. free(bench_override_defines);
  345. }
  346. // bench state
  347. extern const bench_geometry_t *bench_geometries;
  348. extern size_t bench_geometry_count;
  349. const bench_id_t *bench_ids = (const bench_id_t[]) {
  350. {NULL, NULL, 0},
  351. };
  352. size_t bench_id_count = 1;
  353. size_t bench_step_start = 0;
  354. size_t bench_step_stop = -1;
  355. size_t bench_step_step = 1;
  356. const char *bench_disk_path = NULL;
  357. const char *bench_trace_path = NULL;
  358. bool bench_trace_backtrace = false;
  359. uint32_t bench_trace_period = 0;
  360. uint32_t bench_trace_freq = 0;
  361. FILE *bench_trace_file = NULL;
  362. uint32_t bench_trace_cycles = 0;
  363. uint64_t bench_trace_time = 0;
  364. uint64_t bench_trace_open_time = 0;
  365. lfs_emubd_sleep_t bench_read_sleep = 0.0;
  366. lfs_emubd_sleep_t bench_prog_sleep = 0.0;
  367. lfs_emubd_sleep_t bench_erase_sleep = 0.0;
  368. // this determines both the backtrace buffer and the trace printf buffer, if
  369. // trace ends up interleaved or truncated this may need to be increased
  370. #ifndef BENCH_TRACE_BACKTRACE_BUFFER_SIZE
  371. #define BENCH_TRACE_BACKTRACE_BUFFER_SIZE 8192
  372. #endif
  373. void *bench_trace_backtrace_buffer[
  374. BENCH_TRACE_BACKTRACE_BUFFER_SIZE / sizeof(void*)];
  375. // trace printing
  376. void bench_trace(const char *fmt, ...) {
  377. if (bench_trace_path) {
  378. // sample at a specific period?
  379. if (bench_trace_period) {
  380. if (bench_trace_cycles % bench_trace_period != 0) {
  381. bench_trace_cycles += 1;
  382. return;
  383. }
  384. bench_trace_cycles += 1;
  385. }
  386. // sample at a specific frequency?
  387. if (bench_trace_freq) {
  388. struct timespec t;
  389. clock_gettime(CLOCK_MONOTONIC, &t);
  390. uint64_t now = (uint64_t)t.tv_sec*1000*1000*1000
  391. + (uint64_t)t.tv_nsec;
  392. if (now - bench_trace_time < (1000*1000*1000) / bench_trace_freq) {
  393. return;
  394. }
  395. bench_trace_time = now;
  396. }
  397. if (!bench_trace_file) {
  398. // Tracing output is heavy and trying to open every trace
  399. // call is slow, so we only try to open the trace file every
  400. // so often. Note this doesn't affect successfully opened files
  401. struct timespec t;
  402. clock_gettime(CLOCK_MONOTONIC, &t);
  403. uint64_t now = (uint64_t)t.tv_sec*1000*1000*1000
  404. + (uint64_t)t.tv_nsec;
  405. if (now - bench_trace_open_time < 100*1000*1000) {
  406. return;
  407. }
  408. bench_trace_open_time = now;
  409. // try to open the trace file
  410. int fd;
  411. if (strcmp(bench_trace_path, "-") == 0) {
  412. fd = dup(1);
  413. if (fd < 0) {
  414. return;
  415. }
  416. } else {
  417. fd = open(
  418. bench_trace_path,
  419. O_WRONLY | O_CREAT | O_APPEND | O_NONBLOCK,
  420. 0666);
  421. if (fd < 0) {
  422. return;
  423. }
  424. int err = fcntl(fd, F_SETFL, O_WRONLY | O_CREAT | O_APPEND);
  425. assert(!err);
  426. }
  427. FILE *f = fdopen(fd, "a");
  428. assert(f);
  429. int err = setvbuf(f, NULL, _IOFBF,
  430. BENCH_TRACE_BACKTRACE_BUFFER_SIZE);
  431. assert(!err);
  432. bench_trace_file = f;
  433. }
  434. // print trace
  435. va_list va;
  436. va_start(va, fmt);
  437. int res = vfprintf(bench_trace_file, fmt, va);
  438. va_end(va);
  439. if (res < 0) {
  440. fclose(bench_trace_file);
  441. bench_trace_file = NULL;
  442. return;
  443. }
  444. if (bench_trace_backtrace) {
  445. // print backtrace
  446. size_t count = backtrace(
  447. bench_trace_backtrace_buffer,
  448. BENCH_TRACE_BACKTRACE_BUFFER_SIZE);
  449. // note we skip our own stack frame
  450. for (size_t i = 1; i < count; i++) {
  451. res = fprintf(bench_trace_file, "\tat %p\n",
  452. bench_trace_backtrace_buffer[i]);
  453. if (res < 0) {
  454. fclose(bench_trace_file);
  455. bench_trace_file = NULL;
  456. return;
  457. }
  458. }
  459. }
  460. // flush immediately
  461. fflush(bench_trace_file);
  462. }
  463. }
  464. // bench prng
  465. uint32_t bench_prng(uint32_t *state) {
  466. // A simple xorshift32 generator, easily reproducible. Keep in mind
  467. // determinism is much more important than actual randomness here.
  468. uint32_t x = *state;
  469. x ^= x << 13;
  470. x ^= x >> 17;
  471. x ^= x << 5;
  472. *state = x;
  473. return x;
  474. }
  475. // bench recording state
  476. static struct lfs_config *bench_cfg = NULL;
  477. static lfs_emubd_io_t bench_last_readed = 0;
  478. static lfs_emubd_io_t bench_last_proged = 0;
  479. static lfs_emubd_io_t bench_last_erased = 0;
  480. lfs_emubd_io_t bench_readed = 0;
  481. lfs_emubd_io_t bench_proged = 0;
  482. lfs_emubd_io_t bench_erased = 0;
  483. void bench_reset(void) {
  484. bench_readed = 0;
  485. bench_proged = 0;
  486. bench_erased = 0;
  487. bench_last_readed = 0;
  488. bench_last_proged = 0;
  489. bench_last_erased = 0;
  490. }
  491. void bench_start(void) {
  492. assert(bench_cfg);
  493. lfs_emubd_sio_t readed = lfs_emubd_readed(bench_cfg);
  494. assert(readed >= 0);
  495. lfs_emubd_sio_t proged = lfs_emubd_proged(bench_cfg);
  496. assert(proged >= 0);
  497. lfs_emubd_sio_t erased = lfs_emubd_erased(bench_cfg);
  498. assert(erased >= 0);
  499. bench_last_readed = readed;
  500. bench_last_proged = proged;
  501. bench_last_erased = erased;
  502. }
  503. void bench_stop(void) {
  504. assert(bench_cfg);
  505. lfs_emubd_sio_t readed = lfs_emubd_readed(bench_cfg);
  506. assert(readed >= 0);
  507. lfs_emubd_sio_t proged = lfs_emubd_proged(bench_cfg);
  508. assert(proged >= 0);
  509. lfs_emubd_sio_t erased = lfs_emubd_erased(bench_cfg);
  510. assert(erased >= 0);
  511. bench_readed += readed - bench_last_readed;
  512. bench_proged += proged - bench_last_proged;
  513. bench_erased += erased - bench_last_erased;
  514. }
  515. // encode our permutation into a reusable id
  516. static void perm_printid(
  517. const struct bench_suite *suite,
  518. const struct bench_case *case_) {
  519. (void)suite;
  520. // case[:permutation]
  521. printf("%s:", case_->name);
  522. for (size_t d = 0;
  523. d < lfs_max(
  524. suite->define_count,
  525. BENCH_IMPLICIT_DEFINE_COUNT);
  526. d++) {
  527. if (bench_define_ispermutation(d)) {
  528. leb16_print(d);
  529. leb16_print(BENCH_DEFINE(d));
  530. }
  531. }
  532. }
  533. // a quick trie for keeping track of permutations we've seen
  534. typedef struct bench_seen {
  535. struct bench_seen_branch *branches;
  536. size_t branch_count;
  537. size_t branch_capacity;
  538. } bench_seen_t;
  539. struct bench_seen_branch {
  540. intmax_t define;
  541. struct bench_seen branch;
  542. };
  543. bool bench_seen_insert(
  544. bench_seen_t *seen,
  545. const struct bench_suite *suite,
  546. const struct bench_case *case_) {
  547. (void)case_;
  548. bool was_seen = true;
  549. // use the currently set defines
  550. for (size_t d = 0;
  551. d < lfs_max(
  552. suite->define_count,
  553. BENCH_IMPLICIT_DEFINE_COUNT);
  554. d++) {
  555. // treat unpermuted defines the same as 0
  556. intmax_t define = bench_define_ispermutation(d) ? BENCH_DEFINE(d) : 0;
  557. // already seen?
  558. struct bench_seen_branch *branch = NULL;
  559. for (size_t i = 0; i < seen->branch_count; i++) {
  560. if (seen->branches[i].define == define) {
  561. branch = &seen->branches[i];
  562. break;
  563. }
  564. }
  565. // need to create a new node
  566. if (!branch) {
  567. was_seen = false;
  568. branch = mappend(
  569. (void**)&seen->branches,
  570. sizeof(struct bench_seen_branch),
  571. &seen->branch_count,
  572. &seen->branch_capacity);
  573. branch->define = define;
  574. branch->branch = (bench_seen_t){NULL, 0, 0};
  575. }
  576. seen = &branch->branch;
  577. }
  578. return was_seen;
  579. }
  580. void bench_seen_cleanup(bench_seen_t *seen) {
  581. for (size_t i = 0; i < seen->branch_count; i++) {
  582. bench_seen_cleanup(&seen->branches[i].branch);
  583. }
  584. free(seen->branches);
  585. }
  586. // iterate through permutations in a bench case
  587. static void case_forperm(
  588. const struct bench_suite *suite,
  589. const struct bench_case *case_,
  590. const bench_define_t *defines,
  591. size_t define_count,
  592. void (*cb)(
  593. void *data,
  594. const struct bench_suite *suite,
  595. const struct bench_case *case_),
  596. void *data) {
  597. // explicit permutation?
  598. if (defines) {
  599. bench_define_explicit(defines, define_count);
  600. for (size_t v = 0; v < bench_override_define_permutations; v++) {
  601. // define override permutation
  602. bench_define_override(v);
  603. bench_define_flush();
  604. cb(data, suite, case_);
  605. }
  606. return;
  607. }
  608. bench_seen_t seen = {NULL, 0, 0};
  609. for (size_t k = 0; k < case_->permutations; k++) {
  610. // define permutation
  611. bench_define_perm(suite, case_, k);
  612. for (size_t v = 0; v < bench_override_define_permutations; v++) {
  613. // define override permutation
  614. bench_define_override(v);
  615. for (size_t g = 0; g < bench_geometry_count; g++) {
  616. // define geometry
  617. bench_define_geometry(&bench_geometries[g]);
  618. bench_define_flush();
  619. // have we seen this permutation before?
  620. bool was_seen = bench_seen_insert(&seen, suite, case_);
  621. if (!(k == 0 && v == 0 && g == 0) && was_seen) {
  622. continue;
  623. }
  624. cb(data, suite, case_);
  625. }
  626. }
  627. }
  628. bench_seen_cleanup(&seen);
  629. }
  630. // how many permutations are there actually in a bench case
  631. struct perm_count_state {
  632. size_t total;
  633. size_t filtered;
  634. };
  635. void perm_count(
  636. void *data,
  637. const struct bench_suite *suite,
  638. const struct bench_case *case_) {
  639. struct perm_count_state *state = data;
  640. (void)suite;
  641. (void)case_;
  642. state->total += 1;
  643. if (case_->filter && !case_->filter()) {
  644. return;
  645. }
  646. state->filtered += 1;
  647. }
  648. // operations we can do
  649. static void summary(void) {
  650. printf("%-23s %7s %7s %7s %11s\n",
  651. "", "flags", "suites", "cases", "perms");
  652. size_t suites = 0;
  653. size_t cases = 0;
  654. bench_flags_t flags = 0;
  655. struct perm_count_state perms = {0, 0};
  656. for (size_t t = 0; t < bench_id_count; t++) {
  657. for (size_t i = 0; i < BENCH_SUITE_COUNT; i++) {
  658. bench_define_suite(&bench_suites[i]);
  659. for (size_t j = 0; j < bench_suites[i].case_count; j++) {
  660. // does neither suite nor case name match?
  661. if (bench_ids[t].name && !(
  662. strcmp(bench_ids[t].name,
  663. bench_suites[i].name) == 0
  664. || strcmp(bench_ids[t].name,
  665. bench_suites[i].cases[j].name) == 0)) {
  666. continue;
  667. }
  668. cases += 1;
  669. case_forperm(
  670. &bench_suites[i],
  671. &bench_suites[i].cases[j],
  672. bench_ids[t].defines,
  673. bench_ids[t].define_count,
  674. perm_count,
  675. &perms);
  676. }
  677. suites += 1;
  678. flags |= bench_suites[i].flags;
  679. }
  680. }
  681. char perm_buf[64];
  682. sprintf(perm_buf, "%zu/%zu", perms.filtered, perms.total);
  683. char flag_buf[64];
  684. sprintf(flag_buf, "%s%s",
  685. (flags & BENCH_REENTRANT) ? "r" : "",
  686. (!flags) ? "-" : "");
  687. printf("%-23s %7s %7zu %7zu %11s\n",
  688. "TOTAL",
  689. flag_buf,
  690. suites,
  691. cases,
  692. perm_buf);
  693. }
  694. static void list_suites(void) {
  695. // at least size so that names fit
  696. unsigned name_width = 23;
  697. for (size_t i = 0; i < BENCH_SUITE_COUNT; i++) {
  698. size_t len = strlen(bench_suites[i].name);
  699. if (len > name_width) {
  700. name_width = len;
  701. }
  702. }
  703. name_width = 4*((name_width+1+4-1)/4)-1;
  704. printf("%-*s %7s %7s %11s\n",
  705. name_width, "suite", "flags", "cases", "perms");
  706. for (size_t t = 0; t < bench_id_count; t++) {
  707. for (size_t i = 0; i < BENCH_SUITE_COUNT; i++) {
  708. bench_define_suite(&bench_suites[i]);
  709. size_t cases = 0;
  710. struct perm_count_state perms = {0, 0};
  711. for (size_t j = 0; j < bench_suites[i].case_count; j++) {
  712. // does neither suite nor case name match?
  713. if (bench_ids[t].name && !(
  714. strcmp(bench_ids[t].name,
  715. bench_suites[i].name) == 0
  716. || strcmp(bench_ids[t].name,
  717. bench_suites[i].cases[j].name) == 0)) {
  718. continue;
  719. }
  720. cases += 1;
  721. case_forperm(
  722. &bench_suites[i],
  723. &bench_suites[i].cases[j],
  724. bench_ids[t].defines,
  725. bench_ids[t].define_count,
  726. perm_count,
  727. &perms);
  728. }
  729. // no benches found?
  730. if (!cases) {
  731. continue;
  732. }
  733. char perm_buf[64];
  734. sprintf(perm_buf, "%zu/%zu", perms.filtered, perms.total);
  735. char flag_buf[64];
  736. sprintf(flag_buf, "%s%s",
  737. (bench_suites[i].flags & BENCH_REENTRANT) ? "r" : "",
  738. (!bench_suites[i].flags) ? "-" : "");
  739. printf("%-*s %7s %7zu %11s\n",
  740. name_width,
  741. bench_suites[i].name,
  742. flag_buf,
  743. cases,
  744. perm_buf);
  745. }
  746. }
  747. }
  748. static void list_cases(void) {
  749. // at least size so that names fit
  750. unsigned name_width = 23;
  751. for (size_t i = 0; i < BENCH_SUITE_COUNT; i++) {
  752. for (size_t j = 0; j < bench_suites[i].case_count; j++) {
  753. size_t len = strlen(bench_suites[i].cases[j].name);
  754. if (len > name_width) {
  755. name_width = len;
  756. }
  757. }
  758. }
  759. name_width = 4*((name_width+1+4-1)/4)-1;
  760. printf("%-*s %7s %11s\n", name_width, "case", "flags", "perms");
  761. for (size_t t = 0; t < bench_id_count; t++) {
  762. for (size_t i = 0; i < BENCH_SUITE_COUNT; i++) {
  763. bench_define_suite(&bench_suites[i]);
  764. for (size_t j = 0; j < bench_suites[i].case_count; j++) {
  765. // does neither suite nor case name match?
  766. if (bench_ids[t].name && !(
  767. strcmp(bench_ids[t].name,
  768. bench_suites[i].name) == 0
  769. || strcmp(bench_ids[t].name,
  770. bench_suites[i].cases[j].name) == 0)) {
  771. continue;
  772. }
  773. struct perm_count_state perms = {0, 0};
  774. case_forperm(
  775. &bench_suites[i],
  776. &bench_suites[i].cases[j],
  777. bench_ids[t].defines,
  778. bench_ids[t].define_count,
  779. perm_count,
  780. &perms);
  781. char perm_buf[64];
  782. sprintf(perm_buf, "%zu/%zu", perms.filtered, perms.total);
  783. char flag_buf[64];
  784. sprintf(flag_buf, "%s%s",
  785. (bench_suites[i].cases[j].flags & BENCH_REENTRANT)
  786. ? "r" : "",
  787. (!bench_suites[i].cases[j].flags)
  788. ? "-" : "");
  789. printf("%-*s %7s %11s\n",
  790. name_width,
  791. bench_suites[i].cases[j].name,
  792. flag_buf,
  793. perm_buf);
  794. }
  795. }
  796. }
  797. }
  798. static void list_suite_paths(void) {
  799. // at least size so that names fit
  800. unsigned name_width = 23;
  801. for (size_t i = 0; i < BENCH_SUITE_COUNT; i++) {
  802. size_t len = strlen(bench_suites[i].name);
  803. if (len > name_width) {
  804. name_width = len;
  805. }
  806. }
  807. name_width = 4*((name_width+1+4-1)/4)-1;
  808. printf("%-*s %s\n", name_width, "suite", "path");
  809. for (size_t t = 0; t < bench_id_count; t++) {
  810. for (size_t i = 0; i < BENCH_SUITE_COUNT; i++) {
  811. size_t cases = 0;
  812. for (size_t j = 0; j < bench_suites[i].case_count; j++) {
  813. // does neither suite nor case name match?
  814. if (bench_ids[t].name && !(
  815. strcmp(bench_ids[t].name,
  816. bench_suites[i].name) == 0
  817. || strcmp(bench_ids[t].name,
  818. bench_suites[i].cases[j].name) == 0)) {
  819. continue;
  820. cases += 1;
  821. }
  822. }
  823. // no benches found?
  824. if (!cases) {
  825. continue;
  826. }
  827. printf("%-*s %s\n",
  828. name_width,
  829. bench_suites[i].name,
  830. bench_suites[i].path);
  831. }
  832. }
  833. }
  834. static void list_case_paths(void) {
  835. // at least size so that names fit
  836. unsigned name_width = 23;
  837. for (size_t i = 0; i < BENCH_SUITE_COUNT; i++) {
  838. for (size_t j = 0; j < bench_suites[i].case_count; j++) {
  839. size_t len = strlen(bench_suites[i].cases[j].name);
  840. if (len > name_width) {
  841. name_width = len;
  842. }
  843. }
  844. }
  845. name_width = 4*((name_width+1+4-1)/4)-1;
  846. printf("%-*s %s\n", name_width, "case", "path");
  847. for (size_t t = 0; t < bench_id_count; t++) {
  848. for (size_t i = 0; i < BENCH_SUITE_COUNT; i++) {
  849. for (size_t j = 0; j < bench_suites[i].case_count; j++) {
  850. // does neither suite nor case name match?
  851. if (bench_ids[t].name && !(
  852. strcmp(bench_ids[t].name,
  853. bench_suites[i].name) == 0
  854. || strcmp(bench_ids[t].name,
  855. bench_suites[i].cases[j].name) == 0)) {
  856. continue;
  857. }
  858. printf("%-*s %s\n",
  859. name_width,
  860. bench_suites[i].cases[j].name,
  861. bench_suites[i].cases[j].path);
  862. }
  863. }
  864. }
  865. }
  866. struct list_defines_define {
  867. const char *name;
  868. intmax_t *values;
  869. size_t value_count;
  870. size_t value_capacity;
  871. };
  872. struct list_defines_defines {
  873. struct list_defines_define *defines;
  874. size_t define_count;
  875. size_t define_capacity;
  876. };
  877. static void list_defines_add(
  878. struct list_defines_defines *defines,
  879. size_t d) {
  880. const char *name = bench_define_name(d);
  881. intmax_t value = BENCH_DEFINE(d);
  882. // define already in defines?
  883. for (size_t i = 0; i < defines->define_count; i++) {
  884. if (strcmp(defines->defines[i].name, name) == 0) {
  885. // value already in values?
  886. for (size_t j = 0; j < defines->defines[i].value_count; j++) {
  887. if (defines->defines[i].values[j] == value) {
  888. return;
  889. }
  890. }
  891. *(intmax_t*)mappend(
  892. (void**)&defines->defines[i].values,
  893. sizeof(intmax_t),
  894. &defines->defines[i].value_count,
  895. &defines->defines[i].value_capacity) = value;
  896. return;
  897. }
  898. }
  899. // new define?
  900. struct list_defines_define *define = mappend(
  901. (void**)&defines->defines,
  902. sizeof(struct list_defines_define),
  903. &defines->define_count,
  904. &defines->define_capacity);
  905. define->name = name;
  906. define->values = malloc(sizeof(intmax_t));
  907. define->values[0] = value;
  908. define->value_count = 1;
  909. define->value_capacity = 1;
  910. }
  911. void perm_list_defines(
  912. void *data,
  913. const struct bench_suite *suite,
  914. const struct bench_case *case_) {
  915. struct list_defines_defines *defines = data;
  916. (void)suite;
  917. (void)case_;
  918. // collect defines
  919. for (size_t d = 0;
  920. d < lfs_max(suite->define_count,
  921. BENCH_IMPLICIT_DEFINE_COUNT);
  922. d++) {
  923. if (d < BENCH_IMPLICIT_DEFINE_COUNT
  924. || bench_define_ispermutation(d)) {
  925. list_defines_add(defines, d);
  926. }
  927. }
  928. }
  929. void perm_list_permutation_defines(
  930. void *data,
  931. const struct bench_suite *suite,
  932. const struct bench_case *case_) {
  933. struct list_defines_defines *defines = data;
  934. (void)suite;
  935. (void)case_;
  936. // collect permutation_defines
  937. for (size_t d = 0;
  938. d < lfs_max(suite->define_count,
  939. BENCH_IMPLICIT_DEFINE_COUNT);
  940. d++) {
  941. if (bench_define_ispermutation(d)) {
  942. list_defines_add(defines, d);
  943. }
  944. }
  945. }
  946. extern const bench_geometry_t builtin_geometries[];
  947. static void list_defines(void) {
  948. struct list_defines_defines defines = {NULL, 0, 0};
  949. // add defines
  950. for (size_t t = 0; t < bench_id_count; t++) {
  951. for (size_t i = 0; i < BENCH_SUITE_COUNT; i++) {
  952. bench_define_suite(&bench_suites[i]);
  953. for (size_t j = 0; j < bench_suites[i].case_count; j++) {
  954. // does neither suite nor case name match?
  955. if (bench_ids[t].name && !(
  956. strcmp(bench_ids[t].name,
  957. bench_suites[i].name) == 0
  958. || strcmp(bench_ids[t].name,
  959. bench_suites[i].cases[j].name) == 0)) {
  960. continue;
  961. }
  962. case_forperm(
  963. &bench_suites[i],
  964. &bench_suites[i].cases[j],
  965. bench_ids[t].defines,
  966. bench_ids[t].define_count,
  967. perm_list_defines,
  968. &defines);
  969. }
  970. }
  971. }
  972. for (size_t i = 0; i < defines.define_count; i++) {
  973. printf("%s=", defines.defines[i].name);
  974. for (size_t j = 0; j < defines.defines[i].value_count; j++) {
  975. printf("%jd", defines.defines[i].values[j]);
  976. if (j != defines.defines[i].value_count-1) {
  977. printf(",");
  978. }
  979. }
  980. printf("\n");
  981. }
  982. for (size_t i = 0; i < defines.define_count; i++) {
  983. free(defines.defines[i].values);
  984. }
  985. free(defines.defines);
  986. }
  987. static void list_permutation_defines(void) {
  988. struct list_defines_defines defines = {NULL, 0, 0};
  989. // add permutation defines
  990. for (size_t t = 0; t < bench_id_count; t++) {
  991. for (size_t i = 0; i < BENCH_SUITE_COUNT; i++) {
  992. bench_define_suite(&bench_suites[i]);
  993. for (size_t j = 0; j < bench_suites[i].case_count; j++) {
  994. // does neither suite nor case name match?
  995. if (bench_ids[t].name && !(
  996. strcmp(bench_ids[t].name,
  997. bench_suites[i].name) == 0
  998. || strcmp(bench_ids[t].name,
  999. bench_suites[i].cases[j].name) == 0)) {
  1000. continue;
  1001. }
  1002. case_forperm(
  1003. &bench_suites[i],
  1004. &bench_suites[i].cases[j],
  1005. bench_ids[t].defines,
  1006. bench_ids[t].define_count,
  1007. perm_list_permutation_defines,
  1008. &defines);
  1009. }
  1010. }
  1011. }
  1012. for (size_t i = 0; i < defines.define_count; i++) {
  1013. printf("%s=", defines.defines[i].name);
  1014. for (size_t j = 0; j < defines.defines[i].value_count; j++) {
  1015. printf("%jd", defines.defines[i].values[j]);
  1016. if (j != defines.defines[i].value_count-1) {
  1017. printf(",");
  1018. }
  1019. }
  1020. printf("\n");
  1021. }
  1022. for (size_t i = 0; i < defines.define_count; i++) {
  1023. free(defines.defines[i].values);
  1024. }
  1025. free(defines.defines);
  1026. }
  1027. static void list_implicit_defines(void) {
  1028. struct list_defines_defines defines = {NULL, 0, 0};
  1029. // yes we do need to define a suite, this does a bit of bookeeping
  1030. // such as setting up the define cache
  1031. bench_define_suite(&(const struct bench_suite){0});
  1032. // make sure to include builtin geometries here
  1033. extern const bench_geometry_t builtin_geometries[];
  1034. for (size_t g = 0; builtin_geometries[g].name; g++) {
  1035. bench_define_geometry(&builtin_geometries[g]);
  1036. bench_define_flush();
  1037. // add implicit defines
  1038. for (size_t d = 0; d < BENCH_IMPLICIT_DEFINE_COUNT; d++) {
  1039. list_defines_add(&defines, d);
  1040. }
  1041. }
  1042. for (size_t i = 0; i < defines.define_count; i++) {
  1043. printf("%s=", defines.defines[i].name);
  1044. for (size_t j = 0; j < defines.defines[i].value_count; j++) {
  1045. printf("%jd", defines.defines[i].values[j]);
  1046. if (j != defines.defines[i].value_count-1) {
  1047. printf(",");
  1048. }
  1049. }
  1050. printf("\n");
  1051. }
  1052. for (size_t i = 0; i < defines.define_count; i++) {
  1053. free(defines.defines[i].values);
  1054. }
  1055. free(defines.defines);
  1056. }
  1057. // geometries to bench
  1058. const bench_geometry_t builtin_geometries[] = {
  1059. {"default", {{0}, BENCH_CONST(16), BENCH_CONST(512), {0}}},
  1060. {"eeprom", {{0}, BENCH_CONST(1), BENCH_CONST(512), {0}}},
  1061. {"emmc", {{0}, {0}, BENCH_CONST(512), {0}}},
  1062. {"nor", {{0}, BENCH_CONST(1), BENCH_CONST(4096), {0}}},
  1063. {"nand", {{0}, BENCH_CONST(4096), BENCH_CONST(32768), {0}}},
  1064. {NULL, {{0}, {0}, {0}, {0}}},
  1065. };
  1066. const bench_geometry_t *bench_geometries = builtin_geometries;
  1067. size_t bench_geometry_count = 5;
  1068. static void list_geometries(void) {
  1069. // at least size so that names fit
  1070. unsigned name_width = 23;
  1071. for (size_t g = 0; builtin_geometries[g].name; g++) {
  1072. size_t len = strlen(builtin_geometries[g].name);
  1073. if (len > name_width) {
  1074. name_width = len;
  1075. }
  1076. }
  1077. name_width = 4*((name_width+1+4-1)/4)-1;
  1078. // yes we do need to define a suite, this does a bit of bookeeping
  1079. // such as setting up the define cache
  1080. bench_define_suite(&(const struct bench_suite){0});
  1081. printf("%-*s %7s %7s %7s %7s %11s\n",
  1082. name_width, "geometry", "read", "prog", "erase", "count", "size");
  1083. for (size_t g = 0; builtin_geometries[g].name; g++) {
  1084. bench_define_geometry(&builtin_geometries[g]);
  1085. bench_define_flush();
  1086. printf("%-*s %7ju %7ju %7ju %7ju %11ju\n",
  1087. name_width,
  1088. builtin_geometries[g].name,
  1089. READ_SIZE,
  1090. PROG_SIZE,
  1091. BLOCK_SIZE,
  1092. BLOCK_COUNT,
  1093. BLOCK_SIZE*BLOCK_COUNT);
  1094. }
  1095. }
  1096. // global bench step count
  1097. size_t bench_step = 0;
  1098. void perm_run(
  1099. void *data,
  1100. const struct bench_suite *suite,
  1101. const struct bench_case *case_) {
  1102. (void)data;
  1103. // skip this step?
  1104. if (!(bench_step >= bench_step_start
  1105. && bench_step < bench_step_stop
  1106. && (bench_step-bench_step_start) % bench_step_step == 0)) {
  1107. bench_step += 1;
  1108. return;
  1109. }
  1110. bench_step += 1;
  1111. // filter?
  1112. if (case_->filter && !case_->filter()) {
  1113. printf("skipped ");
  1114. perm_printid(suite, case_);
  1115. printf("\n");
  1116. return;
  1117. }
  1118. // create block device and configuration
  1119. lfs_emubd_t bd;
  1120. struct lfs_config cfg = {
  1121. .context = &bd,
  1122. .read = lfs_emubd_read,
  1123. .prog = lfs_emubd_prog,
  1124. .erase = lfs_emubd_erase,
  1125. .sync = lfs_emubd_sync,
  1126. .read_size = READ_SIZE,
  1127. .prog_size = PROG_SIZE,
  1128. .block_size = BLOCK_SIZE,
  1129. .block_count = BLOCK_COUNT,
  1130. .block_cycles = BLOCK_CYCLES,
  1131. .cache_size = CACHE_SIZE,
  1132. .lookahead_size = LOOKAHEAD_SIZE,
  1133. };
  1134. struct lfs_emubd_config bdcfg = {
  1135. .erase_value = ERASE_VALUE,
  1136. .erase_cycles = ERASE_CYCLES,
  1137. .badblock_behavior = BADBLOCK_BEHAVIOR,
  1138. .disk_path = bench_disk_path,
  1139. .read_sleep = bench_read_sleep,
  1140. .prog_sleep = bench_prog_sleep,
  1141. .erase_sleep = bench_erase_sleep,
  1142. };
  1143. int err = lfs_emubd_createcfg(&cfg, bench_disk_path, &bdcfg);
  1144. if (err) {
  1145. fprintf(stderr, "error: could not create block device: %d\n", err);
  1146. exit(-1);
  1147. }
  1148. // run the bench
  1149. bench_cfg = &cfg;
  1150. bench_reset();
  1151. printf("running ");
  1152. perm_printid(suite, case_);
  1153. printf("\n");
  1154. case_->run(&cfg);
  1155. printf("finished ");
  1156. perm_printid(suite, case_);
  1157. printf(" %"PRIu64" %"PRIu64" %"PRIu64,
  1158. bench_readed,
  1159. bench_proged,
  1160. bench_erased);
  1161. printf("\n");
  1162. // cleanup
  1163. err = lfs_emubd_destroy(&cfg);
  1164. if (err) {
  1165. fprintf(stderr, "error: could not destroy block device: %d\n", err);
  1166. exit(-1);
  1167. }
  1168. }
  1169. static void run(void) {
  1170. // ignore disconnected pipes
  1171. signal(SIGPIPE, SIG_IGN);
  1172. for (size_t t = 0; t < bench_id_count; t++) {
  1173. for (size_t i = 0; i < BENCH_SUITE_COUNT; i++) {
  1174. bench_define_suite(&bench_suites[i]);
  1175. for (size_t j = 0; j < bench_suites[i].case_count; j++) {
  1176. // does neither suite nor case name match?
  1177. if (bench_ids[t].name && !(
  1178. strcmp(bench_ids[t].name,
  1179. bench_suites[i].name) == 0
  1180. || strcmp(bench_ids[t].name,
  1181. bench_suites[i].cases[j].name) == 0)) {
  1182. continue;
  1183. }
  1184. case_forperm(
  1185. &bench_suites[i],
  1186. &bench_suites[i].cases[j],
  1187. bench_ids[t].defines,
  1188. bench_ids[t].define_count,
  1189. perm_run,
  1190. NULL);
  1191. }
  1192. }
  1193. }
  1194. }
  1195. // option handling
  1196. enum opt_flags {
  1197. OPT_HELP = 'h',
  1198. OPT_SUMMARY = 'Y',
  1199. OPT_LIST_SUITES = 'l',
  1200. OPT_LIST_CASES = 'L',
  1201. OPT_LIST_SUITE_PATHS = 1,
  1202. OPT_LIST_CASE_PATHS = 2,
  1203. OPT_LIST_DEFINES = 3,
  1204. OPT_LIST_PERMUTATION_DEFINES = 4,
  1205. OPT_LIST_IMPLICIT_DEFINES = 5,
  1206. OPT_LIST_GEOMETRIES = 6,
  1207. OPT_DEFINE = 'D',
  1208. OPT_GEOMETRY = 'G',
  1209. OPT_STEP = 's',
  1210. OPT_DISK = 'd',
  1211. OPT_TRACE = 't',
  1212. OPT_TRACE_BACKTRACE = 7,
  1213. OPT_TRACE_PERIOD = 8,
  1214. OPT_TRACE_FREQ = 9,
  1215. OPT_READ_SLEEP = 10,
  1216. OPT_PROG_SLEEP = 11,
  1217. OPT_ERASE_SLEEP = 12,
  1218. };
  1219. const char *short_opts = "hYlLD:G:s:d:t:";
  1220. const struct option long_opts[] = {
  1221. {"help", no_argument, NULL, OPT_HELP},
  1222. {"summary", no_argument, NULL, OPT_SUMMARY},
  1223. {"list-suites", no_argument, NULL, OPT_LIST_SUITES},
  1224. {"list-cases", no_argument, NULL, OPT_LIST_CASES},
  1225. {"list-suite-paths", no_argument, NULL, OPT_LIST_SUITE_PATHS},
  1226. {"list-case-paths", no_argument, NULL, OPT_LIST_CASE_PATHS},
  1227. {"list-defines", no_argument, NULL, OPT_LIST_DEFINES},
  1228. {"list-permutation-defines",
  1229. no_argument, NULL, OPT_LIST_PERMUTATION_DEFINES},
  1230. {"list-implicit-defines",
  1231. no_argument, NULL, OPT_LIST_IMPLICIT_DEFINES},
  1232. {"list-geometries", no_argument, NULL, OPT_LIST_GEOMETRIES},
  1233. {"define", required_argument, NULL, OPT_DEFINE},
  1234. {"geometry", required_argument, NULL, OPT_GEOMETRY},
  1235. {"step", required_argument, NULL, OPT_STEP},
  1236. {"disk", required_argument, NULL, OPT_DISK},
  1237. {"trace", required_argument, NULL, OPT_TRACE},
  1238. {"trace-backtrace", no_argument, NULL, OPT_TRACE_BACKTRACE},
  1239. {"trace-period", required_argument, NULL, OPT_TRACE_PERIOD},
  1240. {"trace-freq", required_argument, NULL, OPT_TRACE_FREQ},
  1241. {"read-sleep", required_argument, NULL, OPT_READ_SLEEP},
  1242. {"prog-sleep", required_argument, NULL, OPT_PROG_SLEEP},
  1243. {"erase-sleep", required_argument, NULL, OPT_ERASE_SLEEP},
  1244. {NULL, 0, NULL, 0},
  1245. };
  1246. const char *const help_text[] = {
  1247. "Show this help message.",
  1248. "Show quick summary.",
  1249. "List bench suites.",
  1250. "List bench cases.",
  1251. "List the path for each bench suite.",
  1252. "List the path and line number for each bench case.",
  1253. "List all defines in this bench-runner.",
  1254. "List explicit defines in this bench-runner.",
  1255. "List implicit defines in this bench-runner.",
  1256. "List the available disk geometries.",
  1257. "Override a bench define.",
  1258. "Comma-separated list of disk geometries to bench.",
  1259. "Comma-separated range of bench permutations to run (start,stop,step).",
  1260. "Direct block device operations to this file.",
  1261. "Direct trace output to this file.",
  1262. "Include a backtrace with every trace statement.",
  1263. "Sample trace output at this period in cycles.",
  1264. "Sample trace output at this frequency in hz.",
  1265. "Artificial read delay in seconds.",
  1266. "Artificial prog delay in seconds.",
  1267. "Artificial erase delay in seconds.",
  1268. };
  1269. int main(int argc, char **argv) {
  1270. void (*op)(void) = run;
  1271. size_t bench_override_capacity = 0;
  1272. size_t bench_geometry_capacity = 0;
  1273. size_t bench_id_capacity = 0;
  1274. // parse options
  1275. while (true) {
  1276. int c = getopt_long(argc, argv, short_opts, long_opts, NULL);
  1277. switch (c) {
  1278. // generate help message
  1279. case OPT_HELP: {
  1280. printf("usage: %s [options] [bench_id]\n", argv[0]);
  1281. printf("\n");
  1282. printf("options:\n");
  1283. size_t i = 0;
  1284. while (long_opts[i].name) {
  1285. size_t indent;
  1286. if (long_opts[i].has_arg == no_argument) {
  1287. if (long_opts[i].val >= '0' && long_opts[i].val < 'z') {
  1288. indent = printf(" -%c, --%s ",
  1289. long_opts[i].val,
  1290. long_opts[i].name);
  1291. } else {
  1292. indent = printf(" --%s ",
  1293. long_opts[i].name);
  1294. }
  1295. } else {
  1296. if (long_opts[i].val >= '0' && long_opts[i].val < 'z') {
  1297. indent = printf(" -%c %s, --%s %s ",
  1298. long_opts[i].val,
  1299. long_opts[i].name,
  1300. long_opts[i].name,
  1301. long_opts[i].name);
  1302. } else {
  1303. indent = printf(" --%s %s ",
  1304. long_opts[i].name,
  1305. long_opts[i].name);
  1306. }
  1307. }
  1308. // a quick, hacky, byte-level method for text wrapping
  1309. size_t len = strlen(help_text[i]);
  1310. size_t j = 0;
  1311. if (indent < 24) {
  1312. printf("%*s %.80s\n",
  1313. (int)(24-1-indent),
  1314. "",
  1315. &help_text[i][j]);
  1316. j += 80;
  1317. } else {
  1318. printf("\n");
  1319. }
  1320. while (j < len) {
  1321. printf("%24s%.80s\n", "", &help_text[i][j]);
  1322. j += 80;
  1323. }
  1324. i += 1;
  1325. }
  1326. printf("\n");
  1327. exit(0);
  1328. }
  1329. // summary/list flags
  1330. case OPT_SUMMARY:
  1331. op = summary;
  1332. break;
  1333. case OPT_LIST_SUITES:
  1334. op = list_suites;
  1335. break;
  1336. case OPT_LIST_CASES:
  1337. op = list_cases;
  1338. break;
  1339. case OPT_LIST_SUITE_PATHS:
  1340. op = list_suite_paths;
  1341. break;
  1342. case OPT_LIST_CASE_PATHS:
  1343. op = list_case_paths;
  1344. break;
  1345. case OPT_LIST_DEFINES:
  1346. op = list_defines;
  1347. break;
  1348. case OPT_LIST_PERMUTATION_DEFINES:
  1349. op = list_permutation_defines;
  1350. break;
  1351. case OPT_LIST_IMPLICIT_DEFINES:
  1352. op = list_implicit_defines;
  1353. break;
  1354. case OPT_LIST_GEOMETRIES:
  1355. op = list_geometries;
  1356. break;
  1357. // configuration
  1358. case OPT_DEFINE: {
  1359. // allocate space
  1360. bench_override_t *override = mappend(
  1361. (void**)&bench_overrides,
  1362. sizeof(bench_override_t),
  1363. &bench_override_count,
  1364. &bench_override_capacity);
  1365. // parse into string key/intmax_t value, cannibalizing the
  1366. // arg in the process
  1367. char *sep = strchr(optarg, '=');
  1368. char *parsed = NULL;
  1369. if (!sep) {
  1370. goto invalid_define;
  1371. }
  1372. *sep = '\0';
  1373. override->name = optarg;
  1374. optarg = sep+1;
  1375. // parse comma-separated permutations
  1376. {
  1377. override->defines = NULL;
  1378. override->permutations = 0;
  1379. size_t override_capacity = 0;
  1380. while (true) {
  1381. optarg += strspn(optarg, " ");
  1382. if (strncmp(optarg, "range", strlen("range")) == 0) {
  1383. // range of values
  1384. optarg += strlen("range");
  1385. optarg += strspn(optarg, " ");
  1386. if (*optarg != '(') {
  1387. goto invalid_define;
  1388. }
  1389. optarg += 1;
  1390. intmax_t start = strtoumax(optarg, &parsed, 0);
  1391. intmax_t stop = -1;
  1392. intmax_t step = 1;
  1393. // allow empty string for start=0
  1394. if (parsed == optarg) {
  1395. start = 0;
  1396. }
  1397. optarg = parsed + strspn(parsed, " ");
  1398. if (*optarg != ',' && *optarg != ')') {
  1399. goto invalid_define;
  1400. }
  1401. if (*optarg == ',') {
  1402. optarg += 1;
  1403. stop = strtoumax(optarg, &parsed, 0);
  1404. // allow empty string for stop=end
  1405. if (parsed == optarg) {
  1406. stop = -1;
  1407. }
  1408. optarg = parsed + strspn(parsed, " ");
  1409. if (*optarg != ',' && *optarg != ')') {
  1410. goto invalid_define;
  1411. }
  1412. if (*optarg == ',') {
  1413. optarg += 1;
  1414. step = strtoumax(optarg, &parsed, 0);
  1415. // allow empty string for stop=1
  1416. if (parsed == optarg) {
  1417. step = 1;
  1418. }
  1419. optarg = parsed + strspn(parsed, " ");
  1420. if (*optarg != ')') {
  1421. goto invalid_define;
  1422. }
  1423. }
  1424. } else {
  1425. // single value = stop only
  1426. stop = start;
  1427. start = 0;
  1428. }
  1429. if (*optarg != ')') {
  1430. goto invalid_define;
  1431. }
  1432. optarg += 1;
  1433. // calculate the range of values
  1434. assert(step != 0);
  1435. for (intmax_t i = start;
  1436. (step < 0)
  1437. ? i > stop
  1438. : (uintmax_t)i < (uintmax_t)stop;
  1439. i += step) {
  1440. *(intmax_t*)mappend(
  1441. (void**)&override->defines,
  1442. sizeof(intmax_t),
  1443. &override->permutations,
  1444. &override_capacity) = i;
  1445. }
  1446. } else if (*optarg != '\0') {
  1447. // single value
  1448. intmax_t define = strtoimax(optarg, &parsed, 0);
  1449. if (parsed == optarg) {
  1450. goto invalid_define;
  1451. }
  1452. optarg = parsed + strspn(parsed, " ");
  1453. *(intmax_t*)mappend(
  1454. (void**)&override->defines,
  1455. sizeof(intmax_t),
  1456. &override->permutations,
  1457. &override_capacity) = define;
  1458. } else {
  1459. break;
  1460. }
  1461. if (*optarg == ',') {
  1462. optarg += 1;
  1463. }
  1464. }
  1465. }
  1466. assert(override->permutations > 0);
  1467. break;
  1468. invalid_define:
  1469. fprintf(stderr, "error: invalid define: %s\n", optarg);
  1470. exit(-1);
  1471. }
  1472. case OPT_GEOMETRY: {
  1473. // reset our geometry scenarios
  1474. if (bench_geometry_capacity > 0) {
  1475. free((bench_geometry_t*)bench_geometries);
  1476. }
  1477. bench_geometries = NULL;
  1478. bench_geometry_count = 0;
  1479. bench_geometry_capacity = 0;
  1480. // parse the comma separated list of disk geometries
  1481. while (*optarg) {
  1482. // allocate space
  1483. bench_geometry_t *geometry = mappend(
  1484. (void**)&bench_geometries,
  1485. sizeof(bench_geometry_t),
  1486. &bench_geometry_count,
  1487. &bench_geometry_capacity);
  1488. // parse the disk geometry
  1489. optarg += strspn(optarg, " ");
  1490. // named disk geometry
  1491. size_t len = strcspn(optarg, " ,");
  1492. for (size_t i = 0; builtin_geometries[i].name; i++) {
  1493. if (len == strlen(builtin_geometries[i].name)
  1494. && memcmp(optarg,
  1495. builtin_geometries[i].name,
  1496. len) == 0) {
  1497. *geometry = builtin_geometries[i];
  1498. optarg += len;
  1499. goto geometry_next;
  1500. }
  1501. }
  1502. // comma-separated read/prog/erase/count
  1503. if (*optarg == '{') {
  1504. lfs_size_t sizes[4];
  1505. size_t count = 0;
  1506. char *s = optarg + 1;
  1507. while (count < 4) {
  1508. char *parsed = NULL;
  1509. sizes[count] = strtoumax(s, &parsed, 0);
  1510. count += 1;
  1511. s = parsed + strspn(parsed, " ");
  1512. if (*s == ',') {
  1513. s += 1;
  1514. continue;
  1515. } else if (*s == '}') {
  1516. s += 1;
  1517. break;
  1518. } else {
  1519. goto geometry_unknown;
  1520. }
  1521. }
  1522. // allow implicit r=p and p=e for common geometries
  1523. memset(geometry, 0, sizeof(bench_geometry_t));
  1524. if (count >= 3) {
  1525. geometry->defines[READ_SIZE_i]
  1526. = BENCH_LIT(sizes[0]);
  1527. geometry->defines[PROG_SIZE_i]
  1528. = BENCH_LIT(sizes[1]);
  1529. geometry->defines[BLOCK_SIZE_i]
  1530. = BENCH_LIT(sizes[2]);
  1531. } else if (count >= 2) {
  1532. geometry->defines[PROG_SIZE_i]
  1533. = BENCH_LIT(sizes[0]);
  1534. geometry->defines[BLOCK_SIZE_i]
  1535. = BENCH_LIT(sizes[1]);
  1536. } else {
  1537. geometry->defines[BLOCK_SIZE_i]
  1538. = BENCH_LIT(sizes[0]);
  1539. }
  1540. if (count >= 4) {
  1541. geometry->defines[BLOCK_COUNT_i]
  1542. = BENCH_LIT(sizes[3]);
  1543. }
  1544. optarg = s;
  1545. goto geometry_next;
  1546. }
  1547. // leb16-encoded read/prog/erase/count
  1548. if (*optarg == ':') {
  1549. lfs_size_t sizes[4];
  1550. size_t count = 0;
  1551. char *s = optarg + 1;
  1552. while (true) {
  1553. char *parsed = NULL;
  1554. uintmax_t x = leb16_parse(s, &parsed);
  1555. if (parsed == s || count >= 4) {
  1556. break;
  1557. }
  1558. sizes[count] = x;
  1559. count += 1;
  1560. s = parsed;
  1561. }
  1562. // allow implicit r=p and p=e for common geometries
  1563. memset(geometry, 0, sizeof(bench_geometry_t));
  1564. if (count >= 3) {
  1565. geometry->defines[READ_SIZE_i]
  1566. = BENCH_LIT(sizes[0]);
  1567. geometry->defines[PROG_SIZE_i]
  1568. = BENCH_LIT(sizes[1]);
  1569. geometry->defines[BLOCK_SIZE_i]
  1570. = BENCH_LIT(sizes[2]);
  1571. } else if (count >= 2) {
  1572. geometry->defines[PROG_SIZE_i]
  1573. = BENCH_LIT(sizes[0]);
  1574. geometry->defines[BLOCK_SIZE_i]
  1575. = BENCH_LIT(sizes[1]);
  1576. } else {
  1577. geometry->defines[BLOCK_SIZE_i]
  1578. = BENCH_LIT(sizes[0]);
  1579. }
  1580. if (count >= 4) {
  1581. geometry->defines[BLOCK_COUNT_i]
  1582. = BENCH_LIT(sizes[3]);
  1583. }
  1584. optarg = s;
  1585. goto geometry_next;
  1586. }
  1587. geometry_unknown:
  1588. // unknown scenario?
  1589. fprintf(stderr, "error: unknown disk geometry: %s\n",
  1590. optarg);
  1591. exit(-1);
  1592. geometry_next:
  1593. optarg += strspn(optarg, " ");
  1594. if (*optarg == ',') {
  1595. optarg += 1;
  1596. } else if (*optarg == '\0') {
  1597. break;
  1598. } else {
  1599. goto geometry_unknown;
  1600. }
  1601. }
  1602. break;
  1603. }
  1604. case OPT_STEP: {
  1605. char *parsed = NULL;
  1606. bench_step_start = strtoumax(optarg, &parsed, 0);
  1607. bench_step_stop = -1;
  1608. bench_step_step = 1;
  1609. // allow empty string for start=0
  1610. if (parsed == optarg) {
  1611. bench_step_start = 0;
  1612. }
  1613. optarg = parsed + strspn(parsed, " ");
  1614. if (*optarg != ',' && *optarg != '\0') {
  1615. goto step_unknown;
  1616. }
  1617. if (*optarg == ',') {
  1618. optarg += 1;
  1619. bench_step_stop = strtoumax(optarg, &parsed, 0);
  1620. // allow empty string for stop=end
  1621. if (parsed == optarg) {
  1622. bench_step_stop = -1;
  1623. }
  1624. optarg = parsed + strspn(parsed, " ");
  1625. if (*optarg != ',' && *optarg != '\0') {
  1626. goto step_unknown;
  1627. }
  1628. if (*optarg == ',') {
  1629. optarg += 1;
  1630. bench_step_step = strtoumax(optarg, &parsed, 0);
  1631. // allow empty string for stop=1
  1632. if (parsed == optarg) {
  1633. bench_step_step = 1;
  1634. }
  1635. optarg = parsed + strspn(parsed, " ");
  1636. if (*optarg != '\0') {
  1637. goto step_unknown;
  1638. }
  1639. }
  1640. } else {
  1641. // single value = stop only
  1642. bench_step_stop = bench_step_start;
  1643. bench_step_start = 0;
  1644. }
  1645. break;
  1646. step_unknown:
  1647. fprintf(stderr, "error: invalid step: %s\n", optarg);
  1648. exit(-1);
  1649. }
  1650. case OPT_DISK:
  1651. bench_disk_path = optarg;
  1652. break;
  1653. case OPT_TRACE:
  1654. bench_trace_path = optarg;
  1655. break;
  1656. case OPT_TRACE_BACKTRACE:
  1657. bench_trace_backtrace = true;
  1658. break;
  1659. case OPT_TRACE_PERIOD: {
  1660. char *parsed = NULL;
  1661. bench_trace_period = strtoumax(optarg, &parsed, 0);
  1662. if (parsed == optarg) {
  1663. fprintf(stderr, "error: invalid trace-period: %s\n", optarg);
  1664. exit(-1);
  1665. }
  1666. break;
  1667. }
  1668. case OPT_TRACE_FREQ: {
  1669. char *parsed = NULL;
  1670. bench_trace_freq = strtoumax(optarg, &parsed, 0);
  1671. if (parsed == optarg) {
  1672. fprintf(stderr, "error: invalid trace-freq: %s\n", optarg);
  1673. exit(-1);
  1674. }
  1675. break;
  1676. }
  1677. case OPT_READ_SLEEP: {
  1678. char *parsed = NULL;
  1679. double read_sleep = strtod(optarg, &parsed);
  1680. if (parsed == optarg) {
  1681. fprintf(stderr, "error: invalid read-sleep: %s\n", optarg);
  1682. exit(-1);
  1683. }
  1684. bench_read_sleep = read_sleep*1.0e9;
  1685. break;
  1686. }
  1687. case OPT_PROG_SLEEP: {
  1688. char *parsed = NULL;
  1689. double prog_sleep = strtod(optarg, &parsed);
  1690. if (parsed == optarg) {
  1691. fprintf(stderr, "error: invalid prog-sleep: %s\n", optarg);
  1692. exit(-1);
  1693. }
  1694. bench_prog_sleep = prog_sleep*1.0e9;
  1695. break;
  1696. }
  1697. case OPT_ERASE_SLEEP: {
  1698. char *parsed = NULL;
  1699. double erase_sleep = strtod(optarg, &parsed);
  1700. if (parsed == optarg) {
  1701. fprintf(stderr, "error: invalid erase-sleep: %s\n", optarg);
  1702. exit(-1);
  1703. }
  1704. bench_erase_sleep = erase_sleep*1.0e9;
  1705. break;
  1706. }
  1707. // done parsing
  1708. case -1:
  1709. goto getopt_done;
  1710. // unknown arg, getopt prints a message for us
  1711. default:
  1712. exit(-1);
  1713. }
  1714. }
  1715. getopt_done: ;
  1716. if (argc > optind) {
  1717. // reset our bench identifier list
  1718. bench_ids = NULL;
  1719. bench_id_count = 0;
  1720. bench_id_capacity = 0;
  1721. }
  1722. // parse bench identifier, if any, cannibalizing the arg in the process
  1723. for (; argc > optind; optind++) {
  1724. bench_define_t *defines = NULL;
  1725. size_t define_count = 0;
  1726. // parse name, can be suite or case
  1727. char *name = argv[optind];
  1728. char *defines_ = strchr(name, ':');
  1729. if (defines_) {
  1730. *defines_ = '\0';
  1731. defines_ += 1;
  1732. }
  1733. // remove optional path and .toml suffix
  1734. char *slash = strrchr(name, '/');
  1735. if (slash) {
  1736. name = slash+1;
  1737. }
  1738. size_t name_len = strlen(name);
  1739. if (name_len > 5 && strcmp(&name[name_len-5], ".toml") == 0) {
  1740. name[name_len-5] = '\0';
  1741. }
  1742. if (defines_) {
  1743. // parse defines
  1744. while (true) {
  1745. char *parsed;
  1746. size_t d = leb16_parse(defines_, &parsed);
  1747. intmax_t v = leb16_parse(parsed, &parsed);
  1748. if (parsed == defines_) {
  1749. break;
  1750. }
  1751. defines_ = parsed;
  1752. if (d >= define_count) {
  1753. // align to power of two to avoid any superlinear growth
  1754. size_t ncount = 1 << lfs_npw2(d+1);
  1755. defines = realloc(defines,
  1756. ncount*sizeof(bench_define_t));
  1757. memset(defines+define_count, 0,
  1758. (ncount-define_count)*sizeof(bench_define_t));
  1759. define_count = ncount;
  1760. }
  1761. defines[d] = BENCH_LIT(v);
  1762. }
  1763. }
  1764. // append to identifier list
  1765. *(bench_id_t*)mappend(
  1766. (void**)&bench_ids,
  1767. sizeof(bench_id_t),
  1768. &bench_id_count,
  1769. &bench_id_capacity) = (bench_id_t){
  1770. .name = name,
  1771. .defines = defines,
  1772. .define_count = define_count,
  1773. };
  1774. }
  1775. // do the thing
  1776. op();
  1777. // cleanup (need to be done for valgrind benching)
  1778. bench_define_cleanup();
  1779. if (bench_overrides) {
  1780. for (size_t i = 0; i < bench_override_count; i++) {
  1781. free((void*)bench_overrides[i].defines);
  1782. }
  1783. free((void*)bench_overrides);
  1784. }
  1785. if (bench_geometry_capacity) {
  1786. free((void*)bench_geometries);
  1787. }
  1788. if (bench_id_capacity) {
  1789. for (size_t i = 0; i < bench_id_count; i++) {
  1790. free((void*)bench_ids[i].defines);
  1791. }
  1792. free((void*)bench_ids);
  1793. }
  1794. }