logo

utils-std

Collection of commonly available Unix tools git clone https://anongit.hacktivis.me/git/utils-std.git/

basename.c (2760B)

    

  1. // utils-std: Collection of commonly available Unix tools
  2. // SPDX-FileCopyrightText: 2017 Haelwenn (lanodan) Monnier <contact+utils@hacktivis.me>
  3. // SPDX-License-Identifier: MPL-2.0
  5. #define _POSIX_C_SOURCE 200809L
  7. #include "../config.h"
  8. #include "../libutils/getopt_nolong.h"
  10. #include <libgen.h> // basename
  11. #include <stdbool.h>
  12. #include <stdio.h>  // puts, perror
  13. #include <string.h> // strlen, strncmp
  14. #include <unistd.h> // getopt
  15. #ifdef HAS_GETOPT_LONG
  16. #include <getopt.h>
  17. #endif
  19. const char *argv0 = "basename";
  21. #ifdef __GNUC__
  22. #define _NonNull __attribute__((nonnull))
  23. #else
  24. #define _NonNull
  25. #endif
  27. _NonNull static char *
  28. suffix_basename(char *name, char *suffix)
  29. {
  30. 	char *string = basename(name);
  32. 	size_t suflen = strlen(suffix);
  33. 	size_t len = strlen(string);
  35. 	if(suflen < len && strcmp(&string[len - suflen], suffix) == 0)
  36. 	{
  37. 		string[len - suflen] = '\0';
  38. 	}
  40. 	return string;
  41. }
  43. static void
  44. usage(void)
  45. {
  46. 	fputs("Usage: basename [-z] [path] [suffix]\n"
  47. 	      "       basename [-az] [-s suffix] [path...]\n",
  48. 	      stderr);
  49. }
  51. int
  52. main(int argc, char *argv[])
  53. {
  54. 	bool opt_a = false;
  55. 	char *suffix = NULL;
  56. 	char delim = '\n';
  58. #ifdef HAS_GETOPT_LONG
  59. 	// Strictly for GNUisms compatibility so no long-only options
  60. 	// clang-format off
  61. 	static struct option opts[] = {
  62. 		{"multiple", no_argument, NULL, 'a'},
  63. 		{"suffix", required_argument, NULL, 's'},
  64. 		{"zero", no_argument, NULL, 'z'},
  65. 		{0, 0, 0, 0},
  66. 	};
  67. 	// clang-format on
  69. 	// Need + as first character to get POSIX-style option parsing
  70. 	for(int c = -1; (c = getopt_long(argc, argv, "+:as:z", opts, NULL)) != -1;)
  71. #else
  72. 	for(int c = -1; (c = getopt_nolong(argc, argv, ":as:z")) != -1;)
  73. #endif
  74. 	{
  75. 		switch(c)
  76. 		{
  77. 		case 'a':
  78. 			opt_a = true;
  79. 			break;
  80. 		case 's':
  81. 			opt_a = true;
  82. 			suffix = optarg;
  83. 			break;
  84. 		case 'z':
  85. 			delim = '\0';
  86. 			break;
  87. 		case ':':
  88. 			fprintf(stderr, "%s: error: Missing operand for option '-%c'\n", argv0, optopt);
  89. 			usage();
  90. 			return 1;
  91. 		case '?':
  92. 			GETOPT_UNKNOWN_OPT
  93. 			usage();
  94. 			return 1;
  95. 		}
  96. 	}
  98. 	argc -= optind;
  99. 	argv += optind;
  101. 	if(!opt_a || argc == 0)
  102. 	{
  103. 		int ret = 0;
  105. 		switch(argc)
  106. 		{
  107. 		case 0:
  108. 			ret = printf(".%c", delim);
  109. 			break;
  110. 		case 1:
  111. 			ret = printf("%s%c", basename(argv[0]), delim);
  112. 			break;
  113. 		case 2:
  114. 			ret = printf("%s%c", suffix_basename(argv[0], argv[1]), delim);
  115. 			break;
  116. 		default:
  117. 			usage();
  118. 			return 1;
  119. 		}
  121. 		if(ret < 0)
  122. 		{
  123. 			perror("basename: error: Failed to print result");
  124. 			return 1;
  125. 		}
  127. 		return 0;
  128. 	}
  130. 	for(int argi = 0; argi < argc; argi++)
  131. 	{
  132. 		char *res = NULL;
  133. 		if(suffix != NULL)
  134. 			res = suffix_basename(argv[argi], suffix);
  135. 		else
  136. 			res = basename(argv[argi]);
  138. 		if(printf("%s%c", res, delim) < 0)
  139. 		{
  140. 			perror("basename: error: Failed to print result");
  141. 			return 1;
  142. 		}
  143. 	}
  145. 	return 0;
  146. }