illumos gate

#include <sys/types.h>

#include <sys/stat.h>

#include <fcntl.h>

#include <stdio.h>

#include <unistd.h>

#include <string.h>

#include <stdlib.h>

#include <ctype.h>

#include <limits.h>

static int

convert_int(int *val, char *str)

{

	long lval;

	if (str == NULL || !isdigit(*str))

		return (-1);

	lval = strtol(str, &str, 10);

	if (*str != '\0' || lval > INT_MAX)

		return (-2);

	*val = (int)lval;

	return (0);

}

void

lock_cmd_usage(void) {

	printf("Usage: lock r|R|w|W|u|U start len filename\n");

}

void

test_cmd_usage(void) {

	printf("Usage: test r|w|u start len filename\n");

}

int

locktest_cmd(int argc, char *argv[])

{

	int f;

	struct flock fl;

	int cmd;

	int t;

	void (*usage_fnc)(void);

	if (argc < 1)

		return -1;

	if (strcmp(argv[0], "lock") == 0) {

		usage_fnc = lock_cmd_usage;

		cmd = F_SETLK;

	} else if (strcmp(argv[0], "test") == 0) {

		usage_fnc = test_cmd_usage;

		cmd = F_GETLK;

	} else {

		return -1;

	}

	if (argc != 5 || strlen(argv[1]) != 1) {

		usage_fnc();

		return -1;

	}

	f = open(argv[4], O_RDWR | O_CREAT, 0777);

	if (f == -1) {

		printf("open failed\n");

		return -1;

	}

	switch (argv[1][0]) {

	case 'r':

	case 'w':

	case 'u': break;

	case 'R':

	case 'W':

	case 'U':

		if (cmd == F_SETLK) {

			cmd = F_SETLKW;

			break;

		}

	default: usage_fnc(); return -1;

	}

	switch (argv[1][0]) {

	case 'r':

	case 'R': fl.l_type = F_RDLCK; break;

	case 'w':

	case 'W': fl.l_type = F_WRLCK; break;

	case 'u':

	case 'U': fl.l_type = F_UNLCK; break;

	default: usage_fnc(); return -1;

	}

	fl.l_whence = SEEK_SET;

	if (convert_int(&t, argv[2]) != 0) {

		usage_fnc(); return -1;

	}

	fl.l_start = t;

	if (convert_int(&t, argv[3]) != 0) {

		usage_fnc(); return -1;

	}

	fl.l_len = t;

	if (fcntl(f, cmd, &fl) == -1) {

		printf("fcntl failed\n");

		return 0;

	}

	if (cmd == F_GETLK) {

		char *s;

		switch (fl.l_type) {

		case F_WRLCK: s = "F_WRLCK"; break;

		case F_RDLCK: s = "F_RDLCK"; break;

		case F_UNLCK: s = "F_UNLCK"; break;

		default: s = "unknown"; break;

		}

		printf("F_GETLK: %s %ld %ld\n", s, fl.l_start, fl.l_len);

	}

	return 0;

}

int

sleep_cmd(int argc, char *argv[])

{

	int sec;

	if (argc != 2 || convert_int(&sec, argv[1]) != 0)

		return -1;

	(void) sleep(sec);

	return 0;

}

void

parse_cmd(char *s)

{

	char *argv[5];

	int argc = 0;

	int r = -1;

	for (;;) {

		while (*s != '\0' && !isgraph(*s))

			*s++ = '\0';

		if (*s == '\0')

			break;

		argv[argc++] = s;

		while (*s != '\0' && isgraph(*s))

			s++;

	}

	if (argv == 0)

		return;

	if (strcmp(argv[0], "lock") == 0)

		r = locktest_cmd(argc, argv);

	else if (strcmp(argv[0], "test") == 0)

		r = locktest_cmd(argc, argv);

	else if (strcmp(argv[0], "sleep") == 0)

		r = sleep_cmd(argc, argv);

	if (r != 0)

		printf("syntax error\n");

}

int

main(int argc, char *argv[])

{

	char *line;

	size_t len;

	for (;;) {

		line = NULL;

		len = 0;

		printf("> ");

		fflush(stdout);

		if (getline(&line, &len, stdin) == -1)

			break;

		parse_cmd(line);

		free(line);

	}

	return 0;

}