logo

utils-std

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

time.c (3774B)

    

  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 "../lib/getopt_nolong.h"
  8. #include <errno.h>        // errno
  9. #include <stdio.h>        // perror, fprintf
  10. #include <stdlib.h>       // abort
  11. #include <string.h>       // strerror
  12. #include <sys/resource.h> // getrusage
  13. #include <sys/times.h>    // times
  14. #include <sys/wait.h>     // waitpid
  15. #include <unistd.h>       // sysconf, fork, execvp, getopt
  17. const char *argv0 = "time";
  19. enum cmd_time_mode
  20. {
  21. 	CMD_TIME_POSIX = 0,
  22. 	CMD_TIME_VERBOSE = 1,
  23. };
  25. static void
  26. usage(void)
  27. {
  28. 	fprintf(stderr, "Usage: time [-p|-v] command [argument ...]\n");
  29. }
  31. int
  32. main(int argc, char *argv[])
  33. {
  34. 	struct tms tms;
  35. 	int ret = 0;
  36. 	enum cmd_time_mode mode = CMD_TIME_POSIX;
  38. 	if(argc <= 1)
  39. 	{
  40. 		usage();
  41. 		return 0;
  42. 	}
  44. 	for(int c = -1; (c = getopt_nolong(argc, argv, ":pv")) != -1;)
  45. 	{
  46. 		switch(c)
  47. 		{
  48. 		case 'p': // POSIX format (default)
  49. 			mode = CMD_TIME_POSIX;
  50. 			break;
  51. 		case 'v':
  52. 			mode = CMD_TIME_VERBOSE;
  53. 			break;
  54. 		case ':':
  55. 			fprintf(stderr, "time: error: Missing operand for option: '-%c'\n", optopt);
  56. 			usage();
  57. 			return 1;
  58. 		case '?':
  59. 			if(!got_long_opt) fprintf(stderr, "time: error: Unrecognised option: '-%c'\n", optopt);
  60. 			usage();
  61. 			return 1;
  62. 		}
  63. 	}
  65. 	argc -= optind;
  66. 	argv += optind;
  67. 	(void)argc;
  69. 	long ticks = sysconf(_SC_CLK_TCK);
  70. 	if(ticks <= 0)
  71. 	{
  72. 		perror("time: error: sysconf(_SC_CLK_TCK)");
  73. 		return 1;
  74. 	}
  76. 	clock_t t0 = times(&tms);
  77. 	if(t0 == (clock_t)-1)
  78. 	{
  79. 		perror("time: error: times");
  80. 		return 1;
  81. 	}
  83. 	// Note: posix_spawnp seems to lack enough error information
  84. 	pid_t pid = fork();
  85. 	switch(pid)
  86. 	{
  87. 	case -1:
  88. 		perror("time: error: fork");
  89. 		return 1;
  90. 	case 0:
  91. 		execvp(argv[0], argv);
  92. 		ret = 126 + (errno == ENOENT);
  93. 		fprintf(stderr, "time: error: Failed executing '%s': %s\n", argv[0], strerror(errno));
  94. 		return ret;
  95. 	default:
  96. 		break;
  97. 	}
  99. 	int status = 0;
  100. 	waitpid(pid, &status, 0);
  102. 	int t1 = times(&tms);
  103. 	if(t1 == (clock_t)-1)
  104. 	{
  105. 		perror("time: error: times");
  106. 		return 1;
  107. 	}
  109. 	if(WIFSIGNALED(status))
  110. 	{
  111. 		fprintf(stderr, "time: error: Command terminated by signal %d\n", WTERMSIG(status));
  112. 		ret = 128 + WTERMSIG(status);
  113. 	}
  115. 	struct rusage usage;
  116. 	if(getrusage(RUSAGE_CHILDREN, &usage) != 0)
  117. 	{
  118. 		fprintf(
  119. 		    stderr, "time: error: Failed getting resource usage of children: %s\n", strerror(errno));
  120. 		return 1;
  121. 	}
  123. 	double utime_s = usage.ru_utime.tv_sec + (usage.ru_utime.tv_usec * 0.000001);
  124. 	double stime_s = usage.ru_stime.tv_sec + (usage.ru_stime.tv_usec * 0.000001);
  126. 	switch(mode)
  127. 	{
  128. 	case CMD_TIME_POSIX:
  129. 		fprintf(stderr, "real %f\nuser %f\nsys %f\n", (t1 - t0) / (double)ticks, utime_s, stime_s);
  130. 		break;
  131. 	case CMD_TIME_VERBOSE:
  132. 		fprintf(stderr, "\tExecutable: %s\n", argv[0]);
  133. 		fprintf(stderr, "\tReal time  : %06f\n", (t1 - t0) / (double)ticks);
  134. 		fprintf(stderr, "\tUser time  : %06f\n", utime_s);
  135. 		fprintf(stderr, "\tSystem time: %06f\n", stime_s);
  136. #ifdef __linux__
  137. 		fprintf(stderr, "\tMaximum resident set size (maxrss): %ld kB\n", usage.ru_maxrss);
  138. 		fprintf(stderr, "\tMinor (no I/O required) page faults (minflt): %ld\n", usage.ru_minflt);
  139. 		fprintf(stderr, "\tMajor (I/O required) page faults (majflt): %ld\n", usage.ru_majflt);
  140. 		fprintf(stderr, "\tMajor (I/O required) page faults (majflt): %ld\n", usage.ru_majflt);
  141. 		fprintf(stderr, "\tVoluntary context switch (nvcsw): %ld\n", usage.ru_nvcsw);
  142. 		fprintf(stderr, "\tInvoluntary context switch (nivcsw): %ld\n", usage.ru_nivcsw);
  143. #endif
  145. 		if(WIFEXITED(status)) fprintf(stderr, "\tExit status: %d\n", WEXITSTATUS(status));
  147. 		break;
  148. 	default:
  149. 		abort();
  150. 	}
  152. 	if(WIFEXITED(status))
  153. 	{
  154. 		ret = WEXITSTATUS(status);
  155. 	}
  157. 	return ret;
  158. }