apps/zjab/net.c

   1 /*
   2  *  ZeX/OS
   3  *  Copyright (C) 2009  Tomas 'ZeXx86' Jedrzejek (zexx86@zexos.org)
   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 3 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, see <http://www.gnu.org/licenses/>.
  17  */
  18
  19
  20 #include <fcntl.h>
  21 #include <netdb.h>
  22 #include <stdio.h>
  23 #include <string.h>
  24 #include <stdlib.h>
  25 #include <unistd.h>
  26 #include <arpa/inet.h>
  27 #include <netinet/in.h>
  28 #include <sys/socket.h>
  29 #include "net.h"
  30
  31 int sock;
  32
  33 static struct sockaddr_in serverSock;           // Remote socket
  34 static struct hostent *host;                    // Remote host
  35
  36 int net_nonblock ()
  37 {
  38         /* set socket "sock" to non-blocking */
  39         int oldFlag = fcntl (sock, F_GETFL, 0);
  40         if (fcntl (sock, F_SETFL, oldFlag | O_NONBLOCK) == -1) {
  41                 printf ("Cant set socket to nonblocking mode\n");
  42                 return 1;
  43         }
  44
  45         return 0;
  46 }
  47
  48 int net_send (char *buf, unsigned len)
  49 {
  50         return send (sock, buf, len, 0);
  51 }
  52
  53 int net_recv (char *buf, unsigned len)
  54 {
  55         return recv (sock, buf, len, 0);
  56 }
  57
  58 int net_connect (char *server, int port)
  59 {
  60         // Let's get info about remote computer
  61         if ((host = gethostbyname ((char *) server)) == NULL) {
  62                 printf ("Wrong address -> %s:%d\n", server, port);
  63                 return 0;
  64         }
  65
  66         // Create socket
  67         if ((sock = socket (AF_INET, SOCK_STREAM, IPPROTO_TCP)) == -1) {
  68                 printf ("Cant create socket\n");
  69                 return 0;
  70         }
  71
  72         // Fill structure sockaddr_in
  73         // 1) Family of protocols
  74         serverSock.sin_family = AF_INET;
  75         // 2) Number of server port
  76         serverSock.sin_port = htons (port);
  77         // 3) Setup ip address of server, where we want to connect
  78         memcpy (&(serverSock.sin_addr), host->h_addr, host->h_length);
  79
  80         // Now we are able to connect to remote server
  81         if (connect (sock, (struct sockaddr *) &serverSock, sizeof (serverSock)) == -1) {
  82                 printf ("Connection cant be estabilished -> %s:%d\n", server, port);
  83                 return -1;
  84         }
  85
  86         printf ("zjab -> connected to %s:%d server\n", server, port);
  87
  88         return 0;
  89 }
  90
  91 int net_close ()
  92 {
  93         close (sock);
  94
  95         return 0;
  96 }