这部分代码在接收文件时由客户端使用:
void do_retr_cmd(int f_sockd){
int fd;
ssize_t nread = 0;
uint32_t fsize, fsize_tmp, total_bytes_read, size_to_receive;
char *filename = NULL, *conferma = NULL, *filebuffer = NULL;
char buf[256], dirp[256], t_buf[256];
memset(dirp, 0, sizeof(dirp));
memset(buf, 0, sizeof(buf));
memset(t_buf, 0, sizeof(t_buf));
printf("Write the name of file to download: ");
fgets(dirp, BUFFGETS, stdin)
filename = NULL;
filename = strtok(dirp, "\n");
sprintf(buf, "RETR %s", dirp);
if(send(f_sockd, buf, strlen(buf), 0) < 0){
perror("Errore durante l'invio del nome del file");
onexit(f_sockd, 0, 0, 1);
}
fsize = 0;
recv(f_sockd, t_buf, sizeof(t_buf), 0)
fsize = atoi(t_buf);
fd = open(filename, O_CREAT | O_WRONLY, 0644);
fsize_tmp = fsize;
filebuffer = (char *)malloc(fsize);
total_bytes_read = 0;
nread = 0;
for(size_to_receive = fsize; size_to_receive > 0;){
nread = read(f_sockd, filebuffer, size_to_receive);
if(nread < 0){
perror("read error on retr");
onexit(f_sockd, 0, 0, 1);
}
if(write(fd, filebuffer, nread) != nread){
perror("write error on retr");
onexit(f_sockd, 0, 0, 1);
}
size_to_receive -= nread;
}
close(fd);
fflush(stdout);
fflush(stdin);
memset(buf, 0, sizeof(buf));
recv(f_sockd, buf, 21, 0)
printf("%s", buf);
memset(buf, 0, sizeof(buf));
memset(t_buf, 0, sizeof(t_buf));
memset(dirp, 0, sizeof(dirp));
free(filebuffer);
}
这部分代码在发送文件时由服务器使用:
void do_server_retr_cmd(f_sockd, m_sockd){
int fd, rc;
uint32_t fsize, size_to_send;
char *filename = NULL, *other = NULL;
char buf[512], t_buf[256];
off_t offset;
struct stat fileStat;
memset(buf, 0, sizeof(buf));
memset(t_buf, 0, sizeof(t_buf));
recv(f_sockd, buf, sizeof(buf), 0)
other = NULL;
filename = NULL;
other = strtok(buf, " ");
filename = strtok(NULL, "\n");
if(strcmp(other, "RETR") == 0){
printf("Ricevuta richiesta RETR\n");
} else /* do something */
fd = open(filename, O_RDONLY);
memset(&fileStat, 0, sizeof(fileStat));
fileStat.st_size = 0;
fstat(fd, &fileStat)
fsize = fileStat.st_size;
snprintf(t_buf, 255, "%" PRIu32, fsize);
send(f_sockd, t_buf, sizeof(t_buf), 0)
offset = 0;
for (size_to_send = fsize; size_to_send > 0; ){
rc = sendfile(f_sockd, fd, &offset, size_to_send);
if (rc <= 0){
perror("sendfile");
onexit(f_sockd, m_sockd, fd, 3);
}
size_to_send -= rc;
}
close(fd);
fflush(stdout);
fflush(stdin);
memset(buf, 0, sizeof(buf));
strcpy(buf, "226 File transfered\n");
send(f_sockd, buf, strlen(buf), 0)
memset(buf, 0, sizeof(buf));
memset(t_buf, 0, sizeof(t_buf));
}
- &GT;错误检查已被省略&lt; -
这两段代码我遇到了很大的问题。当我启动主程序时,我必须写:
1. retr
然后我按下
2. Write the filename to download:
我写了文件名然后按回车
问题是,有时文件正确下载,有时它不下载,但它的一部分显示在stdout(在终端上)。
我不明白为什么我得到了这个奇怪的行为。左:PS:我知道我的代码很丑,但我是C-newbie!
我正在开发Ubuntu amd64并使用GCC-4.6.3(C语言)。
答案 0 :(得分:3)
TCP连接为您提供了可靠的双向字节流,但是您的应用程序消息的界限是&#34;未保留,这意味着可以在另一侧的多个send()
中接收一个recv()
,反之,可以将多个send()
折叠为一个recv()
(你可以收到你发送的最后一块的部分)。好处是你按照发送的顺序接收你发送的字节。
服务器代码中的行recv(f_sockd, buf, sizeof(buf), 0);
假设您在此处读取文件名,而实际上您可以获得客户端发送的256
字节。
您需要在裸TCP上施加某种应用程序级协议。一个非常简单的方法是在文件内容中以文本形式发送文本标题:
file-size file-name\n
因此,您的服务器可以查找第一个换行符,在第一个空格中分割行并具有预期的字节数,以及用于保存这些字节的文件名。在该换行符之后不要忽略接收缓冲区的其余部分,将其保存到文件中。这也使您可以重新使用该连接进行多个文件传输。
希望这有帮助。
答案 1 :(得分:1)
recv(f_sockd, buf, 21, 0)
printf("%s", buf);
这printf
打印了大量的随机垃圾,因为没有实际的协议来控制它接收的内容和打印的内容。例如,printf
如何知道要打印多少字节?
答案 2 :(得分:1)
我的答案的先前版本不太正确,但这就是为什么你看到这种奇怪的行为。您将文件大小发送为
snprintf(t_buf, 255, "%" PRIu32, fsize);
然后您将收到
recv(f_sockd, t_buf, sizeof(t_buf), 0)
但不能保证实际读取sizeof(t_buf)
个字节。然后atoi
有时会返回不正确的大小,文件的其余部分将被视为状态消息,该消息将在末尾打印(直到第一个空字符)。
由于recv
可能无法同时返回您要求的所有数据,因此您必须检查其返回值并可能重复调用recv
:
size_t to_recv = sizeof(t_buf);
size_t rcvd = 0;
while (to_recv > 0) {
ssize_t r = recv(f_sockd, t_buf + rcvd, sizeof(t_buf) - rcvd, 0);
if (r < 0) {
//error
}
else {
to_recv -= r;
rcvd += r;
}
}
显然,你要么必须知道预期的数据量,要么提出其他答案中建议的更好的协议(例如,寻找终结器以确定何时读取了大小)。