Return to duplicheck.c CVS log

Up to [ELWIX - Embedded LightWeight unIX -] / embedaddon / strongswan / src / libcharon / plugins / duplicheck

Annotation of embedaddon/strongswan/src/libcharon/plugins/duplicheck/duplicheck.c, revision 1.1.1.1

1.1       misho       1: /*
                      2:  * Copyright (C) 2011 Martin Willi
                      3:  * Copyright (C) 2011 revosec AG
                      4:  *
                      5:  * This program is free software; you can redistribute it and/or modify it
                      6:  * under the terms of the GNU General Public License as published by the
                      7:  * Free Software Foundation; either version 2 of the License, or (at your
                      8:  * option) any later version.  See <http://www.fsf.org/copyleft/gpl.txt>.
                      9:  *
                     10:  * This program is distributed in the hope that it will be useful, but
                     11:  * WITHOUT ANY WARRANTY; without even the implied warranty of MERCHANTABILITY
                     12:  * or FITNESS FOR A PARTICULAR PURPOSE.  See the GNU General Public License
                     13:  * for more details.
                     14:  */
                     15: 
                     16: #include <sys/socket.h>
                     17: #include <sys/un.h>
                     18: #include <unistd.h>
                     19: #include <stdlib.h>
                     20: #include <stddef.h>
                     21: #include <stdio.h>
                     22: #include <string.h>
                     23: #include <errno.h>
                     24: #include <arpa/inet.h>
                     25: #include <netinet/in.h>
                     26: 
                     27: #include "duplicheck_msg.h"
                     28: 
                     29: /**
                     30:  * Connect to the daemon, return FD
                     31:  */
                     32: static int make_connection()
                     33: {
                     34:        union {
                     35:                struct sockaddr_un un;
                     36:                struct sockaddr_in in;
                     37:                struct sockaddr sa;
                     38:        } addr;
                     39:        int fd, len;
                     40: 
                     41:        if (getenv("TCP_PORT"))
                     42:        {
                     43:                addr.in.sin_family = AF_INET;
                     44:                addr.in.sin_addr.s_addr = htonl(INADDR_LOOPBACK);
                     45:                addr.in.sin_port = htons(atoi(getenv("TCP_PORT")));
                     46:                len = sizeof(addr.in);
                     47:        }
                     48:        else
                     49:        {
                     50:                addr.un.sun_family = AF_UNIX;
                     51:                strcpy(addr.un.sun_path, DUPLICHECK_SOCKET);
                     52: 
                     53:                len = offsetof(struct sockaddr_un, sun_path) + strlen(addr.un.sun_path);
                     54:        }
                     55:        fd = socket(addr.sa.sa_family, SOCK_STREAM, 0);
                     56:        if (fd < 0)
                     57:        {
                     58:                fprintf(stderr, "opening socket failed: %s\n", strerror(errno));
                     59:                return -1;
                     60:        }
                     61:        if (connect(fd, &addr.sa, len) < 0)
                     62:        {
                     63:                fprintf(stderr, "connecting failed: %s\n", strerror(errno));
                     64:                close(fd);
                     65:                return -1;
                     66:        }
                     67:        return fd;
                     68: }
                     69: 
                     70: int main(int argc, char *argv[])
                     71: {
                     72:        char buf[128];
                     73:        int fd, len;
                     74:        uint16_t msglen;
                     75: 
                     76:        fd = make_connection();
                     77:        if (fd < 0)
                     78:        {
                     79:                return 1;
                     80:        }
                     81:        while (1)
                     82:        {
                     83:                len = recv(fd, &msglen, sizeof(msglen), 0);
                     84:                if (len != sizeof(msglen))
                     85:                {
                     86:                        break;
                     87:                }
                     88:                msglen = ntohs(msglen);
                     89:                while (msglen)
                     90:                {
                     91:                        if (sizeof(buf) > msglen)
                     92:                        {
                     93:                                len = msglen;
                     94:                        }
                     95:                        else
                     96:                        {
                     97:                                len = sizeof(buf);
                     98:                        }
                     99:                        len = recv(fd, &buf, len, 0);
                    100:                        if (len < 0)
                    101:                        {
                    102:                                break;
                    103:                        }
                    104:                        msglen -= len;
                    105:                        printf("%.*s", len, buf);
                    106:                }
                    107:                printf("\n");
                    108:                if (len < 0)
                    109:                {
                    110:                        break;
                    111:                }
                    112:        }
                    113:        fprintf(stderr, "reading from socket failed: %s\n", strerror(errno));
                    114:        close(fd);
                    115:        return 1;
                    116: }