WINGs/connect.c

   1 /*
   2  *  WINGs connect.c: example how to create a netwrok client using WMConnection
   3  *
   4  *  Copyright (c) 1999 Dan Pascu
   5  *
   6  *  This program is free software; you can redistribute it and/or modify
   7  *  it under the terms of the GNU General Public License as published by
   8  *  the Free Software Foundation; either version 2 of the License, or
   9  *  (at your option) any later version.
  10  *
  11  *  This program is distributed in the hope that it will be useful,
  12  *  but WITHOUT ANY WARRANTY; without even the implied warranty of
  13  *  MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
  14  *  GNU General Public License for more details.
  15  *
  16  *  You should have received a copy of the GNU General Public License
  17  *  along with this program; if not, write to the Free Software
  18  *  Foundation, Inc., 675 Mass Ave, Cambridge, MA 02139, USA.
  19  */
  20
  21
  22 #include <stdio.h>
  23 #include <unistd.h>
  24 #include <string.h>
  25
  26 #include "WINGs.h"
  27
  28
  29
  30 static int initialized = 0;
  31
  32
  33
  34 static void didReceiveInput(ConnectionDelegate *self, WMConnection *cPtr);
  35
  36 static void connectionDidDie(ConnectionDelegate *self, WMConnection *cPtr);
  37
  38 static void didInitialize(ConnectionDelegate *self, WMConnection *cPtr);
  39
  40
  41
  42 static ConnectionDelegate socketDelegate = {
  43     NULL,              /* data */
  44     NULL,              /* didCatchException */
  45     connectionDidDie,  /* didDie */
  46     didInitialize,     /* didInitialize */
  47     didReceiveInput,   /* didReceiveInput */
  48     NULL               /* didTimeout */
  49 };
  50
  51
  52
  53 void
  54 wAbort(Bool foo) /*FOLD00*/
  55 {
  56     exit(1);
  57 }
  58
  59
  60 static void
  61 printHelp(char *progname) /*FOLD00*/
  62 {
  63     printf("usage: %s [host [port]]\n\n", progname);
  64     printf(" --help                     print this message\n");
  65 }
  66
  67
  68 static char*
  69 getMessage(WMConnection *cPtr)
  70 {
  71     char *buffer;
  72     WMData *aData;
  73     int length;
  74
  75     aData = WMGetConnectionAvailableData(cPtr);
  76     if (!aData)
  77         return NULL;
  78     if ((length=WMGetDataLength(aData))==0) {
  79         WMReleaseData(aData);
  80         return NULL;
  81     }
  82
  83     buffer = (char*)wmalloc(length+1);
  84     WMGetDataBytes(aData, buffer);
  85     buffer[length]= '\0';
  86     WMReleaseData(aData);
  87
  88     return buffer;
  89 }
  90
  91
  92 static void
  93 inputHandler(int fd, int mask, void *clientData)
  94 {
  95     WMConnection *cPtr = (WMConnection*)clientData;
  96     WMData *aData;
  97     char buf[4096];
  98     int n;
  99
 100     if (!initialized)
 101         return;
 102
 103     n = read(fd, buf, 4096);
 104     if (n>0) {
 105         aData = WMCreateDataWithBytes(buf, n);
 106         WMSendConnectionData(cPtr, aData);
 107         WMReleaseData(aData);
 108     }
 109 }
 110
 111
 112 static void
 113 didReceiveInput(ConnectionDelegate *self, WMConnection *cPtr) /*FOLD00*/
 114 {
 115     char *buffer;
 116
 117     buffer = getMessage(cPtr);
 118     if (!buffer) {
 119         fprintf(stderr, "Connection closed by peer.\n");
 120         exit(0);
 121     }
 122
 123     printf("%s", buffer);
 124
 125     wfree(buffer);
 126 }
 127
 128
 129 static void
 130 connectionDidDie(ConnectionDelegate *self, WMConnection *cPtr) /*FOLD00*/
 131 {
 132     WMCloseConnection(cPtr);
 133
 134     fprintf(stderr, "Connection closed by peer.\n");
 135     exit(0);
 136 }
 137
 138
 139 static void
 140 didInitialize(ConnectionDelegate *self, WMConnection *cPtr)
 141 {
 142     int state = WMGetConnectionState(cPtr);
 143
 144     if (state == WCConnected) {
 145         fprintf(stderr, "connected to '%s:%s'\n", WMGetConnectionAddress(cPtr),
 146                 WMGetConnectionService(cPtr));
 147         initialized = 1;
 148         return;
 149     } else {
 150         wsyserrorwithcode(WCErrorCode, "Unable to connect");
 151         exit(1);
 152     }
 153 }
 154
 155
 156 int
 157 main(int argc, char **argv) /*FOLD00*/
 158 {
 159     char *ProgName, *host, *port;
 160     int i;
 161     WMConnection *sPtr;
 162
 163     wsetabort(wAbort);
 164
 165     WMInitializeApplication("connect", &argc, argv);
 166
 167     ProgName = strrchr(argv[0],'/');
 168     if (!ProgName)
 169         ProgName = argv[0];
 170     else
 171         ProgName++;
 172
 173     host = NULL;
 174     port = "7000";
 175
 176     if (argc>1) {
 177         for (i=1; i<argc; i++) {
 178             if (strcmp(argv[i], "--help")==0) {
 179                 printHelp(argv[0]);
 180                 exit(0);
 181             } else {
 182                 if (!host)
 183                     host = argv[i];
 184                 else
 185                     port = argv[i];
 186             }
 187         }
 188     }
 189
 190     sPtr = WMCreateConnectionToAddressAndNotify(host, port, NULL);
 191     if (!sPtr) {
 192         wfatal("could not create connection. exiting");
 193         exit(1);
 194     }
 195
 196     WMSetConnectionDelegate(sPtr, &socketDelegate);
 197
 198     /* watch what user types and send it over the connection */
 199     WMAddInputHandler(0, WIReadMask, inputHandler, sPtr);
 200
 201     while (1) {
 202         WHandleEvents();
 203     }
 204
 205     return 0;
 206
 207 }
 208
 209