logo

utils-std

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

basename.c (2242B)

    

  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 <libgen.h> // basename
  8. #include <stdbool.h>
  9. #include <stdio.h>  // puts, perror
  10. #include <string.h> // strlen, strncmp
  11. #include <unistd.h> // getopt
  13. const char *argv0 = "basename";
  15. #ifdef __GNUC__
  16. #define _NonNull __attribute__((nonnull))
  17. #else
  18. #define _NonNull
  19. #endif
  21. _NonNull static char *
  22. suffix_basename(char *name, char *suffix)
  23. {
  24. 	char *string = basename(name);
  26. 	size_t suflen = strlen(suffix);
  27. 	size_t len = strlen(string);
  29. 	if(suflen < len && strcmp(&string[len - suflen], suffix) == 0)
  30. 	{
  31. 		string[len - suflen] = '\0';
  32. 	}
  34. 	return string;
  35. }
  37. static void
  38. usage(void)
  39. {
  40. 	fputs("Usage: basename [-z] [path] [suffix]\n"
  41. 	      "       basename [-az] [-s suffix] [path...]\n",
  42. 	      stderr);
  43. }
  45. int
  46. main(int argc, char *argv[])
  47. {
  48. 	bool opt_a = false;
  49. 	char *suffix = NULL;
  50. 	char delim = '\n';
  52. 	int c = -1;
  53. 	while((c = getopt(argc, argv, ":as:z")) != -1)
  54. 	{
  55. 		switch(c)
  56. 		{
  57. 		case 'a':
  58. 			opt_a = true;
  59. 			break;
  60. 		case 's':
  61. 			opt_a = true;
  62. 			suffix = optarg;
  63. 			break;
  64. 		case 'z':
  65. 			delim = '\0';
  66. 			break;
  67. 		case ':':
  68. 			fprintf(stderr, "%s: error: Missing operand for option '-%c'\n", argv0, optopt);
  69. 			usage();
  70. 			return 1;
  71. 		case '?':
  72. 			fprintf(stderr, "%s: error: Unrecognised option '-%c'\n", argv0, optopt);
  73. 			usage();
  74. 			return 1;
  75. 		}
  76. 	}
  78. 	argc -= optind;
  79. 	argv += optind;
  81. 	if(!opt_a || argc == 0)
  82. 	{
  83. 		int ret = 0;
  85. 		switch(argc)
  86. 		{
  87. 		case 0:
  88. 			ret = printf(".%c", delim);
  89. 			break;
  90. 		case 1:
  91. 			ret = printf("%s%c", basename(argv[0]), delim);
  92. 			break;
  93. 		case 2:
  94. 			ret = printf("%s%c", suffix_basename(argv[0], argv[1]), delim);
  95. 			break;
  96. 		default:
  97. 			usage();
  98. 			return 1;
  99. 		}
  101. 		if(ret < 0)
  102. 		{
  103. 			perror("basename: error: Failed to print result");
  104. 			return 1;
  105. 		}
  107. 		return 0;
  108. 	}
  110. 	for(int argi = 0; argi < argc; argi++)
  111. 	{
  112. 		char *res = NULL;
  113. 		if(suffix != NULL)
  114. 			res = suffix_basename(argv[argi], suffix);
  115. 		else
  116. 			res = basename(argv[argi]);
  118. 		if(printf("%s%c", res, delim) < 0)
  119. 		{
  120. 			perror("basename: error: Failed to print result");
  121. 			return 1;
  122. 		}
  123. 	}
  125. 	return 0;
  126. }