[doldaconnect.git] / clients / tty / dcsh.c

/*
 *  Dolda Connect - Modular multiuser Direct Connect-style client
 *  Copyright (C) 2007 Fredrik Tolf <fredrik@dolda2000.com>
 *  
 *  This program is free software; you can redistribute it and/or modify
 *  it under the terms of the GNU General Public License as published by
 *  the Free Software Foundation; either version 2 of the License, or
 *  (at your option) any later version.
 *  
 *  This program is distributed in the hope that it will be useful,
 *  but WITHOUT ANY WARRANTY; without even the implied warranty of
 *  MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
 *  GNU General Public License for more details.
 *  
 *  You should have received a copy of the GNU General Public License
 *  along with this program; if not, write to the Free Software
 *  Foundation, Inc., 59 Temple Place, Suite 330, Boston, MA  02111-1307  USA
*/

#include <stdlib.h>
#include <stdio.h>
#include <unistd.h>
#include <wchar.h>
#include <sys/poll.h>
#include <string.h>
#include <errno.h>
#include <locale.h>

#ifdef HAVE_CONFIG_H
#include <config.h>
#endif
#include <doldaconnect/uilib.h>
#include <doldaconnect/uimisc.h>
#include <doldaconnect/utils.h>

int interactive = 0;
int verbose = 0;
int dcfd;

struct cmd {
    wchar_t *name;
    int (*handler)(int argc, wchar_t **argv);
};

void wcslimit(wchar_t *wcs, int limit) {
    int i;
    
    if(wcslen(wcs) > limit) {
	for(i = limit - 3; i < limit; i++)
	    wcs[i] = L'.';
	wcs[i] = L'\0';
    }
}

void wcslimitr(wchar_t *wcs, int limit) {
    int i;
    
    if((i = wcslen(wcs)) > limit) {
	memmove(wcs + 3, wcs + (i - limit + 3), sizeof(wchar_t) * (limit - 2));
	for(i = 0; i < 3; i++)
	    wcs[i] = L'.';
    }
}

int cmd_lsnodes(int argc, wchar_t **argv)
{
    struct dc_response *resp;
    struct dc_intresp *ires;
    
    resp = dc_gettaggedrespsync(dc_queuecmd(NULL, NULL, L"lsnodes", NULL));
    if(resp->code == 200) {
	if(interactive) {
	    printf("ID    NET  USERS  S NAME\n");
	    printf("----- ---- ------ - ----------------------------------------------------------\n");
	}
	while((ires = dc_interpret(resp)) != NULL) {
	    if(interactive)
		wcslimit(ires->argv[2].val.str, 58);
	    printf("%-5i %-4ls %-6i %c %ls\n", ires->argv[0].val.num, ires->argv[1].val.str, ires->argv[3].val.num, "SHED"[ires->argv[4].val.num], ires->argv[2].val.str);
	    dc_freeires(ires);
	}
    } else if(resp->code == 201) {
    } else {
	fprintf(stderr, "dcsh: %ls\n", resp->rlines[0].argv[0]);
	return(1);
    }
    return(0);
}

int cmd_lsdl(int argc, wchar_t **argv)
{
    struct dc_response *resp;
    struct dc_intresp *ires;
    
    resp = dc_gettaggedrespsync(dc_queuecmd(NULL, NULL, L"lstrans", NULL));
    if(resp->code == 200) {
	if(interactive) {
	    printf("ID      S USER            FILE\n");
	    printf("------- - --------------- ----------------------------------------------------\n");
	}
	while((ires = dc_interpret(resp)) != NULL) {
	    if(ires->argv[1].val.num == DC_TRNSD_DOWN) {
		if(interactive) {
		    wcslimit(ires->argv[4].val.str, 15);
		    wcslimitr(ires->argv[5].val.str, 52);
		}
		printf("%-7i %c %-15ls %ls\n", ires->argv[0].val.num, "SHED"[ires->argv[2].val.num], ires->argv[4].val.str, ires->argv[5].val.str);
	    }
	    dc_freeires(ires);
	}
    } else if(resp->code == 201) {
    } else {
	fprintf(stderr, "dcsh: %ls\n", resp->rlines[0].argv[0]);
	return(1);
    }
    return(0);
}

