在Ubuntu 12.10上使用C ++解决TCP聊天程序问题

时间:2013-03-12 20:36:35

标签: c++ sockets ubuntu tcp

我正在开发一个通过TCP的聊天程序,允许我通过WAN在两台机器之间进行通信。我是C ++的新手(来自Java)并且对TCP来说很新,所以不要对我太过刻薄! (:我已经浏览了相当多的教程并继续只找到Echo程序,没有任何东西可以让连接打开更长的时间以允许聊天功能。我当前的代码看起来像这样:

#include "ClientManager.h"

ClientManager::ClientManager() {

}

void ClientManager::connectCom(char* ipAdd) {



portno = atoi(PORT);
sockfd = socket(AF_INET, SOCK_STREAM, 0);
if (sockfd < 0) 
    error("ERROR opening socket");
server = gethostbyname(ipAdd);
if (server == NULL) {
    fprintf(stderr,"ERROR, no such host\n");
    exit(0);
}
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("ERROR connecting");
}

void ClientManager::message(std::string msg) {
// printf("Please enter the message: ");

char * buffer = new char[msg.size() + 1];
std::copy(msg.begin(), msg.end(), buffer);
buffer[msg.size()] = '\0';
// bzero(buffer,256);
// fgets(buffer,255,stdin);
n = write(sockfd,buffer,strlen(buffer));
if (n < 0) 
     error("ERROR writing to socket");
bzero(buffer,256);
n = read(sockfd,buffer,255);
if (n < 0) 
     error("ERROR reading from socket");
printf("%s\n",buffer);
bzero(buffer,256);
}

void ClientManager::closeCom() {
close(sockfd);
}

void ClientManager::error(const char *msg)
{
perror(msg);
exit(0);
}

我的服务器管理器看起来像这样:

#include "ServerManager.h"

ServerManager::ServerManager() {
    // int sockfd, portno, n;
 //         struct sockaddr_in serv_addr;
 //     struct hostent *server;

}

void ServerManager::openCom() {
char buffer[256];
struct sockaddr_in serv_addr, cli_addr;
int n;
     socklen_t clilen;


sockfd = socket(AF_INET, SOCK_STREAM, 0);
if (sockfd < 0) 
    error("ERROR opening socket");
bzero((char *) &serv_addr, sizeof(serv_addr));
portno = atoi(PORT);
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("ERROR on binding");
listen(sockfd,5);

clilen = sizeof(cli_addr);
newsockfd = accept(sockfd, 
            (struct sockaddr *) &cli_addr, 
            &clilen);
if (newsockfd < 0) 
    error("ERROR on accept");
bzero(buffer,256);
// n = read(newsockfd,buffer,255);
// if (n < 0) error("ERROR reading from socket");
//  printf("Here is the message: %s\n",buffer);
// n = write(newsockfd,"I got your message",18);
// if (n < 0) error("ERROR writing to socket");
}

int ServerManager::readCom() {
bzero(buffer,256);
n = read(newsockfd,buffer,255);
if (n < 0) error("ERROR reading from socket");
    printf("Here is the message: %s\n",buffer);
n = write(newsockfd,"I got your message",18);
if (n < 0) error("ERROR writing to socket");
if (buffer[0] == '0') 
    return 1;
return 0;
}

void ServerManager::closeCom() {
close(newsockfd);
close(sockfd);

}

void ServerManager::error(const char *msg) {
perror(msg);
exit(1);
}

这两个类都是由调用必要函数的独立主函数实现的。我知道代码本身可以发送一条消息 - 它现在已经做了一段时间了。它恰好在客户端调用多条消息()时遇到错误,特别是分段错误。这只发生在第二条消息上,第一条消息被正确发送和接收。

如果有人能帮助我,我们将不胜感激。谢谢!

1 个答案:

答案 0 :(得分:0)

在ClientManager :: message中,您忘记删除方法末尾的缓冲区。 或者,您应该在本地堆栈上声明缓冲区:

char buffer[msg.size() + 1];

因此,缓冲区将在呼叫结束时自动解除分配。 阅读以后的代码,你最好这样做:

char buffer[256];

我认为,如果你的msg.size()&lt;而且,读取服务器回复的代码会导致缓冲区溢出。 serverReply.size,即使你认为你有256个字符(你只是在方法启动时没有分配它们)。请注意这个缓冲区长度声明和稍后的处理,因为C ++会让你写出超出它的结束,破坏相邻变量并导致段错误。