int
main(int argc, char *argv[])
{
int sockFd;
struct sockaddr_in servAddr;
int n;
int msgLen;
/* a simple command line argument processing */
if (argc != 2)
{
fprintf(stderr, "Usage: %s <IPaddress>", argv[0]);
exit(1);
}
/* open a socket */
if ((sockFd = socket(AF_INET, SOCK_STREAM, 0)) < 0)
{
perror("socket error");
exit(1);
}
/* setup the server address and connect to it */
bzero(&servAddr, sizeof(servAddr));
servAddr.sin_family = AF_INET;
servAddr.sin_port = htons(NETCALC_SERV_PORT);
/*
* Inet_pton(AF_INET, argv[1], &servAddr.sin_addr) is only supported by
* the book lib. There is no inet_pton in native Solaris.
*/
servAddr.sin_addr.s_addr = inet_addr(argv[1]);
if (connect(sockFd, (struct sockaddr *) &servAddr, sizeof(servAddr)) < 0)
{
perror("connect error");
exit(1);
}
/* main loop */
while ((fgets(ioBuf, BUF_SIZE, stdin)) != NULL)
{
if (buildReq(ioBuf, strlen(ioBuf), msgBuf, &msgLen) == SUCCESS)
{
/* send request to the server */
if (write(sockFd, msgBuf, msgLen) < 0)
{
perror("write error");
exit(1);
}
/* wait for response from the server */
msgLen = read(sockFd, msgBuf, BUF_SIZE);
if (msgLen <= 0)
break;
if (parseRes(msgBuf, msgLen, ioBuf) == SUCCESS)
{
fputs(ioBuf, stdout);
}
else
{
printf("!!%s: parseRes failed\n", __FUNCTION__);
DUMP_BUF(msgBuf, msgLen);
}
}
else
{
printf("!!%s: buildReq failed\n", __FUNCTION__);
DUMP_BUF(ioBuf, strlen(ioBuf));
}
}
close(sockFd);
exit(0);
}
int main (void){
char * req = NULL;
char mess[BUFFSIZE];
char * data = NULL;
char file[MAX_FILE_SIZE];
int sockNum = 0, recSock = 0;
int i = 0;
int rc=0, rcr = 0;
int accepted = 0;
int cs=0;
int chunkiterator = 0;
int offset= 0;
struct sockaddr_in from;
socklen_t fromlen;
FILE * fp = NULL;
int bytecounter = 0;
while(1){
req = NULL;
for(i = 0; i < BUFFSIZE ; i++) mess[i] = 0;
data = NULL;
for(i = 0; i < MAX_FILE_SIZE ; i++) file[i] = 0;
sockNum = 0;
recSock = 0;
i = 0;
rc=0;
rcr = 0;
accepted = 0;
cs=0;
chunkiterator = 0;
offset= 0;
fromlen = 0;
fp = NULL;
bytecounter = 0;
//build the request string
req = buildReq(NUMCHUNKS, DATAPORT, FILENAME);
//report request
printf("sent request for \"%s\" in %d chunks over port %d\n", FILENAME, NUMCHUNKS, DATAPORT);
//initialize buffers
//open socket
sockNum = Open("localhost", 11708);
if (DEBUG) fprintf(stderr, "Socket Open\n");
//start listening for response
recSock = Listen(DATAPORT);
//send the request
rc = send( sockNum , req , (strlen(&req[4])+ 4) , 0 ) ;
printf( "Sent %d\n%s\n" , rc , req ) ;
//clean up
free(req);
req = NULL;
close(sockNum);
//start loop for each connection
for (chunkiterator = 0; chunkiterator < NUMCHUNKS; chunkiterator++){
//initialize variables
for(i = 0; i < BUFFSIZE ; i++) mess[i] = 0;
accepted = 0;
offset =0;
//try to accept a connection
while (!accepted){
if (DEBUG) fprintf(stderr, "trying to accept connection. accepted = %d\n", accepted);
fromlen = sizeof(from);
cs = accept( recSock , (struct sockaddr *)&from , &fromlen );
if (DEBUG) fprintf(stderr, "cs = %d\n", cs);
//if it failed, report it
//if it succeeded, raise tthe flafg
if (cs == -1){
perror("accept:");
}else{
(accepted = 1);
}
}
if (DEBUG) fprintf(stderr, "done accepting. cs = %d\n", cs);
do {
//cleart buffer
if (DEBUG) fprintf(stderr, "clearing buffer\n");
for(i = 0; i < BUFFSIZE ; i++) mess[i] = 0;
//read from the socket into the buffer
rc = recv( cs , &mess , sizeof(mess) , 0 ) ;
if (DEBUG) fprintf(stderr, "finished reading %d bytes from socket. offset is currently %d\n", rc, offset);
if (rc == -1)
perror("recv");
//if something was in fact read
if (rc){
//print it out
//if (DEBUG) fprintf(stderr, "recieved: [%s]\n", mess);
//if this is the start of the packet find the offset
if (offset == 0){
offset = ntohl(*((int*)mess));
if(DEBUG) fprintf(stderr, "firstoffset = %d\n", offset);
data = &mess[4];
rcr = rc - 4;
// otherwise deal with it as normal
} else {
data = mess;
rcr = rc;
}
if (DEBUG) fprintf(stderr, "done calculating offset and data pointers\n");
//copy the data character by character into the file buffer
for(i = 0; i < rcr; i++){
//if (DEBUG) fprintf(stderr, "112: writing \'%c\' from index %d in buffer to index %d in file\n", data[i], i, *offset+i);
file[offset] = data[i];
offset++;
bytecounter++;
}
if (DEBUG) fprintf(stderr, "done copying data\n");
}
} while (rc);
close(cs);
}
//fprintf(stderr, "Final File:\n%s\nEnd of File\n", file);
fp = fopen(WRITENAME, "w+");
fwrite(file, bytecounter, 1, fp);
fclose(fp);
printf("wrote %d bytes to the file\n", bytecounter);
close(recSock);
usleep(5000000);
}
return 0;
}
