---

Muxer.C

Go to the documentation of this file.

#include <stdio.h>
#include <sched.h>
#include <unistd.h>
#include <time.h>
#include <ctype.h>
#include <list>
#include "Connector.h"
#include "ConnectionWrap.h"
#include "BufferedFile.h"
#include "BufferedFileFactory.h"
#include "ConnectionFactory.h"
#include "Connection.h"
#include "FilterConnection.h"
#include "OutputConnection.h"
#include "Clock.h"
#include "connect.h"
#define BUFFSIZE 512

int connect(hostent * he, int port);
bool isNum(char *arg);
string getIP(int ip);
int main(int argc, char *argv[]){
        try {
                string * name = NULL;
                list<hostent> hosts;
                list<int> ports;
                try {
                        if (argc >= 3) {
                                int index = 2;
                                int offset = 0;
                                if ((argc+1)%2 == 1) { //name included  app10 name21 host32 port43
                                        cout << "NAME\n";
                                        name = new string(argv[1]);
                                        index = 2;
                                        offset = 0;
                                } else if ((argc+1)%2 == 0) { //noname app10 host21 port3
                                        cout << "NONAME\n";
                                        name = NULL;
                                        index = 1;
                                        offset = 1;
                                }
                                for (; index < argc; index++) {
                                        if ((index+offset)%2==1) {
                                                if (!isNum(argv[index])) {
                                                        string error = string("Arguement Invalid: ");
                                                        error += string(argv[index]);
                                                        throw error.c_str();
                                                }
                                                int port = atoi(argv[index]);
                                                ports.push_back(port);
                                        } else {
                                                struct hostent *he = gethostbyname(argv[index]);
                                                if (he == NULL) {
                                                        string errstr = "Could not resolve host ";
                                                        errstr += string(argv[index]);
                                                        throw errstr.c_str();
                                                }
                                                hosts.push_back(*he);
                                        }
                                }
                        } else {
                                throw (char*)NULL;
                        }
                } catch (char * err) {
                        if (err != NULL) {
                                cerr << "ERROR: " << err << "\n";
                        }
                        cout << argv[0] << " [name] [host] [port] [host] [port] ...\n";
                        return 0;
                }
                list<ConnectionWrap *> outgoing;
                list<hostent>::iterator hostIter = hosts.begin();
                list<int>::iterator portIter = ports.begin();
                BufferedFileFactory bf;
                BufferedFile * inputBf = bf.getNewBufferedFile(STDIN_FILENO);
                inputBf->setReadOnly();
                //InputConnection<short> * ic = new InputConnection<short>(inputBf);
                //ConnectionWrap * icw = new ConnectionWrap(ic,SHORTCONN);
                char * buffer = new char[BUFFSIZE * 2];//(icw->getDataSize())];
                int framesize = 2;
                while (hostIter != hosts.end() && portIter != ports.end()) {
                        int port = *portIter;
                        hostent host = *hostIter;
                        int fd =  connect(&host,port);
                        cout << host.h_name << " " << port << "\n";
                        if (fd!=-1) {
                                cout << "FD: " << fd <<"\n";
                                BufferedFile * outputBf = bf.getNewBufferedFile(fd);
                                outputBf->setWriteOnly();
                                OutputConnection<short> * oc = new OutputConnection<short>(outputBf);
                                outgoing.push_back(new ConnectionWrap(oc,SHORTCONN));
                                //oc->connectFrom((Connection<short>*)ic);
                        }
                        hostIter++;
                        portIter++;
                }
                if (name!=NULL) {
                        string headStart("<header><name>");
                        string nameEnd("</name>");
                        string headEnd("</header>");
                        int length = headStart.length() + headEnd.length() + nameEnd.length() + name->length();
                        if (length%framesize!=0) {
                                int count = framesize - length%framesize - 1;
                                for (int i = 0 ; i < count; i++) {
                                        nameEnd += " ";
                                }
                                length = headStart.length() + headEnd.length() + nameEnd.length() + name->length();
                        }
                        const char * header = (headStart+(*name)+nameEnd+headEnd).c_str();
                        
                        bf.process(100);
                        list<ConnectionWrap *>::iterator iter;
                        for (iter = outgoing.begin(); iter != outgoing.end(); iter++) {
                                ConnectionWrap * cw = (*iter);
                                //if (cw->writeReady()) {
                                        cw->write((char*)header,length);
                                //}
                        }
                        
                }
                for (;;) {
                        int rw = bf.process(100);
                        Clock::clock++;
                        if (rw >= 2) {
                                        //int r = icw->read((char*)buffer,BUFFSIZE);
                                        //int r = ic->read((short*)buffer,BUFFSIZE);
                                        read(STDIN_FILENO,buffer,BUFFSIZE*2);
                                        list<ConnectionWrap *>::iterator iter;
                                        for (iter = outgoing.begin(); iter != outgoing.end(); iter++) {
                                                ConnectionWrap * cw = (*iter);
                                                if (cw->writeReady()) {
                                                        cw->write((char*)buffer,BUFFSIZE);
                                                }
                                        }
                        }
                }
        } catch (char * str) {
                cerr<< "G EXCEPTION: " <<str<<"\n";
        } catch (string * str) {
                cerr<< "G EXCEPTION: " <<*str<<"\n";
        }
        return 0;
}
bool isNum(char *arg) {
        for (int i = 0; arg[i]!='\0' ; i++) {
                if (!isdigit(arg[i])) {
                        return false;
                }
        }
        return true;
}
string getIP(int ip) {
        int i1 = ip >> 24;
        int i2 = (ip << 8 )>> 24;
        int i3 = (ip << 16)>> 24;
        int i4 = (ip << 24)>> 24;
        char data[12];
        string out = "";
        out+=string((snprintf(data,12,"%d",i1),data));//Bad coding style uses , operator
        out+=string((snprintf(data,12,"%d",i2),data));
        out+=string((snprintf(data,12,"%d",i3),data));
        out+=string((snprintf(data,12,"%d",i4),data));
        return out;
}

---