Client.cpp

// CLIENT TCP PROGRAM
// Revised and tidied up by
// J.W. Atwood
// 1999 June 30



char* getmessage(char *);



/* send and receive codes between client and server */
/* This is your basic WINSOCK shell */
#pragma comment( linker, "/defaultlib:ws2_32.lib" )
#include <winsock2.h>
#include <ws2tcpip.h>

#include <winsock.h>
#include <stdio.h>
#include <iostream>

#include <windows.h>

#include <fstream>
#include <string>

using namespace std;

//user defined port number
#define REQUEST_PORT 0x5000;

int port=REQUEST_PORT;



//socket data types
SOCKET s;
SOCKADDR_IN sa;         // filled by bind
SOCKADDR_IN sa_in;      // fill with server info, IP, port



//buffer data types
char szbuffer[1024];

char *buffer;

int ibufferlen=0;

int ibytessent;
int ibytesrecv=0;



//host data types
HOSTENT *hp;
HOSTENT *rp;

char localhost[11],
     remotehost[11];


//other

HANDLE test;

DWORD dwtest;


//packet

struct Packet {
    char header[7];
    char sequence[1];
    char message[1016];
};


//reference for used structures

/*  * Host structure

    struct  hostent {
    char    FAR * h_name;             official name of host *
    char    FAR * FAR * h_aliases;    alias list *
    short   h_addrtype;               host address type *
    short   h_length;                 length of address *
    char    FAR * FAR * h_addr_list;  list of addresses *
#define h_addr  h_addr_list[0]            address, for backward compat *
};

 * Socket address structure

 struct sockaddr_in {
 short   sin_family;
 u_short sin_port;
 struct  in_addr sin_addr;
 char    sin_zero[8];
 }; */


