logo

utils-std

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

truncation.c (2928B)

    

  1. // Collection of Unix tools, comparable to coreutils
  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 "truncation.h"
  8. #include <assert.h>
  9. #include <ctype.h> // toupper
  10. #include <errno.h>
  11. #include <stdio.h>    // fprintf
  12. #include <stdlib.h>   // strtol
  13. #include <string.h>   // strerror
  14. #include <sys/stat.h> // fstat
  15. #include <unistd.h>   // ftruncate
  17. extern char *argv0;
  19. int
  20. apply_truncation(int fd, struct truncation tr, char *arg)
  21. {
  22. 	if(tr.op != OP_SET)
  23. 	{
  24. 		struct stat stats;
  25. 		if(fstat(fd, &stats) < 0)
  26. 		{
  27. 			fprintf(
  28. 			    stderr, "truncate: error: Failed to get status of file '%s': %s\n", arg, strerror(errno));
  29. 			return -1;
  30. 		}
  32. 		int q;
  33. 		switch(tr.op)
  34. 		{
  35. 		case OP_INC:
  36. 			tr.size += stats.st_size;
  37. 			break;
  38. 		case OP_DEC:
  39. 			tr.size = stats.st_size - tr.size;
  40. 			if(tr.size < 0) tr.size = 0;
  41. 			break;
  42. 		case OP_CHK_DOWN:
  43. 			q = stats.st_size / tr.size;
  45. 			tr.size = tr.size * q;
  46. 			break;
  47. 		case OP_CHK_UP:
  48. 			q = stats.st_size / tr.size;
  49. 			if(stats.st_size % tr.size) q++;
  50. 			tr.size = tr.size * q;
  51. 			break;
  52. 		default:
  53. 			abort();
  54. 		}
  55. 	}
  57. 	if(ftruncate(fd, tr.size) < 0)
  58. 	{
  59. 		fprintf(stderr,
  60. 		        "truncate: error: Failed to truncate '%s' to %ld: %s\n",
  61. 		        arg,
  62. 		        tr.size,
  63. 		        strerror(errno));
  64. 		return -1;
  65. 	}
  67. 	return 0;
  68. }
  70. int
  71. apply_size_suffix(unsigned long *size, char *endptr)
  72. {
  73. 	char units[] = "KMGTPEZ";
  75. 	if(endptr[0] == '\0') return 0;
  77. 	size_t i = 0;
  78. 	unsigned long si = 1, iec = 1;
  79. 	char pfx = toupper(endptr[0]);
  80. 	for(; i < sizeof(units); i++)
  81. 	{
  82. 		si *= 1000;
  83. 		iec *= 1024;
  84. 		if(units[i] == pfx) break;
  85. 	}
  86. 	if(i >= sizeof(units) || si == 1)
  87. 	{
  88. 		fprintf(stderr, "%s: error: Unrecognised unit '%s'\n", argv0, endptr);
  89. 		return -1;
  90. 	}
  92. 	if(endptr[1] == 0 || (endptr[1] == 'i' && endptr[2] == 'B' && endptr[3] == 0))
  93. 	{
  94. 		*size *= iec;
  95. 		return 0;
  96. 	}
  97. 	if(endptr[1] == 'B' && endptr[2] == 0)
  98. 	{
  99. 		*size *= si;
  100. 		return 0;
  101. 	}
  103. 	fprintf(stderr, "%s: error: Unrecognised suffix '%s'\n", argv0, endptr);
  104. 	return -1;
  105. }
  107. int
  108. parse_size(const char *arg, struct truncation *buf)
  109. {
  110. 	assert(arg != NULL);
  111. 	assert(buf != NULL);
  113. 	if(arg[0] == 0)
  114. 	{
  115. 		fprintf(stderr, "%s: error: Got empty size argument\n", argv0);
  116. 		buf->size = 0;
  117. 		return -1;
  118. 	}
  120. 	enum operation_e op = OP_SET;
  122. 	switch(arg[0])
  123. 	{
  124. 	case '+':
  125. 		op = OP_INC;
  126. 		arg++;
  127. 		break;
  128. 	case '-':
  129. 		op = OP_DEC;
  130. 		arg++;
  131. 		break;
  132. 	case '/':
  133. 		op = OP_CHK_DOWN;
  134. 		arg++;
  135. 		break;
  136. 	case '%':
  137. 		op = OP_CHK_UP;
  138. 		arg++;
  139. 		break;
  140. 		// no default case intended
  141. 	}
  143. 	char *endptr = NULL;
  144. 	unsigned long size = strtoul(arg, &endptr, 10);
  145. 	if(errno != 0)
  146. 	{
  147. 		fprintf(
  148. 		    stderr, "%s: error: Failed parsing '%s' as a number: %s\n", argv0, arg, strerror(errno));
  149. 		return -1;
  150. 	}
  152. 	if(endptr != NULL)
  153. 	{
  154. 		if(apply_size_suffix(&size, endptr) < 0) return -1;
  155. 	}
  157. 	buf->size = size;
  158. 	buf->op = op;
  159. 	return 0;
  160. }