logo

utils-std

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

consent.c (3242B)

    

  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 "./consent.h"
  9. #include <assert.h>
  10. #include <errno.h>
  11. #include <langinfo.h> // nl_langinfo
  12. #include <regex.h>    // regcomp
  13. #include <stdarg.h>   // va_list
  14. #include <stdio.h>    // fprintf, getline,
  15. #include <stdlib.h>   // free
  16. #include <string.h>   // strerror
  17. #include <unistd.h>   // isatty, write
  19. // Needs to be executed after setlocale
  20. regex_t consent_yesexpr_r;
  21. regex_t consent_noexpr_r;
  22. void
  23. consent_init(void)
  24. {
  25. 	char *yesexpr = nl_langinfo(YESEXPR);
  26. 	int yesexpr_ret = regcomp(&consent_yesexpr_r, yesexpr, REG_EXTENDED | REG_NOSUB);
  27. 	if(yesexpr_ret != 0)
  28. 	{
  29. 		char errstr[64] = "";
  30. 		regerror(yesexpr_ret, &consent_yesexpr_r, errstr, 100);
  31. 		fprintf(stderr,
  32. 		        "%s: warning: Got erroneous yesexpr regex /%s/ from locale: %s\n",
  33. 		        argv0,
  34. 		        yesexpr,
  35. 		        errstr);
  37. 		errno = 0;
  38. 		fprintf(stderr, "%s: Fallbacking to /^[Yy]/\n", argv0);
  39. 		yesexpr_ret = regcomp(&consent_yesexpr_r, "^[Yy]", REG_EXTENDED | REG_NOSUB);
  40. 		assert(yesexpr_ret == 0);
  41. 	}
  43. 	char *noexpr = nl_langinfo(NOEXPR);
  44. 	int noexpr_ret = regcomp(&consent_noexpr_r, noexpr, REG_EXTENDED | REG_NOSUB);
  45. 	if(noexpr_ret != 0)
  46. 	{
  47. 		char errstr[64] = "";
  48. 		regerror(noexpr_ret, &consent_noexpr_r, errstr, 100);
  49. 		fprintf(stderr,
  50. 		        "%s: warning: Got erroneous noexpr regex /%s/ from locale: %s\n",
  51. 		        argv0,
  52. 		        noexpr,
  53. 		        errstr);
  55. 		errno = 0;
  56. 		fprintf(stderr, "%s: Fallbacking to /^[Nn]/\n", argv0);
  57. 		noexpr_ret = regcomp(&consent_noexpr_r, "^[Nn]", REG_EXTENDED | REG_NOSUB);
  58. 		assert(noexpr_ret == 0);
  59. 	}
  60. }
  62. void
  63. consent_finish(void)
  64. {
  65. 	regfree(&consent_yesexpr_r);
  66. 	regfree(&consent_noexpr_r);
  67. }
  69. // Consent therefore defaults to no
  70. bool
  71. consentf(const char *restrict fmt, ...)
  72. {
  73. 	bool result = false;
  74. 	char *line = NULL;
  75. 	size_t len = 0;
  77. 	va_list ap;
  79. 	va_start(ap, fmt);
  80. 	int ret = vfprintf(stderr, fmt, ap);
  81. 	va_end(ap);
  83. 	if(ret < 0)
  84. 	{
  85. 		fprintf(stderr, "%s: error: Failed to print user prompt: %s\n", argv0, strerror(errno));
  86. 		errno = 0;
  87. 		goto end;
  88. 	}
  90. 	ssize_t nread = getline(&line, &len, stdin);
  91. 	if(nread < 0)
  92. 	{
  93. 		fprintf(
  94. 		    stderr, "\n%s: error: Failed getting user entry via getline: %s\n", argv0, strerror(errno));
  95. 		errno = 0;
  96. 		goto end;
  97. 	}
  99. 	if(nread == 0)
  100. 	{
  101. 		fprintf(stderr, "%s: Got empty response, considering it false\n", argv0);
  102. 		errno = 0;
  103. 		goto end;
  104. 	}
  106. 	// Doesn't echoes if not a TTY
  107. 	if(!isatty(0)) write(2, line, nread);
  108. 	// isatty changes errno if not a TTY *sigh*
  109. 	errno = 0;
  111. 	if(line[nread - 1] == '\n') line[--nread] = 0;
  113. 	if(line[0] == 0 || line[0] == '\n' || line[0] == '\r')
  114. 	{
  115. 		fprintf(stderr, "%s: Got empty response, considering it false\n", argv0);
  116. 		goto end;
  117. 	}
  119. 	if(regexec(&consent_yesexpr_r, line, 0, 0, 0) == 0)
  120. 	{
  121. 		result = true;
  122. 		goto end;
  123. 	}
  125. 	if(regexec(&consent_noexpr_r, line, 0, 0, 0) == 0)
  126. 	{
  127. 		result = false;
  128. 		goto end;
  129. 	}
  131. 	fprintf(stderr,
  132. 	        "%s: User entry \"%s\" neither affirmative nor negative, considering it false\n",
  133. 	        argv0,
  134. 	        line);
  136. end:
  137. 	if(len != 0) free(line);
  138. 	return result;
  139. }