struct cmd commands[] = {
    {L"hubs", cmd_lsnodes},
    {L"lsdl", cmd_lsdl},
    {NULL, NULL}
};

int run1(int argc, wchar_t **argv)
{
    struct cmd *c;
    
    for(c = commands; c->handler != NULL; c++) {
	if(!wcscmp(argv[0], c->name))
	    return(c->handler(argc, argv));
    }
    fprintf(stderr, "dcsh: no such command\n");
    return(1);
}

int runchar(int argc, char **argv) {
    int i, rv;
    wchar_t **wargv, *buf;
    size_t wargvsize, wargvdata;
    
    wargv = NULL;
    wargvsize = wargvdata = 0;
    for(i = 0; i < argc; i++) {
	if((buf = icmbstowcs(argv[i], NULL)) == NULL) {
	    fprintf(stderr, "dcsh: could not convert %s to wide char: %s\n", argv[i], strerror(errno));
	    for(i = 0; i < wargvdata; i++)
		free(wargv[i]);
	    if(wargv != NULL)
		free(wargv);
	    return(1);
	}
	addtobuf(wargv, buf);
    }
    addtobuf(wargv, NULL);
    rv = run1(wargvdata - 1, wargv);
    dc_freewcsarr(wargv);
    return(rv);
}

int shell(void)
{
    int ret, argc;
    wchar_t *buf, **argv;
    char *p, cmdbuf[128];
    int cmddata;
    struct pollfd pfd[2];

    if(interactive) {
	fprintf(stderr, "dcsh> ");
	fflush(stderr);
    }
    cmddata = 0;
    while(1) {
	pfd[0].fd = dcfd;
	pfd[0].events = POLLIN;
	if(dc_wantwrite())
	    pfd[0].events |= POLLOUT;
	pfd[1].fd = 0;
	pfd[1].events = POLLIN;
	pfd[1].revents = 0;
	if(poll(pfd, 2, -1) < 0) {
	    if(errno != EINTR) {
		perror("dcsh: poll");
		exit(1);
	    }
	}
	if(((pfd[0].revents & POLLIN) && dc_handleread()) || ((pfd[0].revents & POLLOUT) && dc_handlewrite()))
	    return(1);
	if(pfd[1].revents) {
	    ret = read(0, cmdbuf + cmddata, sizeof(cmdbuf) - cmddata);
	    if(ret < 0) {
		fprintf(stderr, "dcsh: stdin: %s\n", strerror(errno));
		return(1);
	    } else if(ret == 0) {
		if(interactive)
		    fprintf(stderr, "\n");
		return(0);
	    }
	    cmddata += ret;
	    while((p = memchr(cmdbuf, '\n', cmddata)) != NULL) {
		*(p++) = 0;
		if((buf = icmbstowcs(cmdbuf, NULL)) == NULL) {
		    fprintf(stderr, "dcsh: could not convert command to wide chars: %s\n", strerror(errno));
		} else {
		    argv = dc_lexsexpr(buf);
		    free(buf);
		    for(argc = 0; argv[argc] != NULL; argc++);
		    if(argc > 0)
			run1(argc, argv);
		    dc_freewcsarr(argv);
		}
		memmove(cmdbuf, p, cmddata -= (p - cmdbuf));
		if(interactive) {
		    fprintf(stderr, "dcsh> ");
		    fflush(stderr);
		}
	    }
	}
    }
}