int main(void){

    WSADATA wsadata;

    //Blank Packet
    Packet packet;
    memset(&packet,0,1024*sizeof(char));

    try {

        if (WSAStartup(0x0202,&wsadata)!=0){
            cout<<"Error in starting WSAStartup()" << endl;
        } else {
            buffer="WSAStartup was successful\n";
            WriteFile(test,buffer,sizeof(buffer),&dwtest,NULL);

            /* Display the wsadata structure */
            cout<< endl
                << "wsadata.wVersion "       << wsadata.wVersion       << endl
                << "wsadata.wHighVersion "   << wsadata.wHighVersion   << endl
                << "wsadata.szDescription "  << wsadata.szDescription  << endl
                << "wsadata.szSystemStatus " << wsadata.szSystemStatus << endl
                << "wsadata.iMaxSockets "    << wsadata.iMaxSockets    << endl
                << "wsadata.iMaxUdpDg "      << wsadata.iMaxUdpDg      << endl;
        }


        //Display name of local host.

        gethostname(localhost,10);
        cout<<"Local host name is \"" << localhost << "\"" << endl;

        if((hp=gethostbyname(localhost)) == NULL)
            throw "gethostbyname failed\n";

        //Ask for name of remote server

        cout << "please enter your remote server name :" << flush ;
        cin >> remotehost ;
        cout << "Remote host name is: \"" << remotehost << "\"" << endl;

        if((rp=gethostbyname(remotehost)) == NULL)
            throw "remote gethostbyname failed\n";

        //Create the socket
        if((s = socket(AF_INET,SOCK_DGRAM,0))==INVALID_SOCKET)
            throw "Socket failed\n";
        /* For UDP protocol replace SOCK_STREAM with SOCK_DGRAM */

        memset(&sa,0,sizeof(sa));
        sa.sin_family = AF_INET;
        sa.sin_port = htons(port);
        sa.sin_addr.s_addr = htonl(INADDR_ANY); //host to network
        //bind the port to the socket
        if (bind(s,(LPSOCKADDR)&sa,sizeof(sa)) == SOCKET_ERROR)
                throw "can't bind the socket";


        //Specify server address for client to connect to server.
        memset(&sa_in,0,sizeof(sa_in));
        memcpy(&sa_in.sin_addr,rp->h_addr,rp->h_length);
        sa_in.sin_family = rp->h_addrtype;
        sa_in.sin_port = htons(7000);

        //Display the host machine internet address

        cout << "Connecting to remote host:";
        cout << inet_ntoa(sa_in.sin_addr) << endl;

        //Connect Client to the server
        if (connect(s,(LPSOCKADDR)&sa_in,sizeof(sa_in)) == SOCKET_ERROR)
            throw "connect failed\n";

        ibytessent = sendto(s, (const char*)&packet, sizeof(packet), 0,(struct sockaddr*) &sa_in, sizeof(sa_in));
        //ibytessent = send(s, (const char*)&packet, sizeof(packet), 0);

        /* Have an open connection, so, server is

           - waiting for the client request message
           - don't forget to append <carriage return>
           - <line feed> characters after the send buffer to indicate end-of file */

        //append client message to szbuffer + send.

        /*sprintf_s(packet.header,"COMMAND");
        sprintf_s(packet.message,"hello world!\r\n");

        ibytessent=0;
        ibufferlen = 1024;

        if (ibytessent = send(s,szbuffer,1024,0);== SOCKET_ERROR)
            throw "Send failed\n";
        else
            cout << "Message to server: " << szbuffer;

        //wait for reception of server response.
        ibytesrecv=0;
        if((ibytesrecv = recv(s,szbuffer,1024,0)) == SOCKET_ERROR)
            throw "Receive failed\n";
        else
            cout << "hip hip hoorah!: Successful message replied from server: " << szbuffer;*/

        string choice;
        char filename[1500];
        bool error = true;
        //Get command & file name
        cout << "get or put file? ";
        cin >> choice;
        cout << "file name? ";
        cin >> filename;
        while(error) {
            if(choice.compare("get") == 0) {
                cout << "RETR: " << filename << "\n";
                ofstream file (filename, ios::out);
                file.close();
                file.open(filename, ios::out|ios::binary|ios::trunc);
                ofstream::pos_type size;
                sprintf_s(packet.header, "RETR");
                sprintf_s(packet.message, filename);
                memset(szbuffer,0,1024*sizeof(char));
                memcpy(szbuffer,packet.header,8*sizeof(char));
                memcpy(szbuffer+8*sizeof(char),packet.message,1016*sizeof(char));
                //Send file name
                if (ibytessent = send(s,szbuffer,1024,0) == SOCKET_ERROR)
                    throw "Send failed\n";
                memset(packet.message,0,1016*sizeof(char));
                memset(szbuffer+8*sizeof(char),0,1016*sizeof(char));
                //Get size
                if((ibytesrecv = recv(s,szbuffer,1024,0)) == SOCKET_ERROR)
                    throw "Receive failed\n";
                memcpy(packet.message, szbuffer+8*sizeof(char), 1016*sizeof(char));
                cout << "Size: " << atoi(packet.message) << "bytes\n";
                for (size = atoi(packet.message); size > 1016*sizeof(char); size -= 1016) {
                    cout << "Size Remaining: " << size << "bytes\n";
                    memset(packet.message,0,1016*sizeof(char));
                    memset(szbuffer+8*sizeof(char),0,1016*sizeof(char));
                    //Get packet
                    if((ibytesrecv = recv(s,szbuffer,1024,0)) == SOCKET_ERROR)
                        throw "Receive failed\n";
                    memcpy(packet.message, szbuffer+8*sizeof(char), 1016*sizeof(char));
                    file.write(packet.message, 1016*sizeof(char));
                }
                memset(packet.message,0,1016*sizeof(char));
                memset(szbuffer+8*sizeof(char),0,1016*sizeof(char));
                //Get last packet
                if((ibytesrecv = recv(s,szbuffer,1024,0)) == SOCKET_ERROR)
                    throw "Receive failed\n";
                cout << "Last Packet Size: " << size << "bytes\n";
                memcpy(packet.message, szbuffer+8*sizeof(char), size*sizeof(char));
                file.write(packet.message, size*sizeof(char));
                memset(packet.message,0,1016*sizeof(char));
                memset(szbuffer+8*sizeof(char),0,1016*sizeof(char));
                file.close();
                cout << "Transfer Complete!\n";
                error = false;
            } else if(choice.compare("put") == 0) {
                cout << "STOR: " << filename << "\n";
                ifstream file (filename, ios::in|ios::binary|ios::ate);
                ifstream::pos_type size;
                if(file.is_open()) {
                    sprintf_s(packet.header, "STOR");
                    sprintf_s(packet.message, filename);
                    memset(szbuffer,0,1024*sizeof(char));
                    memcpy(szbuffer,packet.header,8*sizeof(char));
                    memcpy(szbuffer+8*sizeof(char),packet.message,1016*sizeof(char));
                    //Send file name
                    if (ibytessent = send(s,szbuffer,1024,0) == SOCKET_ERROR)
                        throw "Send failed\n";
                    size = file.tellg();
                    memset(packet.message,0,1016*sizeof(char));
                    memset(szbuffer+8*sizeof(char),0,1016*sizeof(char));
                    itoa((int)size,packet.message,10);
                    memcpy(szbuffer,packet.header,8*sizeof(char));
                    memcpy(szbuffer+8*sizeof(char),packet.message,1016*sizeof(char));
                    cout << atoi(packet.message) << "\n";
                    //Send size
                    if (ibytessent = send(s,szbuffer,1024,0) == SOCKET_ERROR)
                        throw "Send failed\n";
                    file.seekg(0, ios::beg);
                    memset(packet.message,0,1016*sizeof(char));
                    memset(szbuffer+8*sizeof(char),0,1016*sizeof(char));
                    while(file.read(packet.message,1016*sizeof(char))) {
                        memcpy(szbuffer,packet.header,8*sizeof(char));
                        memcpy(szbuffer+8*sizeof(char),packet.message,1016*sizeof(char));
                        //Send packet
                        if (ibytessent = send(s,szbuffer,1024,0) == SOCKET_ERROR)
                            throw "Send failed\n";
                        memset(packet.message,0,1016*sizeof(char));
                        memset(szbuffer+8*sizeof(char),0,1016*sizeof(char));
                        size -= 1016;
                    }
                    memcpy(szbuffer,packet.header,8*sizeof(char));
                    memcpy(szbuffer+8*sizeof(char),packet.message,size*sizeof(char));
                    //Send last packet
                    if((ibytessent = send(s,szbuffer,1024,0))==SOCKET_ERROR)
                        throw "error in send in server program\n";
                    memset(packet.message,0,1016*sizeof(char));
                    memset(szbuffer+8*sizeof(char),0,1016*sizeof(char));
                    file.close();
                    error = false;
                }
                else {
                    //No file found stuff
                    cout << "File not found or incorrect.\n";
                    cout << "\nnew file name? ";
                    cin >> filename;
                    error = true;
                }
            } else {
                cout << "Invalid command, only get or put accepted!\n";
                cout << "get or put file? ";
                cin >> choice;
                error = true;
            }
        }

    } // try loop

    //Display any needed error response.

    catch (char *str) { cerr<<str<<":"<<dec<<WSAGetLastError()<<endl;}

    //close the client socket
    closesocket(s);

    /* When done uninstall winsock.dll (WSACleanup()) and exit */
    WSACleanup();
    system("pause");
    return 0;
}