WINGs/Examples/connect.c

   1 /*
   2  *  WINGs connect.c: example how to create a network client using WMConnection
   3  *
   4  *  Copyright (c) 1999-2002 Dan Pascu
   5  *
   6  */
   7
   8
   9 #include <stdio.h>
  10 #include <unistd.h>
  11 #include <string.h>
  12
  13 #include <WINGs/WINGs.h>
  14
  15
  16
  17 static int initialized = 0;
  18
  19
  20
  21 static void didReceiveInput(ConnectionDelegate *self, WMConnection *cPtr);
  22
  23 static void connectionDidDie(ConnectionDelegate *self, WMConnection *cPtr);
  24
  25 static void didInitialize(ConnectionDelegate *self, WMConnection *cPtr);
  26
  27
  28
  29 static ConnectionDelegate socketDelegate = {
  30     NULL,              /* data */
  31     NULL,              /* canResumeSending */
  32     NULL,              /* didCatchException */
  33     connectionDidDie,  /* didDie */
  34     didInitialize,     /* didInitialize */
  35     didReceiveInput,   /* didReceiveInput */
  36     NULL               /* didTimeout */
  37 };
  38
  39
  40
  41 void
  42 wAbort(Bool foo)
  43 {
  44     exit(1);
  45 }
  46
  47
  48 static char*
  49 getMessage(WMConnection *cPtr)
  50 {
  51     char *buffer;
  52     WMData *aData;
  53     int length;
  54
  55     aData = WMGetConnectionAvailableData(cPtr);
  56     if (!aData)
  57         return NULL;
  58     if ((length=WMGetDataLength(aData))==0) {
  59         WMReleaseData(aData);
  60         return NULL;
  61     }
  62
  63     buffer = (char*)wmalloc(length+1);
  64     WMGetDataBytes(aData, buffer);
  65     buffer[length]= '\0';
  66     WMReleaseData(aData);
  67
  68     return buffer;
  69 }
  70
  71
  72 static void
  73 inputHandler(int fd, int mask, void *clientData)
  74 {
  75     WMConnection *cPtr = (WMConnection*)clientData;
  76     WMData *aData;
  77     char buf[4096];
  78     int n;
  79
  80     if (!initialized)
  81         return;
  82
  83     n = read(fd, buf, 4096);
  84     if (n>0) {
  85         aData = WMCreateDataWithBytes(buf, n);
  86         WMSendConnectionData(cPtr, aData);
  87         WMReleaseData(aData);
  88     }
  89 }
  90
  91
  92 static void
  93 didReceiveInput(ConnectionDelegate *self, WMConnection *cPtr)
  94 {
  95     char *buffer;
  96
  97     buffer = getMessage(cPtr);
  98     if (!buffer) {
  99         fprintf(stderr, "Connection closed by peer.\n");
 100         exit(0);
 101     }
 102
 103     printf("%s", buffer);
 104
 105     wfree(buffer);
 106 }
 107
 108
 109 static void
 110 connectionDidDie(ConnectionDelegate *self, WMConnection *cPtr)
 111 {
 112     WMCloseConnection(cPtr);
 113
 114     fprintf(stderr, "Connection closed by peer.\n");
 115     exit(0);
 116 }
 117
 118
 119 static void
 120 didInitialize(ConnectionDelegate *self, WMConnection *cPtr)
 121 {
 122     int state = WMGetConnectionState(cPtr);
 123     WMHost *host;
 124
 125     if (state == WCConnected) {
 126         host = WMGetHostWithAddress(WMGetConnectionAddress(cPtr));
 127         fprintf(stderr, "connected to '%s:%s'\n",
 128                 host?WMGetHostName(host):WMGetConnectionAddress(cPtr),
 129                 WMGetConnectionService(cPtr));
 130         initialized = 1;
 131         if (host)
 132             WMReleaseHost(host);
 133         return;
 134     } else {
 135         wsyserrorwithcode(WCErrorCode, "Unable to connect");
 136         exit(1);
 137     }
 138 }
 139
 140
 141 int
 142 main(int argc, char **argv)
 143 {
 144     char *ProgName, *host, *port;
 145     int i;
 146     WMConnection *sPtr;
 147
 148     wsetabort(wAbort);
 149
 150     WMInitializeApplication("connect", &argc, argv);
 151
 152     ProgName = strrchr(argv[0],'/');
 153     if (!ProgName)
 154         ProgName = argv[0];
 155     else
 156         ProgName++;
 157
 158     host = NULL;
 159     port = "34567";
 160
 161     if (argc>1) {
 162         for (i=1; i<argc; i++) {
 163             if (strcmp(argv[i], "--help")==0 || strcmp(argv[i], "-h")==0) {
 164                 printf("usage: %s [host [port]]\n\n", ProgName);
 165                 exit(0);
 166             } else {
 167                 if (!host)
 168                     host = argv[i];
 169                 else
 170                     port = argv[i];
 171             }
 172         }
 173     }
 174
 175     printf("Trying to make connection to '%s:%s'\n",
 176            host?host:"localhost", port);
 177
 178     sPtr = WMCreateConnectionToAddressAndNotify(host, port, NULL);
 179     if (!sPtr) {
 180         wfatal("could not create connection. exiting");
 181         exit(1);
 182     }
 183
 184     WMSetConnectionDelegate(sPtr, &socketDelegate);
 185
 186     /* watch what user types and send it over the connection */
 187     WMAddInputHandler(0, WIReadMask, inputHandler, sPtr);
 188
 189     while (1) {
 190         WHandleEvents();
 191     }
 192
 193     return 0;
 194
 195 }
 196
 197