int main(int argc, char **argv)
{
    int c, rv;
    char *server;
    char *username;
    int authless, authia;
    
    setlocale(LC_ALL, "");
    server = username = NULL;
    authless = authia = 1;
    while((c = getopt(argc, argv, "hvIAs:u:")) != -1) {
	switch(c) {
	case 's':
	    server = optarg;
	    break;
	case 'u':
	    username = optarg;
	    break;
	case 'I':
	    authia = 0;
	    break;
	case 'A':
	    authless = 0;
	    break;
	case 'v':
	    verbose++;
	    break;
	default:
	    fprintf(stderr, "usage: dcsh [-AIhv] [-s SERVER] [-u USERNAME] [COMMAND [ARGS...]]\n");
	    exit((c == 'h')?0:1);
	}
    }
    dc_init();
    
    if(verbose)
	fprintf(stderr, "connecting...\n");
    if((dcfd = dc_connectsync2(server, 1)) < 0) {
	perror("dcsh: could not connect to server");
	exit(1);
    }
    if(verbose)
	fprintf(stderr, "authenticating...\n");
    if(dc_login(username, authless, authia?dc_convtty:dc_convnone, NULL) != DC_LOGIN_ERR_SUCCESS) {
	fprintf(stderr, "dcsh: authentication unsuccessful\n");
	exit(1);
    }
    if(verbose)
	fprintf(stderr, "done\n");
    if(optind < argc) {
	interactive = isatty(1);
	rv = runchar(argc - optind, argv + optind);
    } else {
	interactive = isatty(0);
	rv = shell();
    }
    return(rv);
}
Commit	Line	Data
a24f31ec	1	/*
a24f31ec	2	* Dolda Connect - Modular multiuser Direct Connect-style client
302a2600	3	* Copyright (C) 2007 Fredrik Tolf <fredrik@dolda2000.com>
a24f31ec	4	*
	5	* This program is free software; you can redistribute it and/or modify
	6	* it under the terms of the GNU General Public License as published by
	7	* the Free Software Foundation; either version 2 of the License, or
	8	* (at your option) any later version.
	9	*
	10	* This program is distributed in the hope that it will be useful,
	11	* but WITHOUT ANY WARRANTY; without even the implied warranty of
	12	* MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the
	13	* GNU General Public License for more details.
	14	*
	15	* You should have received a copy of the GNU General Public License
	16	* along with this program; if not, write to the Free Software
	17	* Foundation, Inc., 59 Temple Place, Suite 330, Boston, MA 02111-1307 USA
	18	*/
	19
	20	#include <stdlib.h>
	21	#include <stdio.h>
	22	#include <unistd.h>
	23	#include <wchar.h>
	24	#include <sys/poll.h>
c8817ca2	25	#include <string.h>
15ee5bfb	26	#include <errno.h>
c8817ca2	27	#include <locale.h>
a24f31ec	28
	29	#ifdef HAVE_CONFIG_H
	30	#include <config.h>
	31	#endif
	32	#include <doldaconnect/uilib.h>
	33	#include <doldaconnect/uimisc.h>
	34	#include <doldaconnect/utils.h>
	35
fa51e153	36	int interactive = 0;
a24f31ec	37	int verbose = 0;
	38	int dcfd;
	39
c8817ca2 FT	40	struct cmd {
	41	wchar_t *name;
	42	int (handler)(int argc, wchar_t *argv);
	43	};
	44
fa51e153 FT	45	void wcslimit(wchar_t *wcs, int limit) {
	46	int i;
	47
	48	if(wcslen(wcs) > limit) {
	49	for(i = limit - 3; i < limit; i++)
	50	wcs[i] = L'.';
	51	wcs[i] = L'\0';
	52	}
	53	}
	54
b6e4d01d FT	55	void wcslimitr(wchar_t *wcs, int limit) {
	56	int i;
	57
	58	if((i = wcslen(wcs)) > limit) {
	59	memmove(wcs + 3, wcs + (i - limit + 3), sizeof(wchar_t) * (limit - 2));
	60	for(i = 0; i < 3; i++)
	61	wcs[i] = L'.';
	62	}
	63	}
	64
