我必须编写一个包含整数的 TCP 服务器程序,该整数应该可以由客户端程序修改。它应该有点像银行账户。一切正常,除了一件事:
当客户端第一次连接到服务器时,它会等待欢迎消息(服务器必须是迭代的,所以它一次只能处理一个客户端)。服务器总是只发送欢迎消息的前几个字母。所有其他消息均已完整且正确地传输。
在第 49 行,欢迎消息首先被复制到一个字符数组,然后写入套接字。这就是错误所在...仅发送前 1-5 个字母(每次新客户端连接时都不同)。在我使用 sprintf() 将消息复制到字符数组然后将其写入套接字的其他地方,一切都按照我想要的方式工作。 我也尝试过使用 snprintf(),但这也不起作用。我究竟做错了什么? :D
所以这将是客户端的示例输出:
Connected!
Waiting for welcome message...
We
之后,我就可以开始向服务器输入命令了。但是整个欢迎信息在两封信之后就被删掉了。但如上所述,有时只有一个字母,有时是五个 :D。
无论如何,这是我的代码(如果有任何其他错误或我应该避免的事情,请随时告诉我 :D):
客户:
#include <stdio.h>
#include <stdlib.h>
#include <unistd.h>
#include <string.h>
#include <sys/types.h>
#include <sys/socket.h>
#include <netinet/in.h>
#include <netdb.h>
#define BufferSize 99999
void error(const char *msg) {
fprintf(stderr, "%s\n", msg);
exit(EXIT_FAILURE);
}
int main(int argc, char *argv[]) {
int sockfd, portno, n;
struct sockaddr_in serv_addr;
struct hostent *server;
char msg[BufferSize], data[BufferSize];
if (argc < 3) error("usage: <hostname> <port>\n");
server = gethostbyname(argv[1]);
if (server == NULL) error("Host not found!");
portno = atoi(argv[2]);
sockfd = socket(AF_INET, SOCK_STREAM, 0);
if (sockfd < 0) error("socket() error");
bzero((char *) &serv_addr, sizeof (serv_addr));
serv_addr.sin_family = AF_INET;
bcopy((char *) server->h_addr, (char *) &serv_addr.sin_addr.s_addr, server->h_length);
serv_addr.sin_port = htons(portno);
if (connect(sockfd, (struct sockaddr *) &serv_addr, sizeof (serv_addr)) < 0) error("connect() error");
printf("Connected!\nWaiting for welcome message...\n");
memset(msg, 0, BufferSize);
n = read(sockfd, msg, BufferSize - 1);
if (n < 0) error("read() error");
printf("%s\n", msg);
memset(data, 0, BufferSize);
while (fgets(data, BufferSize, stdin) != NULL) {
data[strlen(data) - 1] = '\0'; //remove trailing newline char
n = write(sockfd, data, strlen(data) + 1);
if (n < 0) error("write() error");
if (strcmp(data, "exit") == 0) break;
memset(msg, 0, BufferSize);
n = read(sockfd, msg, BufferSize - 1);
if (n < 0) error("read() error");
if (n==0) error("Server shut down...");
printf("%s\n", msg);
memset(data, 0, BufferSize);
}
close(sockfd);
return 0;
}
服务器:
#include <stdio.h>
#include <stdlib.h>
#include <string.h>
#include <unistd.h>
#include <sys/types.h>
#include <sys/socket.h>
#include <netinet/in.h>
#include <limits.h>
#define BufferSize 99999
#define ClientWaiting 100
void error(const char *msg) {
fprintf(stderr, "%s\n", msg);
exit(EXIT_FAILURE);
}
int main(int argc, char *argv[]) {
int sockfd, newsockfd, portno, n, amount, balance, balOld;
socklen_t clilen;
char msg[BufferSize], data[BufferSize], *splitBuf[2];
struct sockaddr_in serv_addr, cli_addr;
balance = 0;
if (argc < 2) error("usage: <port>");
portno = atoi(argv[1]);
sockfd = socket(AF_INET, SOCK_STREAM, 0);
if (sockfd < 0) error("socket() error");
memset(&serv_addr, 0, sizeof (serv_addr));
serv_addr.sin_family = AF_INET;
serv_addr.sin_addr.s_addr = INADDR_ANY;
serv_addr.sin_port = htons(portno);
if (bind(sockfd, (struct sockaddr *) &serv_addr, sizeof (serv_addr)) < 0) error("bind() error");
listen(sockfd, ClientWaiting);
while (1) {
printf("Waiting for new client...\n");
clilen = sizeof (cli_addr);
newsockfd = accept(sockfd, (struct sockaddr *) &cli_addr, &clilen);
if (newsockfd < 0) error("accept() error");
printf("New connection accepted...\n");
memset(data, 0, BufferSize);
sprintf(data, "Welcome!\nPlease use the following commands:\n<put, get> <positive integer>\nBalance: %d€", balance);
n = write(newsockfd, data, strlen(msg) + 1);
if (n < 0) error("write() error");
while (1) {
splitBuf[0] = NULL;
splitBuf[1] = NULL;
memset(data, 0, BufferSize);
memset(msg, 0, BufferSize);
n = read(newsockfd, msg, BufferSize - 1);
if (n < 0) {
fprintf(stderr, "read() error\n");
break;
}
if (n == 0) {
printf("Client disconnected...\n");
break;
}
printf("Message received: %s\n", msg);
if (strcmp(msg, "exit") == 0) break;
splitBuf[0] = strtok(msg, " ");
splitBuf[1] = strtok(NULL, " ");
if (splitBuf[1] == NULL) {
strcpy(data, "Please use the following commands:\n<put, get> <positive integer>");
} else {
amount = atoi(splitBuf[1]);
if (amount <= 0) {
strcpy(data, "Please use the following commands:\n<put, get> <positive integer>");
} else if (strcmp(splitBuf[0], "put") == 0) {
balOld = balance;
balance += amount;
if (balance < balOld) {
balance = INT_MAX;
sprintf(data, "Warning! Overflow!\nBalance: %d€", balance);
} else {
sprintf(data, "Balance: %d€", balance);
}
printf("New balance: %d€\n", balance);
} else if (strcmp(splitBuf[0], "get") == 0) {
balOld = balance;
balance -= amount;
if (balance > balOld) {
balance = INT_MIN;
sprintf(data, "Warning! Underflow!\nBalance: %d€", balance);
} else {
sprintf(data, "Balance: %d€", balance);
}
printf("New balance: %d€\n", balance);
}
}
n = write(newsockfd, data, strlen(data) + 1);
if (n < 0) error("write() error");
}
close(newsockfd);
}
close(newsockfd);
close(sockfd);
return 0;
}
最佳答案
这个例子对于 SO 上的人来说太长了,无法调试。 (我们不是编译器和调试器。)
您的第一步必须是将程序分解成更小、更容易理解的可以独立调试的部分。有几种方法可以做到这一点:
- 添加测试断言的检查点
- 将代码分解成更容易理解和独立测试的函数
例如,我看:
splitBuf[0] = strtok(msg, " ");
splitBuf[1] = strtok(NULL, " ");
splitBuf 是否包含您期望的内容? (NULL 是 strtok 的有效参数吗?)
我推荐两件事: #包括
# Are my assumptions met?
assert( splitBuf[0]!=null );
assert( splitBuf[1]!=null );
#ifdef DEBUG
printf("splitBuf[0]=%s\n", splitBuf[0]);
printf("splitBuf[1]=%s\n", splitBuf[1]);
#endif
编译时加上 -DDEBUG=1 以确保定义了 DEBUG,或者添加:
#define DEBUG
在文件的顶部。
其次,如果您解决较小的问题,然后独立处理并测试您对这些问题的答案,那么编程会容易得多。假设您需要解析来自网络的消息并提取余额,那么您可以这样写:
int parseBalance(char const* serverMessage) {
...
return balance;
}
您现在可以编写测试了:
void tests()
{
// test parseBalance
assert( 100 == parseBalance("100") )
... more tests
}
至少,您可以在程序开始时调用 tests() 来执行自测(阅读“单元测试”以获得更好的方法)。
如果你用这种方式编程:
- 通常问题会变得更加明显
- 如果您卡住了并发布到 SO,您只需要发布最小的功能和测试用例。
关于c - sprintf/snprintf 没有正确写入缓冲区,我们在Stack Overflow上找到一个类似的问题: https://stackoverflow.com/questions/27045367/