Subversion Repositories planix.SVN

#include <windows.h>
#include <stdio.h>

#pragma comment(lib, "wsock32.lib")
#pragma comment(lib, "shell32.lib")

char    *argv0 = "winplumb";
char errbuf[256];
unsigned long parseip(char*, char*);
typedef unsigned long ulong;
void oserror(void);

void
hnputl(void *p, unsigned long v)
{
        unsigned char *a;

        a = p;
        a[0] = v>>24;
        a[1] = v>>16;
        a[2] = v>>8;
        a[3] = v;
}

void
hnputs(void *p, unsigned short v)
{
        unsigned char *a;

        a = p;
        a[0] = v>>8;
        a[1] = v;
}

unsigned long
nhgetl(void *p)
{
        unsigned char *a;
        a = p;
        return (a[0]<<24)|(a[1]<<16)|(a[2]<<8)|(a[3]<<0);
}

unsigned short
nhgets(void *p)
{
        unsigned char *a;
        a = p;
        return (a[0]<<8)|(a[1]<<0);
}


int
main(int argc, char **argv)
{
        char *addr, *p, *q, to[4];
        char buf[2048];
        int port, fd, nfd, one, len, n, tot;
        ulong ip;
        struct sockaddr_in sin;
        WSADATA wasdat;

        if(argc != 1 && argc != 2){
        usage:
                fprintf(stderr, "usage: winplumb [tcp!ipaddr!port]\n");
                ExitThread(1);
        }

        if(argc == 1)
                addr = "tcp!*!17890";
        else
                addr = argv[1];

        strcpy(buf, addr);
        p = strchr(buf, '!');
        if(p == 0)
                goto usage;
        q = strchr(p+1, '!');
        if(q == 0)
                goto usage;
        *p++ = 0;
        *q++ = 0;

        if(strcmp(buf, "tcp") != 0)
                goto usage;

        port = atoi(q);
        if(strcmp(p, "*") == 0)
                ip = 0;
        else
                ip = parseip(to, p);

        WSAStartup(MAKEWORD(1, 1), &wasdat);


        fd = socket(AF_INET, SOCK_STREAM, 0);
        if(fd < 0){
                oserror();
                fprintf(stderr, "socket: %s\n", errbuf);
                ExitThread(1);
        }

        one = 1;
        if(setsockopt(fd, IPPROTO_TCP, TCP_NODELAY, (char*)&one, sizeof one) != 0){
                oserror();
                fprintf(stderr, "setsockopt nodelay: %s\n", errbuf);
        }

        if(setsockopt(fd, SOL_SOCKET, SO_REUSEADDR, (char*)&one, sizeof one) != 0){
                oserror();
                fprintf(stderr, "setsockopt reuse: %s\n", errbuf);
        }
        memset(&sin, 0, sizeof sin);
        sin.sin_family = AF_INET;
        hnputs(&sin.sin_port, port);
        hnputl(&sin.sin_addr, ip);
        if(bind(fd, (struct sockaddr*)&sin, sizeof sin) < 0){
                oserror();
                fprintf(stderr, "bind: %s\n", errbuf);
                ExitThread(1);
        }

        if(listen(fd, 5) < 0){
                oserror();
                fprintf(stderr, "listen: %s\n", errbuf);
                ExitThread(1);
        }

        for(;;){
                len = sizeof sin;
                nfd = accept(fd, (struct sockaddr*)&sin, &len);
                if(nfd < 0){
                        oserror();
                        fprintf(stderr, "accept: %s\n", errbuf);
                        continue;
                }
                tot = 0;
                while(tot == 0 || buf[tot-1] != '\n'){
                        n = recv(nfd, buf+tot, sizeof buf-tot, 0);
                        if(n < 0)
                                break;
                        tot += n;
                }
                if(buf[tot-1] == '\n'){
                        buf[tot-1] = 0;
                        p = strchr(buf, ' ');
                        if(p)
                                *p++ = 0;
                        ShellExecute(0, 0, buf, p, 0, SW_SHOWNORMAL);
                }
                closesocket(nfd);
        }
}


#define CLASS(p) ((*(unsigned char*)(p))>>6)


unsigned long
parseip(char *to, char *from)
{
        int i;
        char *p;

        p = from;
        memset(to, 0, 4);
        for(i = 0; i < 4 && *p; i++){
                to[i] = strtoul(p, &p, 0);
                if(*p == '.')
                        p++;
        }
        switch(CLASS(to)){
        case 0:  /* class A - 1 byte net */
        case 1:
                if(i == 3){
                        to[3] = to[2];
                        to[2] = to[1];
                        to[1] = 0;
                } else if (i == 2){
                        to[3] = to[1];
                        to[1] = 0;
                }
                break;
        case 2: /* class B - 2 byte net */
                if(i == 3){
                        to[3] = to[2];
                        to[2] = 0;
                }
                break;
        }
        return nhgetl(to);
}

void
oserror(void)
{
        int e, r, i;
        char buf[200];

        e = GetLastError();
        
        r = FormatMessage(FORMAT_MESSAGE_FROM_SYSTEM,
                0, e, MAKELANGID(LANG_NEUTRAL, SUBLANG_DEFAULT),
                buf, sizeof(buf), 0);

        if(r == 0)
                sprintf(buf, "windows error %d", e);


        for(i = strlen(buf)-1; i>=0 && buf[i] == '\n' || buf[i] == '\r'; i--)
                buf[i] = 0;

        strcpy(errbuf, buf);
}

extern int      main(int, char*[]);
static int      args(char *argv[], int n, char *p);

int PASCAL
WinMain(HANDLE hInst, HANDLE hPrev, LPSTR arg, int nshow)
{
        int argc, n;
        char *p, **argv;

        /* conservative guess at the number of args */
        for(argc=5,p=arg; *p; p++)
                if(*p == ' ' || *p == '\t')
                        argc++;

        argv = malloc(argc*sizeof(char*));
        argc = args(argv+1, argc, arg);
        argc++;
        argv[0] = argv0;
        main(argc, argv);
        ExitThread(0);
        return 0;
}

/*
 * Break the command line into arguments
 * The rules for this are not documented but appear to be the following
 * according to the source for the microsoft C library.
 * Words are seperated by space or tab
 * Words containing a space or tab can be quoted using "
 * 2N backslashes + " ==> N backslashes and end quote
 * 2N+1 backslashes + " ==> N backslashes + literal "
 * N backslashes not followed by " ==> N backslashes
 */
static int
args(char *argv[], int n, char *p)
{
        char *p2;
        int i, j, quote, nbs;

        for(i=0; *p && i<n-1; i++) {
                while(*p == ' ' || *p == '\t')
                        p++;
                quote = 0;
                argv[i] = p2 = p;
                for(;*p; p++) {
                        if(!quote && (*p == ' ' || *p == '\t'))
                                break;
                        for(nbs=0; *p == '\\'; p++,nbs++)
                                ;
                        if(*p == '"') {
                                for(j=0; j<(nbs>>1); j++)
                                        *p2++ = '\\';
                                if(nbs&1)
                                        *p2++ = *p;
                                else
                                        quote = !quote;
                        } else {
                                for(j=0; j<nbs; j++)
                                        *p2++ = '\\';
                                *p2++ = *p;
                        }
                }
                /* move p up one to avoid pointing to null at end of p2 */
                if(*p)
                        p++;
                *p2 = 0;        
        }
        argv[i] = 0;

        return i;
}