c8817ca2 FT	65	int cmd_lsnodes(int argc, wchar_t **argv)
c8817ca2 FT	66	{
c8817ca2 FT	67	struct dc_response *resp;
	68	struct dc_intresp *ires;
	69
	70	resp = dc_gettaggedrespsync(dc_queuecmd(NULL, NULL, L"lsnodes", NULL));
	71	if(resp->code == 200) {
fa51e153	72	if(interactive) {
b6e4d01d	73	printf("ID NET USERS S NAME\n");
fa51e153 FT	74	printf("----- ---- ------ - ----------------------------------------------------------\n");
fa51e153 FT	75	}
c8817ca2	76	while((ires = dc_interpret(resp)) != NULL) {
fa51e153 FT	77	if(interactive)
fa51e153 FT	78	wcslimit(ires->argv[2].val.str, 58);
b6e4d01d FT	79	printf("%-5i %-4ls %-6i %c %ls\n", ires->argv[0].val.num, ires->argv[1].val.str, ires->argv[3].val.num, "SHED"[ires->argv[4].val.num], ires->argv[2].val.str);
	80	dc_freeires(ires);
	81	}
	82	} else if(resp->code == 201) {
	83	} else {
	84	fprintf(stderr, "dcsh: %ls\n", resp->rlines[0].argv[0]);
	85	return(1);
	86	}
	87	return(0);
	88	}
	89
	90	int cmd_lsdl(int argc, wchar_t **argv)
	91	{
	92	struct dc_response *resp;
	93	struct dc_intresp *ires;
	94
	95	resp = dc_gettaggedrespsync(dc_queuecmd(NULL, NULL, L"lstrans", NULL));
	96	if(resp->code == 200) {
	97	if(interactive) {
	98	printf("ID S USER FILE\n");
	99	printf("------- - --------------- ----------------------------------------------------\n");
	100	}
	101	while((ires = dc_interpret(resp)) != NULL) {
	102	if(ires->argv[1].val.num == DC_TRNSD_DOWN) {
	103	if(interactive) {
	104	wcslimit(ires->argv[4].val.str, 15);
	105	wcslimitr(ires->argv[5].val.str, 52);
	106	}
	107	printf("%-7i %c %-15ls %ls\n", ires->argv[0].val.num, "SHED"[ires->argv[2].val.num], ires->argv[4].val.str, ires->argv[5].val.str);
	108	}
c8817ca2 FT	109	dc_freeires(ires);
	110	}
	111	} else if(resp->code == 201) {
	112	} else {
	113	fprintf(stderr, "dcsh: %ls\n", resp->rlines[0].argv[0]);
b6e4d01d	114	return(1);
c8817ca2 FT	115	}
	116	return(0);
	117	}
	118
	119	struct cmd commands[] = {
	120	{L"hubs", cmd_lsnodes},
b6e4d01d	121	{L"lsdl", cmd_lsdl},
c8817ca2 FT	122	{NULL, NULL}
	123	};
	124
	125	int run1(int argc, wchar_t **argv)
	126	{
	127	struct cmd *c;
	128
	129	for(c = commands; c->handler != NULL; c++) {
	130	if(!wcscmp(argv[0], c->name))
	131	return(c->handler(argc, argv));
	132	}
	133	fprintf(stderr, "dcsh: no such command\n");
	134	return(1);
	135	}
	136
	137	int runchar(int argc, char **argv) {
	138	int i, rv;
	139	wchar_t *wargv, buf;
	140	size_t wargvsize, wargvdata;
	141
	142	wargv = NULL;
	143	wargvsize = wargvdata = 0;
	144	for(i = 0; i < argc; i++) {
	145	if((buf = icmbstowcs(argv[i], NULL)) == NULL) {
	146	fprintf(stderr, "dcsh: could not convert %s to wide char: %s\n", argv[i], strerror(errno));
	147	for(i = 0; i < wargvdata; i++)
	148	free(wargv[i]);
	149	if(wargv != NULL)
	150	free(wargv);
	151	return(1);
	152	}
	153	addtobuf(wargv, buf);
	154	}
	155	addtobuf(wargv, NULL);
	156	rv = run1(wargvdata - 1, wargv);
	157	dc_freewcsarr(wargv);
	158	return(rv);
	159	}
	160
	161	int shell(void)
	162	{
	163	int ret, argc;
	164	wchar_t buf, *argv;
	165	char *p, cmdbuf[128];
	166	int cmddata;
	167	struct pollfd pfd[2];
	168
fa51e153 FT	169	if(interactive) {
	170	fprintf(stderr, "dcsh> ");
	171	fflush(stderr);
	172	}
c8817ca2 FT	173	cmddata = 0;
	174	while(1) {
	175	pfd[0].fd = dcfd;
	176	pfd[0].events = POLLIN;
	177	if(dc_wantwrite())
	178	pfd[0].events \|= POLLOUT;
	179	pfd[1].fd = 0;
	180	pfd[1].events = POLLIN;
	181	pfd[1].revents = 0;
	182	if(poll(pfd, 2, -1) < 0) {
	183	if(errno != EINTR) {
	184	perror("dcsh: poll");
	185	exit(1);
	186	}
	187	}
	188	if(((pfd[0].revents & POLLIN) && dc_handleread()) \|\| ((pfd[0].revents & POLLOUT) && dc_handlewrite()))
	189	return(1);
	190	if(pfd[1].revents) {
	191	ret = read(0, cmdbuf + cmddata, sizeof(cmdbuf) - cmddata);
	192	if(ret < 0) {
	193	fprintf(stderr, "dcsh: stdin: %s\n", strerror(errno));
	194	return(1);
	195	} else if(ret == 0) {
fa51e153 FT	196	if(interactive)
fa51e153 FT	197	fprintf(stderr, "\n");
c8817ca2 FT	198	return(0);
	199	}
	200	cmddata += ret;
	201	while((p = memchr(cmdbuf, '\n', cmddata)) != NULL) {
	202	*(p++) = 0;
	203	if((buf = icmbstowcs(cmdbuf, NULL)) == NULL) {
	204	fprintf(stderr, "dcsh: could not convert command to wide chars: %s\n", strerror(errno));
	205	} else {
	206	argv = dc_lexsexpr(buf);
	207	free(buf);
	208	for(argc = 0; argv[argc] != NULL; argc++);
	209	if(argc > 0)
	210	run1(argc, argv);
	211	dc_freewcsarr(argv);
	212	}
	213	memmove(cmdbuf, p, cmddata -= (p - cmdbuf));
fa51e153 FT	214	if(interactive) {
	215	fprintf(stderr, "dcsh> ");
	216	fflush(stderr);
	217	}
c8817ca2 FT	218	}
	219	}
	220	}
	221	}
	222
