logo

utils-std

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

mktemp.c (1913B)

    

  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
  6. #include <errno.h>
  7. #include <limits.h> // PATH_MAX
  8. #include <stdbool.h>
  9. #include <stdio.h>  // fprintf
  10. #include <stdlib.h> // getenv
  11. #include <string.h> // strerror
  12. #include <unistd.h> // getopt
  14. int
  15. main(int argc, char *argv[])
  16. {
  17. 	bool o_create_dir = false, o_quiet = false;
  18. 	char template[PATH_MAX] = "tmp.XXXXXXXXXX";
  19. 	const char *tmpdir = NULL;
  21. 	int c = -1;
  22. 	while((c = getopt(argc, argv, ":dqp:t")) != -1)
  23. 	{
  24. 		switch(c)
  25. 		{
  26. 		case 'd':
  27. 			o_create_dir = true;
  28. 			break;
  29. 		case 'q':
  30. 			o_quiet = true;
  31. 			break;
  32. 		case 'p':
  33. 			tmpdir = optarg;
  34. 			break;
  35. 		case 't':
  36. 			if(tmpdir == NULL) tmpdir = getenv("TMPDIR");
  37. 			if(tmpdir == NULL) tmpdir = "/tmp";
  38. 			break;
  39. 		default:
  40. 			abort();
  41. 		}
  42. 	}
  44. 	argc -= optind;
  45. 	argv += optind;
  47. 	if(argc == 1)
  48. 	{
  49. 		strncpy(template, *argv, PATH_MAX);
  50. 	}
  51. 	else if(argc > 1)
  52. 	{
  53. 		fprintf(stderr, "mktemp: Only one template argument is supported, got %d\n", argc);
  54. 		return 1;
  55. 	}
  57. 	if(tmpdir)
  58. 	{
  59. 		if(chdir(tmpdir) < 0)
  60. 		{
  61. 			fprintf(stderr,
  62. 			        "mktemp: Failed changing directory into tmpdir '%s': %s\n",
  63. 			        tmpdir,
  64. 			        strerror(errno));
  65. 			return 1;
  66. 		}
  68. 		printf("%s/", tmpdir);
  69. 	}
  71. 	if(o_create_dir)
  72. 	{
  73. 		char *dir = mkdtemp(template);
  74. 		if(dir == NULL)
  75. 		{
  76. 			if(!o_quiet)
  77. 				fprintf(stderr, "mktemp: Failed creating random directory: %s\n", strerror(errno));
  79. 			return 1;
  80. 		}
  82. 		puts(dir);
  84. 		return 0;
  85. 	}
  87. 	int fd = mkstemp(template);
  88. 	if(fd < 0)
  89. 	{
  90. 		if(!o_quiet) fprintf(stderr, "mktemp: Failed creating random file: %s\n", strerror(errno));
  92. 		return 1;
  93. 	}
  95. 	puts(template);
  97. 	if(close(fd) < 0)
  98. 	{
  99. 		if(!o_quiet) fprintf(stderr, "mktemp: Failed closing file descriptor: %s\n", strerror(errno));
  101. 		return 1;
  102. 	}
  104. 	return 0;
  105. }