utils-std

strings.c (4675B)

// utils-std: Collection of commonly available Unix tools
// SPDX-FileCopyrightText: 2017 Haelwenn (lanodan) Monnier <contact+utils@hacktivis.me>
// SPDX-License-Identifier: MPL-2.0
#define _POSIX_C_SOURCE 200809L
#include "../config.h"
#include "../libutils/getopt_nolong.h"
#include <ctype.h>  /* isprint() */
#include <errno.h>  /* errno */
#include <fcntl.h>  /* open(), O_RDONLY */
#include <limits.h> /* LONG_MAX */
#include <stdio.h>  /* fprintf(), BUFSIZ */
#include <stdlib.h> /* strtol() */
#include <string.h> /* strerror(), strncmp() */
#include <unistd.h> /* read(), write(), close() */
#ifdef HAS_GETOPT_LONG
#include <getopt.h>
#endif
const char *argv0 = "strings";
size_t opt_min_strlen = 4;
const char *opt_offset_format = NULL;
char delim = '\n';
static int
print_string(char *buffer, size_t buflen, size_t offset)
{
	int ret = 0;
	int wrote = 0;
	if(opt_offset_format)
	{
		ret = printf(opt_offset_format, offset);
		if(ret < 0) return ret;
		wrote += ret;
	}
	ret = fwrite(buffer, buflen, 1, stdout);
	if(ret < 0) return ret;
	wrote += ret;
	return wrote;
}
static int
concat(int fd, const char *fdname)
{
	ssize_t c;
	static char read_buf[4096];
	static char write_buf[4096];
	size_t write_pos = 0;
	size_t offset = 0;
	while((c = read(fd, read_buf, sizeof(read_buf))) > 0)
	{
		int read_pos = 0;
		for(; read_pos < c; read_pos++)
		{
			char b = read_buf[read_pos];
			if(isprint(b))
			{
				write_buf[write_pos++] = b;
				if(write_pos < (sizeof(write_buf) - 1)) continue;
			}
			if(write_pos >= opt_min_strlen)
			{
				write_buf[write_pos] = delim;
				if(print_string(write_buf, write_pos + 1, offset) < 0)
				{
					fprintf(stderr, "strings: error: Failed writing resulting string: %s\n", strerror(errno));
					return 1;
				}
			}
			offset += write_pos;
			offset++;
			write_pos = 0;
		}
	}
	if(c < 0)
	{
		fprintf(stderr, "strings: error: Failed reading ‘%s’: %s\n", fdname, strerror(errno));
		errno = 0;
		return 1;
	}
	return 0;
}
static void
usage(void)
{
	fprintf(stderr, "strings: [-az] [-t format] [-n number] [file...]\n");
}
int
main(int argc, char *argv[])
{
#ifdef HAS_GETOPT_LONG
	// Strictly for GNUisms compatibility so no long-only options
	// clang-format off
	static struct option opts[] = {
		{"all", no_argument, NULL, 'a'},
		{"bytes", required_argument, NULL, 'n'},
		{"radix", required_argument, NULL, 't'},
		{0, 0, 0, 0},
	};
	// clang-format on
	// Need + as first character to get POSIX-style option parsing
	for(int c = -1; (c = getopt_long(argc, argv, "+:an:t:z", opts, NULL)) != -1;)
#else
	for(int c = -1; (c = getopt_nolong(argc, argv, ":an:t:z")) != -1;)
#endif
	{
		char *endptr = NULL;
		switch(c)
		{
		case 'a':
			/* Structure is always ignored */
			break;
		case 'n':
			opt_min_strlen = strtol(optarg, &endptr, 10);
			if(endptr && *endptr != 0)
			{
				// extraneous characters is invalid
				errno = EINVAL;
			}
			if(errno != 0)
			{
				fprintf(stderr, "strings: error: Option `-n %s`: %s\n", optarg, strerror(errno));
				usage();
				return 1;
			}
			if(opt_min_strlen < 1)
			{
				fprintf(stderr, "strings: error: Option `-n %s` is too small\n", optarg);
				usage();
				return 1;
			}
			if(opt_min_strlen == LONG_MAX || opt_min_strlen > 4096)
			{
				fprintf(stderr, "strings: error: Option `-n %s` is too large\n", optarg);
				usage();
				return 1;
			}
			break;
		case 't':
			if(strnlen(optarg, 2) > 1)
			{
				usage();
				return 1;
			}
			switch(optarg[0])
			{
			case 'o':
				opt_offset_format = "%zo ";
				break;
			case 'x':
				opt_offset_format = "%zx ";
				break;
			case 'd':
				opt_offset_format = "%zd ";
				break;
			default:
				fprintf(stderr, "strings: error: Unknown format: %s\n", optarg);
				usage();
				return 1;
			}
			break;
		case 'z':
			delim = '\0';
			break;
		case ':':
			fprintf(stderr, "strings: error: Missing operand for option: '-%c'\n", optopt);
			usage();
			return 1;
		case '?':
			GETOPT_UNKNOWN_OPT
			usage();
			return 1;
		default:
			abort();
		}
	}
	argc -= optind;
	argv += optind;
	if(argc < 1)
	{
		return concat(0, "<stdin>");
	}
	for(int argi = 0; argi < argc; argi++)
	{
		if(strncmp(argv[argi], "-", 2) == 0)
		{
			if(concat(0, "<stdin>") != 0)
			{
				return 1;
			}
		}
		else
		{
			int fd = open(argv[argi], O_RDONLY);
			if(fd < 0)
			{
				fprintf(stderr, "strings: error: Failed opening ‘%s’: %s\n", argv[argi], strerror(errno));
				return 1;
			}
			if(concat(fd, argv[argi]) != 0)
			{
				return 1;
			}
			if(close(fd) < 0)
			{
				fprintf(stderr, "strings: error: Failed closing ‘%s’: %s\n", argv[argi], strerror(errno));
				return 1;
			}
		}
	}
	return 0;
}