a24f31ec	223	int main(int argc, char **argv)
a24f31ec	224	{
c8817ca2	225	int c, rv;
a24f31ec	226	char *server;
	227	char *username;
	228	int authless, authia;
a24f31ec	229
c8817ca2	230	setlocale(LC_ALL, "");
a24f31ec	231	server = username = NULL;
	232	authless = authia = 1;
	233	while((c = getopt(argc, argv, "hvIAs:u:")) != -1) {
	234	switch(c) {
	235	case 's':
	236	server = optarg;
	237	break;
	238	case 'u':
	239	username = optarg;
	240	break;
	241	case 'I':
	242	authia = 0;
	243	break;
	244	case 'A':
	245	authless = 0;
	246	break;
	247	case 'v':
	248	verbose++;
	249	break;
	250	default:
c8817ca2	251	fprintf(stderr, "usage: dcsh [-AIhv] [-s SERVER] [-u USERNAME] [COMMAND [ARGS...]]\n");
a24f31ec	252	exit((c == 'h')?0:1);
	253	}
	254	}
	255	dc_init();
	256
	257	if(verbose)
	258	fprintf(stderr, "connecting...\n");
	259	if((dcfd = dc_connectsync2(server, 1)) < 0) {
	260	perror("dcsh: could not connect to server");
	261	exit(1);
	262	}
	263	if(verbose)
	264	fprintf(stderr, "authenticating...\n");
	265	if(dc_login(username, authless, authia?dc_convtty:dc_convnone, NULL) != DC_LOGIN_ERR_SUCCESS) {
	266	fprintf(stderr, "dcsh: authentication unsuccessful\n");
	267	exit(1);
	268	}
	269	if(verbose)
	270	fprintf(stderr, "done\n");
fa51e153 FT	271	if(optind < argc) {
fa51e153 FT	272	interactive = isatty(1);
c8817ca2	273	rv = runchar(argc - optind, argv + optind);
fa51e153 FT	274	} else {
fa51e153 FT	275	interactive = isatty(0);
c8817ca2	276	rv = shell();
fa51e153	277	}
c8817ca2	278	return(rv);
a24f31ec	279	}