rnetclient.c

   1 /*
   2  *  Copyright (C) 2011  Thadeu Lima de Souza Cascardo <cascardo@holoscopio.com>
   3  *
   4  *  This program is free software; you can redistribute it and/or modify
   5  *  it under the terms of the GNU General Public License as published by
   6  *  the Free Software Foundation; either version 2 of the License, or
   7  *  (at your option) any later version.
   8  *
   9  *  This program is distributed in the hope that it will be useful,
  10  *  but WITHOUT ANY WARRANTY; without even the implied warranty of
  11  *  MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
  12  *  GNU General Public License for more details.
  13  *
  14  *  You should have received a copy of the GNU General Public License along
  15  *  with this program; if not, write to the Free Software Foundation, Inc.,
  16  *  51 Franklin Street, Fifth Floor, Boston, MA 02110-1301 USA.
  17  */
  18
  19 #include <string.h>
  20 #include <stdio.h>
  21 #include <stdlib.h>
  22 #include <unistd.h>
  23 #include <sys/socket.h>
  24 #include <netinet/in.h>
  25 #include <arpa/inet.h>
  26 #include <gnutls/gnutls.h>
  27
  28 static void * get_creds(char *certfile)
  29 {
  30         static gnutls_certificate_credentials_t cred;
  31         gnutls_certificate_allocate_credentials(&cred);
  32         gnutls_certificate_set_x509_trust_file(cred, certfile,
  33                                         GNUTLS_X509_FMT_PEM);
  34         return cred;
  35 }
  36
  37 static void session_new(gnutls_session_t *session)
  38 {
  39         static void *cred;
  40         cred = get_creds("cert.pem");
  41         gnutls_init(session, GNUTLS_CLIENT);
  42         gnutls_set_default_priority(*session);
  43         gnutls_credentials_set(*session, GNUTLS_CRD_CERTIFICATE, cred);
  44 }
  45
  46 int main(int argc, char **argv)
  47 {
  48         struct sockaddr_in saddr;
  49         int c;
  50         int r;
  51         char buffer[256];
  52         gnutls_session_t session;
  53         gnutls_global_init();
  54         session_new(&session);
  55         c = socket(PF_INET, SOCK_STREAM, 0);
  56         saddr.sin_family = AF_INET;
  57         saddr.sin_port = htons(3456);
  58         saddr.sin_addr.s_addr = htonl(INADDR_LOOPBACK);
  59         connect(c, (struct sockaddr *) &saddr, sizeof(saddr));
  60         gnutls_transport_set_ptr(session, (gnutls_transport_ptr_t) c);
  61         buffer[0] = 1;
  62         write(c, buffer, 1);
  63         r = read(c, buffer, 1);
  64         if (r != 1 && buffer[0] != 'E')
  65                 exit(1);
  66         write(c, "00000000000000", 14);
  67         r = read(c, buffer, 14);
  68         if (r != 14)
  69                 exit(1);
  70         if ((r = gnutls_handshake(session)) < 0)
  71                 fprintf(stderr, "error in handshake: %s\n",
  72                                 gnutls_strerror(r));
  73         else
  74                 fprintf(stderr, "handshake ok\n");
  75         buffer[0] = 0x40;
  76         gnutls_record_send(session, buffer, 1);
  77         while ((r = gnutls_record_recv(session, buffer, sizeof(buffer))) > 0)
  78                 write(1, buffer, r);
  79         close(c);
  80         gnutls_global_deinit();
  81         return 0;
  82 }