use shared send_message

master
andrew 2009-02-18 02:31:29 -06:00
parent c9d154d357
commit 8fd9562671
1 changed files with 12 additions and 40 deletions

View File

@ -11,6 +11,7 @@
#include "cftp.h" #include "cftp.h"
#include "client.h" #include "client.h"
#include "filesystem.h"
/* /*
* Special client debug wrapper. * Special client debug wrapper.
@ -66,44 +67,6 @@ int connect_to_server(const char *address, int portnum)
return skt; return skt;
} }
/*
* Sends a message to the socket. Pass in the socket and message. Expects msg
* to be already validated by validate_command first! Formats response into
* string buffer given.
*
*/
void send_message(int skt, char *msg)
{
char command[CMDLEN], argument[MSGLEN], errmsg[ERRMSGLEN];
memset(command, '\0', CMDLEN);
memset(argument, '\0', MSGLEN);
memset(errmsg, '\0', ERRMSGLEN);
if(send(skt, msg, strlen(msg), 0) < 0)
{
sprintf(errmsg, "Send Error: %s", strerror(errno));
error(errmsg);
}
else if(recv(skt, msg, MSGLEN, 0) < 0)
{
sprintf(errmsg, "Receive Error: %s", strerror(errno));
error(errmsg);
}
else
{
split_command(msg, command, argument);
if(strlen(argument) > 0)
{
sprintf(errmsg, "Response: %s: %s", command, argument);
}
else
{
sprintf(errmsg, "Response: %s", command);
}
cdebug(errmsg);
}
}
/* /*
* Runs the command processor to the server. Handles commands and responses. * Runs the command processor to the server. Handles commands and responses.
* *
@ -111,7 +74,7 @@ void send_message(int skt, char *msg)
void run_client(const char *address, int portnum) void run_client(const char *address, int portnum)
{ {
int skt; int skt;
char msg[MSGLEN]; char msg[MSGLEN], cmd[CMDLEN], argument[MSGLEN];
memset(msg, '\0', MSGLEN); memset(msg, '\0', MSGLEN);
/* simple informational header */ /* simple informational header */
@ -124,7 +87,16 @@ void run_client(const char *address, int portnum)
if(strlen(msg) > 0 && validate_command(msg)) if(strlen(msg) > 0 && validate_command(msg))
{ {
skt = connect_to_server(address, portnum); skt = connect_to_server(address, portnum);
split_command(msg, cmd, argument);
if(!strcmp(cmd, "put"))
{
send_file(skt, argument);
}
else
{
send_message(skt, msg); send_message(skt, msg);
cdebug(msg);
}
close(skt); close(skt);
} }
else if(strlen(msg) > 0 && !strcmp(msg, "help")) else if(strlen(msg) > 0 && !strcmp(msg, "help"))