]> rtime.felk.cvut.cz Git - can-eth-gw.git/blobdiff - utils/cegw/cegw.c
Sending socket fd over netlink
[can-eth-gw.git] / utils / cegw / cegw.c
index 275d6592ff91864c9707ce0acf1787efa68b6c29..fc91a9ce48a9672b38f2001ba9be5bf7eefe205c 100644 (file)
 #include <arpa/inet.h>
 #include <linux/can.h>
 #include <linux/types.h>
+#include <unistd.h>
 #include "canethgw.h"
 
 /**
- * ToDo
- * [ ] refactor error messages
+ * ToDo:
+ * [ ] print usage, on -h and plain execution
  * [ ] start/stop listening
- * [ ] remove routing job
- * [ ] recv netlink reponse
- * 
+ * [ ] split to files
  */
 
-#define CMD_STOP_LISTEN    1
-#define CMD_START_LISTEN   2
-#define CMD_ADD_RULE       4
+#define CEGW_CMD_ADD     1
+#define CEGW_CMD_LIST    2
+#define CEGW_CMD_FLUSH   4
+#define CEGW_CMD_LISTEN  8
 
-enum {
+enum
+{
        IF_UNDEF,
        IF_CAN,
        IF_ETH_UDP
 };
 
+struct cegw_data
+{
+       int content;
+       int src_if, dst_if;
+       int can_ifidx;
+       struct in_addr eth_addr;
+       unsigned short eth_port;
+       struct in_addr eth_listen_addr;
+       unsigned short eth_listen_port;
+};
+
+struct cegw_nlmsg
+{
+       struct nlmsghdr nh;
+       struct rtmsg rt;
+       char buf[768]; /* enough? */
+};
+
+struct list_item
+{
+       int type;
+       int can;
+       struct in_addr ip;
+       unsigned short port;
+};
+
+unsigned int cegw_errno = 0;
+
+enum
+{
+       CEGW_ERR_UNKNOWN,
+       CEGW_ERR_IF_UNSPEC,
+       CEGW_ERR_IF_SAME,
+       CEGW_ERR_IF_TYPE,
+       CEGW_ERR_IF_CAN,
+       CEGW_ERR_IF_ETH,
+       CEGW_ERR_COLON,
+       CEGW_ERR_ATON,
+       CEGW_ERR_PORT
+};
+
+char* cegw_errlist[] =
+{
+       [ CEGW_ERR_UNKNOWN   ] = "",
+       [ CEGW_ERR_IF_UNSPEC ] = "source or destination not specified",
+       [ CEGW_ERR_IF_SAME   ] = "source and destination have same interface type",
+       [ CEGW_ERR_IF_TYPE   ] = "unknown interface type",
+       [ CEGW_ERR_IF_CAN    ] = "invalid can interface",
+       [ CEGW_ERR_IF_ETH    ] = "invalid eth interface",
+       [ CEGW_ERR_COLON     ] = "expected ':' (<ip>:<port>)",
+       [ CEGW_ERR_ATON      ] = "ip address mismatch",
+       [ CEGW_ERR_PORT      ] = "port number"
+};
+
+static void perr( char* s )
+{
+       if( s )
+       {
+               if( cegw_errno == 0 )
+               {
+                       fprintf( stderr, "error: %s\n", s );
+
+               } else
+               {
+                       fprintf( stderr, "error: %s, %s\n", s,
+                                cegw_errlist[ cegw_errno ] );
+               }
+               return;
+       }
+
+       fprintf( stderr, "error: %s\n", cegw_errlist[ cegw_errno ] );
+}
 
 /**
- * parses @in for eth address, valid input is 
- * e.g. udp@127.0.0.1:10502 or can@vcan0
+ * read_addrport - parses @in for eth address. 
+ * Valid input is e.g. udp@127.0.0.1:10502 or can@vcan0.
  *
+ * @param[in]  in   string to search in
  * @param[out] addr ip address
  * @param[out] port transport layer port
- * @return 0 on success
+ * @return 0 on success, -1 otherwise
  */
-int read_addr_port( char* in, struct in_addr* addr, unsigned short* port )
+int read_addrport( char* in, struct in_addr* addr, unsigned short* port )
 {
        char* delim = NULL;
        char addrstr[16];
        int addrlen;
-       
+
        if( (delim = strchr( in, ':' )) == NULL )
        {
-               fprintf( stderr, "error: ':'\n" );
+               cegw_errno = CEGW_ERR_COLON;
                return -1;
        }
 
@@ -60,222 +134,333 @@ int read_addr_port( char* in, struct in_addr* addr, unsigned short* port )
        addrstr[addrlen] = '\0';
        if( inet_aton( addrstr, addr ) == 0 )
        {
-               fprintf( stderr, "error: aton\n" );
+               cegw_errno = CEGW_ERR_ATON;
                return -1;
        }
 
        /* get port */
-       if( sscanf( delim, ":%hu", port ) != 1 ) /* todo: handle overflow */
+       if( sscanf( delim, ":%hu", port ) != 1 ) /* ToDo: handle overflow */
        {
-               fprintf( stderr, "error: port\n" );
+               cegw_errno = CEGW_ERR_PORT;
                return -1;
        }
 
        return 0;
 }
 
+/**
+ * read_iftype - reads @in for iftype
+ * Iftype type is e.g. "can@" or "udp@".
+ *
+ * @param[in] in string to search in
+ * @param[out] iftype iftype detected
+ * @return pointer to @in after iftype on success, NULL otherwise
+ */
 char* read_iftype( char* in, int* iftype )
 {
        char* ret = in+4;
-       
+
        if( strncmp( "udp@", optarg, 4 ) == 0 )
        {
                *iftype = IF_ETH_UDP;
                return ret;
        }
-       else if( strncmp( "tcp@", optarg, 4 ) == 0 )
+       /*
+       if( strncmp( "tcp@", optarg, 4 ) == 0 )
        {
                return NULL;
        }
-       else if( strncmp( "can@", optarg, 4 ) == 0 )
+       */
+       if( strncmp( "can@", optarg, 4 ) == 0 )
        {
                *iftype = IF_CAN;
                return ret;
        }
-       
+
+       cegw_errno = CEGW_ERR_IF_TYPE;
        return NULL;
 }
 
+/* ToDo: move to common */
+
+/**
+ * read_if - reads interface from @in
+ * Function analyzes @in for interface specification in format
+ * <if>@<ip>:<port>, where <if> is can or udp, <ip> is address in dotted
+ * format
+ *
+ * @param[in]  in string to search in
+ * @param[out] iftype interface type (IF_CAN or IF_ETH_UDP)
+ * @param[out] d ip and port is stored to @d
+ */
+int read_if( char* in, int* iftype, struct cegw_data* d )
+{
+       char* optstr = NULL;
+
+       if( (optstr = read_iftype( in, iftype )) == NULL )
+       {
+               return -1;
+       }
+
+       switch( *iftype )
+       {
+               case IF_CAN:
+                       d->can_ifidx = if_nametoindex( optstr );
+                       if( d->can_ifidx == 0 )
+                       {
+                               cegw_errno = CEGW_ERR_IF_CAN;
+                               return -1;
+                       }
+                       break;
+               case IF_ETH_UDP:
+                       if( read_addrport( optstr, &d->eth_addr, &d->eth_port ) != 0 )
+                       {
+                               return -1;
+                       }
+                       break;
+               default:
+                       return -1;
+                       break;
+       }
+
+       return 0;
+}
+
+inline static int cegw_add( struct cegw_nlmsg* req, struct cegw_data* d )
+{
+       int gwtype = 0;
+
+       req->nh.nlmsg_type  = RTM_NEWROUTE;
+       if( (d->src_if == 0 || d->dst_if == 0) )
+       {
+               cegw_errno = CEGW_ERR_IF_UNSPEC;
+               return -cegw_errno;
+       }
+
+       if( d->src_if == d->dst_if )
+       {
+               cegw_errno = CEGW_ERR_IF_SAME;
+               return -cegw_errno;
+       }
+
+       gwtype = (d->src_if == IF_CAN) ? CEGW_RULE_CAN_ETH : CEGW_RULE_ETH_CAN;
+       addattr_l( &req->nh, sizeof(*req), CEGW_CAN_IFINDEX, &d->can_ifidx, sizeof(d->can_ifidx) );
+       addattr_l( &req->nh, sizeof(*req), CEGW_ETH_IP, &d->eth_addr, sizeof(d->eth_addr) );
+       addattr_l( &req->nh, sizeof(*req), CEGW_ETH_PORT, &d->eth_port, sizeof(d->eth_port) );
+       addattr32( &req->nh, sizeof(*req), CEGW_CMD_INFO, gwtype );
+
+       return 0;
+}
+
+inline static int cegw_listen( struct cegw_nlmsg* req, struct cegw_data* d )
+{
+       req->nh.nlmsg_type = RTM_NEWROUTE;
+       addattr32( &req->nh, sizeof(*req), CEGW_CMD_INFO, CEGW_LISTEN );
+       addattr_l( &req->nh, sizeof(*req), CEGW_ETH_IP, &d->eth_listen_addr, sizeof(d->eth_listen_addr) );
+       addattr_l( &req->nh, sizeof(*req), CEGW_ETH_PORT, &d->eth_listen_port, sizeof(d->eth_listen_port) );
+
+       return 0;
+}
+
+inline static int cegw_list( struct cegw_nlmsg* req, struct cegw_data* d )
+{
+       req->nh.nlmsg_type = RTM_GETROUTE;
+       req->nh.nlmsg_flags = NLM_F_REQUEST | NLM_F_DUMP;
+
+       return 0;
+}
+
+inline static int cegw_flush( struct cegw_nlmsg* req, struct cegw_data* d )
+{
+       addattr32( &req->nh, sizeof(*req), CEGW_CMD_INFO, CEGW_FLUSH );
+       req->nh.nlmsg_type  = RTM_DELROUTE;
+       return 0;
+}
+
+void print_list_item( struct list_item* li )
+{
+       char ifname[IF_NAMESIZE];
+       const char src_width = 21;
+       char* dotaddr;
+       int tmp;
+
+       if( if_indextoname( li->can, ifname ) == NULL )
+       {
+               strncpy( ifname, "unknown", IF_NAMESIZE );
+       }
+
+       /* ToDo listening at */
+       switch( li->type )
+       {
+               case CEGW_RULE_CAN_ETH:
+                       printf( "can@%-*s -> udp@%s:%hu\n", src_width, ifname, \
+                                inet_ntoa(li->ip), li->port );
+                       break;
+               case CEGW_RULE_ETH_CAN:
+                       dotaddr = inet_ntoa(li->ip);
+                       tmp = src_width - strlen(dotaddr) - 1;
+                       printf( "udp@%s:%-*hu -> can@%s\n", inet_ntoa(li->ip), \
+                                tmp, li->port, ifname );
+                       break;
+       }
+}
+
 int main( int argc, char* argv[] )
 {
        int s;
-       int src_if = 0, dst_if = 0;
-       int can_ifidx = 0;
        int tmp = 0;
        int cmd = 0;
-       struct in_addr eth_addr;
-       unsigned short eth_port;
-       struct in_addr eth_listen_addr;
-       unsigned short eth_listen_port;
        char* optstr;
        char opt;
        struct sockaddr_nl nladdr;
        int err = 0;
+       struct cegw_nlmsg req;
+       struct cegw_data d;
+       char rxbuf[8192]; /* ToDo: /linux/netlink.h? */
+       int rsize = 0;
+       struct nlmsghdr* nlh;
+       struct nlmsgerr* rte;
+       struct rtattr* rta;
+       int len;
+       struct list_item li;
+       int gw_can_sock = 0;
+       int gw_eth_sock = 0;
+       struct sockaddr_in gw_eth_addr;
+       struct sockaddr_can gw_can_addr;
+       struct in_addr addr;
 
-       struct option long_opt[] =
-       {
-               { "start-listen", 1, NULL, 'l' },
-               { "stop-listen" , 1, NULL, 'k' },
-               { 0, 0, 0, 0 }
-       };
-       
-       struct {
-               struct nlmsghdr nh;
-               struct rtcanmsg rtcan;
-               char buf[600]; /* enough? */
-       } req;
-       
-       while( 1 )
+       memset( &d, 0, sizeof(d) );
+
+       /* create sockets for gateway  */
+       gw_eth_sock = socket( PF_INET, SOCK_DGRAM, IPPROTO_UDP );
+       gw_can_sock = socket( PF_CAN, SOCK_RAW, CAN_RAW );
+
+       if( gw_eth_sock == -1 )
+               fprintf( stderr, "error: udp socket for gw creation failed\n" );
+       if( gw_can_sock == -1 )
+               fprintf( stderr, "error: can socket for gw creation failed\n" );
+
+       gw_eth_addr.sin_family = AF_INET;
+       gw_eth_addr.sin_port = htons( 10501 );
+       gw_eth_addr.sin_addr.s_addr = INADDR_ANY;
+
+       gw_can_addr.can_family = AF_CAN;
+       gw_can_addr.can_ifindex = 0;
+
+       if( bind( gw_eth_sock, (struct sockaddr*)&gw_eth_addr, sizeof(gw_eth_addr) ) != 0 )
+               fprintf( stderr, "error: eth binding\n" );
+       if( bind( gw_can_sock, (struct sockaddr*)&gw_can_addr, sizeof(gw_can_addr) ) )
+               fprintf( stderr, "error: can binding\n" );
+
+       printf( "sockets created (%i,%i)\n", gw_can_sock, gw_eth_sock );
+
+       /* prepare netlink message */
+       req.nh.nlmsg_len   = NLMSG_LENGTH( sizeof(struct rtmsg) );
+       req.nh.nlmsg_type  = RTM_NEWROUTE;
+       req.nh.nlmsg_flags = NLM_F_REQUEST | NLM_F_ACK;
+       req.nh.nlmsg_seq   = 0;
+       req.nh.nlmsg_pid   = 0; /* ? */
+
+       memset( &req.rt, 0, sizeof(req.rt) );
+       req.rt.rtm_family = AF_CAN;
+
+       addattr32( &req.nh, sizeof(req), CEGW_CAN_SOCK, gw_can_sock );
+       addattr32( &req.nh, sizeof(req), CEGW_ETH_SOCK, gw_eth_sock );
+       addattr32( &req.nh, sizeof(req), CEGW_CAN_IFINDEX, if_nametoindex("vcan0") );
+
+       /* send over netlink socket */
+       s = socket( PF_NETLINK, SOCK_RAW, NETLINK_ROUTE ); /* chck */
+
+       memset( &nladdr, 0, sizeof(nladdr) );
+       nladdr.nl_family = AF_NETLINK;
+       nladdr.nl_pad = 0;
+       nladdr.nl_pid = 0;
+       nladdr.nl_groups = 0;
+
+       err = sendto( s, &req, req.nh.nlmsg_len, 0,
+                     (struct sockaddr*)&nladdr, sizeof(nladdr) );
+       if( err < 0 )
        {
-               opt = getopt_long( argc, argv, "As:d:", long_opt, NULL );
-               if( opt == -1 )
-                       break;
+               perror( "netlink sendto" );
+               return -1;
+       }
 
-               switch( opt )
-               {
-                       case 's':
-                               cmd |= CMD_ADD_RULE;
-                               if( (optstr = read_iftype( optarg, &src_if )) == NULL )
-                               {
-                                       fprintf( stderr, "error: bad input format\n" );
-                                       goto syntax_error;
-                                       break;
-                               }
-                               
-                               switch( src_if )
-                               {
-                                       case IF_CAN:
-                                               can_ifidx = if_nametoindex( optstr );
-                                               break;
-                                       case IF_ETH_UDP:
-                                               read_addr_port( optstr, &eth_addr, &eth_port );
-                                               break;
-                                       default:
-                                               fprintf( stderr, "error: unrecognized interface" );
-                                               goto syntax_error;
-                                               break;
-                               }
-                               break;
-                       case 'd':
-                               cmd |= CMD_ADD_RULE;
-                               if( (optstr = read_iftype( optarg, &dst_if )) == NULL )
-                               {
-                                       fprintf( stderr, "error: bad input format\n" );
-                                       goto syntax_error;
-                                       break;
-                               }
-                               
-                               switch( dst_if )
-                               {
-                                       case IF_CAN:
-                                               can_ifidx = if_nametoindex( optstr ); /*chk*/
-                                               break;
-                                       case IF_ETH_UDP:
-                                               read_addr_port( optstr, &eth_addr, &eth_port ); /*chk*/
-                                               break;
-                                       default:
-                                               fprintf( stderr, "error: unrecognized interface" );
-                                               break;
-                               }
-                               break;
-                       case 'l':
-                               cmd |= CMD_START_LISTEN;
-                               if( (optstr = read_iftype( optarg, &tmp )) == NULL )
-                               {
-                                       fprintf( stderr, "error: -l bad input format\n" );
-                                       goto syntax_error;
-                                       break;
-                               }
-                               
-                               if( tmp != IF_ETH_UDP )
-                               {
-                                       fprintf( stderr, "error: -l bad input format\n" );
-                                       goto syntax_error;
-                                       break;
-                               }
-                                       
-                               read_addr_port( optstr, &eth_listen_addr, &eth_listen_port ); /*chk*/
-                       case 'k':
-                               cmd |= CMD_STOP_LISTEN;
-                               break;
-                       case '?':
-                               break;
-                       default:
-                               fprintf( stderr, "error: unknown option\n" );
-                       break;
-               }                               
+       /* recv */
+       rsize = recv( s, &rxbuf, sizeof(rxbuf), 0 );
+       if( rsize < 0 )
+       {
+               perr( "recv" );
+               return -1;
        }
+       nlh = (struct nlmsghdr*)rxbuf;
+
+       /* REMOVE */
+       sleep( 2 );
+       //printf( "closing sockets\n" );
+       //shutdown( gw_eth_sock, SHUT_RDWR );
+       shutdown( gw_can_sock, SHUT_RDWR );
+       //close( gw_can_sock );
+       //close( gw_eth_sock );
 
-       /* 2. do check on arguments */
-       if( cmd & CMD_ADD_RULE )
+       if( nlh->nlmsg_type == NLMSG_ERROR )
        {
-               if( (src_if == 0 || dst_if == 0) || (src_if == dst_if) )
+               rte = (struct nlmsgerr*)NLMSG_DATA( nlh );
+               err = rte->error;
+
+               if( err == 0  )
                {
-                       fprintf( stderr, "error: source or destination not specified\n" );
-                       return -1;
-               }
-       
-               if( src_if == dst_if )
+                       printf( "%s\n", strerror(abs(err)) );
+                       return 0;
+               } else
                {
-                       fprintf( stderr, "error: source and destination same interface type\n" );
+                       printf( "netlink error: %s\n", strerror(abs(err)) );
+                       return -1;
                }
        }
-       
-       /* 3. prepare netlink message */
-       req.nh.nlmsg_len   = NLMSG_LENGTH( sizeof(struct rtcanmsg) );
-       req.nh.nlmsg_type  = RTM_NEWROUTE;
-       req.nh.nlmsg_flags = NLM_F_REQUEST | NLM_F_ACK;
-       req.nh.nlmsg_seq   = 0;
-       req.nh.nlmsg_pid   = 0; /* ? */
-       
-       req.rtcan.can_family = AF_CAN;
-       req.rtcan.flags = 0;
 
-       if( cmd & CMD_ADD_RULE )
+       if( cmd & CEGW_CMD_LIST )
        {
-               req.rtcan.gwtype = (src_if == IF_CAN) ? CGW_TYPE_CAN_ETH_UDP : CGW_TYPE_ETH_CAN_UDP;
-               addattr_l( &req.nh, sizeof(req), CGW_CAN_IF, &can_ifidx, sizeof(can_ifidx) );
-               switch( req.rtcan.gwtype )
+               /* ToDo recv while */
+               printf( "%10ssource%20sdestination\n", "", "" );
+               while( 1 )
                {
-                       case CGW_TYPE_CAN_ETH_UDP:
-                               addattr_l( &req.nh, sizeof(req), CGW_ETH_IP, &eth_addr, sizeof(eth_addr) );
-                               addattr_l( &req.nh, sizeof(req), CGW_ETH_PORT, &eth_port, sizeof(eth_port) );
-                               break;
-                       case CGW_TYPE_ETH_CAN_UDP:
+                       if( !NLMSG_OK( nlh, rsize ) )
+                       {
                                break;
-                       default:
+                       }
+                       if( nlh->nlmsg_type == NLMSG_DONE )
+                       {
                                break;
+                       }
+                       /* ToDo: NLMSG_ERR */
+                       rta = NLMSG_DATA( nlh );
+                       len = NLMSG_PAYLOAD( nlh, 0 );
+                       for( ;RTA_OK(rta, len); rta = RTA_NEXT(rta,len) )
+                       {
+                               switch( rta->rta_type )
+                               {
+                                       case CEGW_TYPE:
+                                               li.type = *(int*)RTA_DATA(rta);
+                                               break;
+                                       case CEGW_CAN_IFINDEX:
+                                               li.can = *(int*)RTA_DATA(rta);
+                                               break;
+                                       case CEGW_ETH_IP:
+                                               li.ip = *(struct in_addr*)RTA_DATA(rta);
+                                               break;
+                                       case CEGW_ETH_PORT:
+                                               li.port = *(unsigned short*)RTA_DATA(rta);
+                                               break;
+                                       /* case CGW_ETH_PROTO */
+                               }
+                       }
+
+                       print_list_item( &li );
+
+                       nlh = NLMSG_NEXT( nlh, rsize );
                }
        }
-       
-       if( cmd & CMD_START_LISTEN )
-       {
-               req.rtcan.gwtype = CGW_TYPE_CONFIG;
-               addattr_l( &req.nh, sizeof(req), CGW_LISTEN_IP, &eth_listen_addr, sizeof(eth_listen_addr) );
-               addattr_l( &req.nh, sizeof(req), CGW_LISTEN_PORT, &eth_listen_port, sizeof(eth_listen_port) );
-               printf( "start listen: %x, %hu\n", eth_listen_addr, eth_listen_port );
-       }
-       
-       /* 4. send over netlink socket */
-       s = socket(PF_NETLINK, SOCK_RAW, NETLINK_ROUTE); /* chck */
-       
-       memset(&nladdr, 0, sizeof(nladdr));
-       nladdr.nl_family = AF_NETLINK;
-       nladdr.nl_pid    = 0;
-       nladdr.nl_groups = 0;
 
-       err = sendto( s, &req, req.nh.nlmsg_len, 0, 
-                     (struct sockaddr*)&nladdr, sizeof(nladdr));
-       if (err < 0)
-       {
-               perror( "netlink sendto" );
-               return err;
-       }
-       
-       /* ack? */
-       
        return 0;
-syntax_error:
-       return -1;
 }