1333fc21eSDavid E. O'Brien #include <sys/cdefs.h> 2333fc21eSDavid E. O'Brien __FBSDID("$FreeBSD$"); 3333fc21eSDavid E. O'Brien 48ca5c256SDaniel C. Sobral #include <stdio.h> 58ca5c256SDaniel C. Sobral #include <string.h> 68ca5c256SDaniel C. Sobral 78ca5c256SDaniel C. Sobral /* 88ca5c256SDaniel C. Sobral - split - divide a string into fields, like awk split() 98ca5c256SDaniel C. Sobral = int split(char *string, char *fields[], int nfields, char *sep); 108eba7ea3SEnji Cooper - fields: list is not NULL-terminated 118eba7ea3SEnji Cooper - nfields: number of entries available in fields[] 128eba7ea3SEnji Cooper - sep: "" white, "c" single char, "ab" [ab]+ 138ca5c256SDaniel C. Sobral */ 148ca5c256SDaniel C. Sobral int /* number of fields, including overflow */ 158eba7ea3SEnji Cooper split(char *string, char *fields[], int nfields, char *sep) 168ca5c256SDaniel C. Sobral { 178fb3f3f6SDavid E. O'Brien char *p = string; 188fb3f3f6SDavid E. O'Brien char c; /* latest character */ 198fb3f3f6SDavid E. O'Brien char sepc = sep[0]; 208fb3f3f6SDavid E. O'Brien char sepc2; 218fb3f3f6SDavid E. O'Brien int fn; 228fb3f3f6SDavid E. O'Brien char **fp = fields; 238fb3f3f6SDavid E. O'Brien char *sepp; 248fb3f3f6SDavid E. O'Brien int trimtrail; 258ca5c256SDaniel C. Sobral 268ca5c256SDaniel C. Sobral /* white space */ 278ca5c256SDaniel C. Sobral if (sepc == '\0') { 288ca5c256SDaniel C. Sobral while ((c = *p++) == ' ' || c == '\t') 298ca5c256SDaniel C. Sobral continue; 308ca5c256SDaniel C. Sobral p--; 318ca5c256SDaniel C. Sobral trimtrail = 1; 328ca5c256SDaniel C. Sobral sep = " \t"; /* note, code below knows this is 2 long */ 338ca5c256SDaniel C. Sobral sepc = ' '; 348ca5c256SDaniel C. Sobral } else 358ca5c256SDaniel C. Sobral trimtrail = 0; 368ca5c256SDaniel C. Sobral sepc2 = sep[1]; /* now we can safely pick this up */ 378ca5c256SDaniel C. Sobral 388ca5c256SDaniel C. Sobral /* catch empties */ 398ca5c256SDaniel C. Sobral if (*p == '\0') 408ca5c256SDaniel C. Sobral return(0); 418ca5c256SDaniel C. Sobral 428ca5c256SDaniel C. Sobral /* single separator */ 438ca5c256SDaniel C. Sobral if (sepc2 == '\0') { 448ca5c256SDaniel C. Sobral fn = nfields; 458ca5c256SDaniel C. Sobral for (;;) { 468ca5c256SDaniel C. Sobral *fp++ = p; 478ca5c256SDaniel C. Sobral fn--; 488ca5c256SDaniel C. Sobral if (fn == 0) 498ca5c256SDaniel C. Sobral break; 508ca5c256SDaniel C. Sobral while ((c = *p++) != sepc) 518ca5c256SDaniel C. Sobral if (c == '\0') 528ca5c256SDaniel C. Sobral return(nfields - fn); 538ca5c256SDaniel C. Sobral *(p-1) = '\0'; 548ca5c256SDaniel C. Sobral } 558ca5c256SDaniel C. Sobral /* we have overflowed the fields vector -- just count them */ 568ca5c256SDaniel C. Sobral fn = nfields; 578ca5c256SDaniel C. Sobral for (;;) { 588ca5c256SDaniel C. Sobral while ((c = *p++) != sepc) 598ca5c256SDaniel C. Sobral if (c == '\0') 608ca5c256SDaniel C. Sobral return(fn); 618ca5c256SDaniel C. Sobral fn++; 628ca5c256SDaniel C. Sobral } 638ca5c256SDaniel C. Sobral /* not reached */ 648ca5c256SDaniel C. Sobral } 658ca5c256SDaniel C. Sobral 668ca5c256SDaniel C. Sobral /* two separators */ 678ca5c256SDaniel C. Sobral if (sep[2] == '\0') { 688ca5c256SDaniel C. Sobral fn = nfields; 698ca5c256SDaniel C. Sobral for (;;) { 708ca5c256SDaniel C. Sobral *fp++ = p; 718ca5c256SDaniel C. Sobral fn--; 728ca5c256SDaniel C. Sobral while ((c = *p++) != sepc && c != sepc2) 738ca5c256SDaniel C. Sobral if (c == '\0') { 748ca5c256SDaniel C. Sobral if (trimtrail && **(fp-1) == '\0') 758ca5c256SDaniel C. Sobral fn++; 768ca5c256SDaniel C. Sobral return(nfields - fn); 778ca5c256SDaniel C. Sobral } 788ca5c256SDaniel C. Sobral if (fn == 0) 798ca5c256SDaniel C. Sobral break; 808ca5c256SDaniel C. Sobral *(p-1) = '\0'; 818ca5c256SDaniel C. Sobral while ((c = *p++) == sepc || c == sepc2) 828ca5c256SDaniel C. Sobral continue; 838ca5c256SDaniel C. Sobral p--; 848ca5c256SDaniel C. Sobral } 858ca5c256SDaniel C. Sobral /* we have overflowed the fields vector -- just count them */ 868ca5c256SDaniel C. Sobral fn = nfields; 878ca5c256SDaniel C. Sobral while (c != '\0') { 888ca5c256SDaniel C. Sobral while ((c = *p++) == sepc || c == sepc2) 898ca5c256SDaniel C. Sobral continue; 908ca5c256SDaniel C. Sobral p--; 918ca5c256SDaniel C. Sobral fn++; 928ca5c256SDaniel C. Sobral while ((c = *p++) != '\0' && c != sepc && c != sepc2) 938ca5c256SDaniel C. Sobral continue; 948ca5c256SDaniel C. Sobral } 958ca5c256SDaniel C. Sobral /* might have to trim trailing white space */ 968ca5c256SDaniel C. Sobral if (trimtrail) { 978ca5c256SDaniel C. Sobral p--; 988ca5c256SDaniel C. Sobral while ((c = *--p) == sepc || c == sepc2) 998ca5c256SDaniel C. Sobral continue; 1008ca5c256SDaniel C. Sobral p++; 1018ca5c256SDaniel C. Sobral if (*p != '\0') { 1028ca5c256SDaniel C. Sobral if (fn == nfields+1) 1038ca5c256SDaniel C. Sobral *p = '\0'; 1048ca5c256SDaniel C. Sobral fn--; 1058ca5c256SDaniel C. Sobral } 1068ca5c256SDaniel C. Sobral } 1078ca5c256SDaniel C. Sobral return(fn); 1088ca5c256SDaniel C. Sobral } 1098ca5c256SDaniel C. Sobral 1108ca5c256SDaniel C. Sobral /* n separators */ 1118ca5c256SDaniel C. Sobral fn = 0; 1128ca5c256SDaniel C. Sobral for (;;) { 1138ca5c256SDaniel C. Sobral if (fn < nfields) 1148ca5c256SDaniel C. Sobral *fp++ = p; 1158ca5c256SDaniel C. Sobral fn++; 1168ca5c256SDaniel C. Sobral for (;;) { 1178ca5c256SDaniel C. Sobral c = *p++; 1188ca5c256SDaniel C. Sobral if (c == '\0') 1198ca5c256SDaniel C. Sobral return(fn); 1208ca5c256SDaniel C. Sobral sepp = sep; 1218ca5c256SDaniel C. Sobral while ((sepc = *sepp++) != '\0' && sepc != c) 1228ca5c256SDaniel C. Sobral continue; 1238ca5c256SDaniel C. Sobral if (sepc != '\0') /* it was a separator */ 1248ca5c256SDaniel C. Sobral break; 1258ca5c256SDaniel C. Sobral } 1268ca5c256SDaniel C. Sobral if (fn < nfields) 1278ca5c256SDaniel C. Sobral *(p-1) = '\0'; 1288ca5c256SDaniel C. Sobral for (;;) { 1298ca5c256SDaniel C. Sobral c = *p++; 1308ca5c256SDaniel C. Sobral sepp = sep; 1318ca5c256SDaniel C. Sobral while ((sepc = *sepp++) != '\0' && sepc != c) 1328ca5c256SDaniel C. Sobral continue; 1338ca5c256SDaniel C. Sobral if (sepc == '\0') /* it wasn't a separator */ 1348ca5c256SDaniel C. Sobral break; 1358ca5c256SDaniel C. Sobral } 1368ca5c256SDaniel C. Sobral p--; 1378ca5c256SDaniel C. Sobral } 1388ca5c256SDaniel C. Sobral 1398ca5c256SDaniel C. Sobral /* not reached */ 1408ca5c256SDaniel C. Sobral } 1418ca5c256SDaniel C. Sobral 1428ca5c256SDaniel C. Sobral #ifdef TEST_SPLIT 1438ca5c256SDaniel C. Sobral 1448ca5c256SDaniel C. Sobral 1458ca5c256SDaniel C. Sobral /* 1468ca5c256SDaniel C. Sobral * test program 1478ca5c256SDaniel C. Sobral * pgm runs regression 1488ca5c256SDaniel C. Sobral * pgm sep splits stdin lines by sep 1498ca5c256SDaniel C. Sobral * pgm str sep splits str by sep 1508ca5c256SDaniel C. Sobral * pgm str sep n splits str by sep n times 1518ca5c256SDaniel C. Sobral */ 1528ca5c256SDaniel C. Sobral int 1538eba7ea3SEnji Cooper main(int argc, char *argv[]) 1548ca5c256SDaniel C. Sobral { 1558ca5c256SDaniel C. Sobral char buf[512]; 1568fb3f3f6SDavid E. O'Brien int n; 1578ca5c256SDaniel C. Sobral # define MNF 10 1588ca5c256SDaniel C. Sobral char *fields[MNF]; 1598ca5c256SDaniel C. Sobral 1608ca5c256SDaniel C. Sobral if (argc > 4) 1618ca5c256SDaniel C. Sobral for (n = atoi(argv[3]); n > 0; n--) { 1628ca5c256SDaniel C. Sobral (void) strcpy(buf, argv[1]); 1638ca5c256SDaniel C. Sobral } 1648ca5c256SDaniel C. Sobral else if (argc > 3) 1658ca5c256SDaniel C. Sobral for (n = atoi(argv[3]); n > 0; n--) { 1668ca5c256SDaniel C. Sobral (void) strcpy(buf, argv[1]); 1678ca5c256SDaniel C. Sobral (void) split(buf, fields, MNF, argv[2]); 1688ca5c256SDaniel C. Sobral } 1698ca5c256SDaniel C. Sobral else if (argc > 2) 1708ca5c256SDaniel C. Sobral dosplit(argv[1], argv[2]); 1718ca5c256SDaniel C. Sobral else if (argc > 1) 1728ca5c256SDaniel C. Sobral while (fgets(buf, sizeof(buf), stdin) != NULL) { 1738ca5c256SDaniel C. Sobral buf[strlen(buf)-1] = '\0'; /* stomp newline */ 1748ca5c256SDaniel C. Sobral dosplit(buf, argv[1]); 1758ca5c256SDaniel C. Sobral } 1768ca5c256SDaniel C. Sobral else 1778ca5c256SDaniel C. Sobral regress(); 1788ca5c256SDaniel C. Sobral 1798ca5c256SDaniel C. Sobral exit(0); 1808ca5c256SDaniel C. Sobral } 1818ca5c256SDaniel C. Sobral 1828eba7ea3SEnji Cooper void 1838eba7ea3SEnji Cooper dosplit(char *string, char *seps) 1848ca5c256SDaniel C. Sobral { 1858ca5c256SDaniel C. Sobral # define NF 5 1868ca5c256SDaniel C. Sobral char *fields[NF]; 1878fb3f3f6SDavid E. O'Brien int nf; 1888ca5c256SDaniel C. Sobral 1898ca5c256SDaniel C. Sobral nf = split(string, fields, NF, seps); 1908ca5c256SDaniel C. Sobral print(nf, NF, fields); 1918ca5c256SDaniel C. Sobral } 1928ca5c256SDaniel C. Sobral 1938eba7ea3SEnji Cooper void 1948eba7ea3SEnji Cooper print(int nf, int nfp, char *fields[]) 1958ca5c256SDaniel C. Sobral { 1968fb3f3f6SDavid E. O'Brien int fn; 1978fb3f3f6SDavid E. O'Brien int bound; 1988ca5c256SDaniel C. Sobral 1998ca5c256SDaniel C. Sobral bound = (nf > nfp) ? nfp : nf; 2008ca5c256SDaniel C. Sobral printf("%d:\t", nf); 2018ca5c256SDaniel C. Sobral for (fn = 0; fn < bound; fn++) 2028ca5c256SDaniel C. Sobral printf("\"%s\"%s", fields[fn], (fn+1 < nf) ? ", " : "\n"); 2038ca5c256SDaniel C. Sobral } 2048ca5c256SDaniel C. Sobral 2058ca5c256SDaniel C. Sobral #define RNF 5 /* some table entries know this */ 2068ca5c256SDaniel C. Sobral struct { 2078ca5c256SDaniel C. Sobral char *str; 2088ca5c256SDaniel C. Sobral char *seps; 2098ca5c256SDaniel C. Sobral int nf; 2108ca5c256SDaniel C. Sobral char *fi[RNF]; 2118ca5c256SDaniel C. Sobral } tests[] = { 2128ca5c256SDaniel C. Sobral "", " ", 0, { "" }, 2138ca5c256SDaniel C. Sobral " ", " ", 2, { "", "" }, 2148ca5c256SDaniel C. Sobral "x", " ", 1, { "x" }, 2158ca5c256SDaniel C. Sobral "xy", " ", 1, { "xy" }, 2168ca5c256SDaniel C. Sobral "x y", " ", 2, { "x", "y" }, 2178ca5c256SDaniel C. Sobral "abc def g ", " ", 5, { "abc", "def", "", "g", "" }, 2188ca5c256SDaniel C. Sobral " a bcd", " ", 4, { "", "", "a", "bcd" }, 2198ca5c256SDaniel C. Sobral "a b c d e f", " ", 6, { "a", "b", "c", "d", "e f" }, 2208ca5c256SDaniel C. Sobral " a b c d ", " ", 6, { "", "a", "b", "c", "d " }, 2218ca5c256SDaniel C. Sobral 2228ca5c256SDaniel C. Sobral "", " _", 0, { "" }, 2238ca5c256SDaniel C. Sobral " ", " _", 2, { "", "" }, 2248ca5c256SDaniel C. Sobral "x", " _", 1, { "x" }, 2258ca5c256SDaniel C. Sobral "x y", " _", 2, { "x", "y" }, 2268ca5c256SDaniel C. Sobral "ab _ cd", " _", 2, { "ab", "cd" }, 2278ca5c256SDaniel C. Sobral " a_b c ", " _", 5, { "", "a", "b", "c", "" }, 2288ca5c256SDaniel C. Sobral "a b c_d e f", " _", 6, { "a", "b", "c", "d", "e f" }, 2298ca5c256SDaniel C. Sobral " a b c d ", " _", 6, { "", "a", "b", "c", "d " }, 2308ca5c256SDaniel C. Sobral 2318ca5c256SDaniel C. Sobral "", " _~", 0, { "" }, 2328ca5c256SDaniel C. Sobral " ", " _~", 2, { "", "" }, 2338ca5c256SDaniel C. Sobral "x", " _~", 1, { "x" }, 2348ca5c256SDaniel C. Sobral "x y", " _~", 2, { "x", "y" }, 2358ca5c256SDaniel C. Sobral "ab _~ cd", " _~", 2, { "ab", "cd" }, 2368ca5c256SDaniel C. Sobral " a_b c~", " _~", 5, { "", "a", "b", "c", "" }, 2378ca5c256SDaniel C. Sobral "a b_c d~e f", " _~", 6, { "a", "b", "c", "d", "e f" }, 2388ca5c256SDaniel C. Sobral "~a b c d ", " _~", 6, { "", "a", "b", "c", "d " }, 2398ca5c256SDaniel C. Sobral 2408ca5c256SDaniel C. Sobral "", " _~-", 0, { "" }, 2418ca5c256SDaniel C. Sobral " ", " _~-", 2, { "", "" }, 2428ca5c256SDaniel C. Sobral "x", " _~-", 1, { "x" }, 2438ca5c256SDaniel C. Sobral "x y", " _~-", 2, { "x", "y" }, 2448ca5c256SDaniel C. Sobral "ab _~- cd", " _~-", 2, { "ab", "cd" }, 2458ca5c256SDaniel C. Sobral " a_b c~", " _~-", 5, { "", "a", "b", "c", "" }, 2468ca5c256SDaniel C. Sobral "a b_c-d~e f", " _~-", 6, { "a", "b", "c", "d", "e f" }, 2478ca5c256SDaniel C. Sobral "~a-b c d ", " _~-", 6, { "", "a", "b", "c", "d " }, 2488ca5c256SDaniel C. Sobral 2498ca5c256SDaniel C. Sobral "", " ", 0, { "" }, 2508ca5c256SDaniel C. Sobral " ", " ", 2, { "", "" }, 2518ca5c256SDaniel C. Sobral "x", " ", 1, { "x" }, 2528ca5c256SDaniel C. Sobral "xy", " ", 1, { "xy" }, 2538ca5c256SDaniel C. Sobral "x y", " ", 2, { "x", "y" }, 2548ca5c256SDaniel C. Sobral "abc def g ", " ", 4, { "abc", "def", "g", "" }, 2558ca5c256SDaniel C. Sobral " a bcd", " ", 3, { "", "a", "bcd" }, 2568ca5c256SDaniel C. Sobral "a b c d e f", " ", 6, { "a", "b", "c", "d", "e f" }, 2578ca5c256SDaniel C. Sobral " a b c d ", " ", 6, { "", "a", "b", "c", "d " }, 2588ca5c256SDaniel C. Sobral 2598ca5c256SDaniel C. Sobral "", "", 0, { "" }, 2608ca5c256SDaniel C. Sobral " ", "", 0, { "" }, 2618ca5c256SDaniel C. Sobral "x", "", 1, { "x" }, 2628ca5c256SDaniel C. Sobral "xy", "", 1, { "xy" }, 2638ca5c256SDaniel C. Sobral "x y", "", 2, { "x", "y" }, 2648ca5c256SDaniel C. Sobral "abc def g ", "", 3, { "abc", "def", "g" }, 2658ca5c256SDaniel C. Sobral "\t a bcd", "", 2, { "a", "bcd" }, 2668ca5c256SDaniel C. Sobral " a \tb\t c ", "", 3, { "a", "b", "c" }, 2678ca5c256SDaniel C. Sobral "a b c d e ", "", 5, { "a", "b", "c", "d", "e" }, 2688ca5c256SDaniel C. Sobral "a b\tc d e f", "", 6, { "a", "b", "c", "d", "e f" }, 2698ca5c256SDaniel C. Sobral " a b c d e f ", "", 6, { "a", "b", "c", "d", "e f " }, 2708ca5c256SDaniel C. Sobral 2718ca5c256SDaniel C. Sobral NULL, NULL, 0, { NULL }, 2728ca5c256SDaniel C. Sobral }; 2738ca5c256SDaniel C. Sobral 2748eba7ea3SEnji Cooper void 2758eba7ea3SEnji Cooper regress(void) 2768ca5c256SDaniel C. Sobral { 2778ca5c256SDaniel C. Sobral char buf[512]; 2788fb3f3f6SDavid E. O'Brien int n; 2798ca5c256SDaniel C. Sobral char *fields[RNF+1]; 2808fb3f3f6SDavid E. O'Brien int nf; 2818fb3f3f6SDavid E. O'Brien int i; 2828fb3f3f6SDavid E. O'Brien int printit; 2838fb3f3f6SDavid E. O'Brien char *f; 2848ca5c256SDaniel C. Sobral 2858ca5c256SDaniel C. Sobral for (n = 0; tests[n].str != NULL; n++) { 2868ca5c256SDaniel C. Sobral (void) strcpy(buf, tests[n].str); 2878ca5c256SDaniel C. Sobral fields[RNF] = NULL; 2888ca5c256SDaniel C. Sobral nf = split(buf, fields, RNF, tests[n].seps); 2898ca5c256SDaniel C. Sobral printit = 0; 2908ca5c256SDaniel C. Sobral if (nf != tests[n].nf) { 2918ca5c256SDaniel C. Sobral printf("split `%s' by `%s' gave %d fields, not %d\n", 2928ca5c256SDaniel C. Sobral tests[n].str, tests[n].seps, nf, tests[n].nf); 2938ca5c256SDaniel C. Sobral printit = 1; 2948ca5c256SDaniel C. Sobral } else if (fields[RNF] != NULL) { 2958ca5c256SDaniel C. Sobral printf("split() went beyond array end\n"); 2968ca5c256SDaniel C. Sobral printit = 1; 2978ca5c256SDaniel C. Sobral } else { 2988ca5c256SDaniel C. Sobral for (i = 0; i < nf && i < RNF; i++) { 2998ca5c256SDaniel C. Sobral f = fields[i]; 3008ca5c256SDaniel C. Sobral if (f == NULL) 3018ca5c256SDaniel C. Sobral f = "(NULL)"; 3028ca5c256SDaniel C. Sobral if (strcmp(f, tests[n].fi[i]) != 0) { 3038ca5c256SDaniel C. Sobral printf("split `%s' by `%s' field %d is `%s', not `%s'\n", 3048ca5c256SDaniel C. Sobral tests[n].str, tests[n].seps, 3058ca5c256SDaniel C. Sobral i, fields[i], tests[n].fi[i]); 3068ca5c256SDaniel C. Sobral printit = 1; 3078ca5c256SDaniel C. Sobral } 3088ca5c256SDaniel C. Sobral } 3098ca5c256SDaniel C. Sobral } 3108ca5c256SDaniel C. Sobral if (printit) 3118ca5c256SDaniel C. Sobral print(nf, RNF, fields); 3128ca5c256SDaniel C. Sobral } 3138ca5c256SDaniel C. Sobral } 3148ca5c256SDaniel C. Sobral #endif 315