//widget.h
#ifndef WIDGET_H
#define WIDGET_H
#include <QWidget>
#include <QTcpSocket>
#include <QPushButton>
#include <QLineEdit>
#include <QLabel>
#include <QCloseEvent>
#include <QTextBrowser>
class Widget : public QWidget
{
Q_OBJECT
public:
Widget(QWidget *parent = );
~Widget();
private:
QTcpSocket *tcpsocket;
QPushButton *btn1,*btn2;
QLineEdit *edit1,*edit2,*edit3;
QLabel *label1,*label2,*label3;
QTextBrowser *textb;
void closeEvent(QCloseEvent *event);
private slots:
void btn1_click();
void btn2_click();
void myrecvdata();
};
#endif // WIDGET_H
//widget.cpp
#include "widget.h"
#include <QHBoxLayout>
#include <QVBoxLayout>
#include <QMessageBox>
#include <QHostAddress>
Widget::Widget(QWidget *parent)
: QWidget(parent)
{
tcpsocket=new QTcpSocket(this);
btn1=new QPushButton(tr("连接"));
btn2=new QPushButton(tr("发送"));
connect(btn1,SIGNAL(clicked()),this,SLOT(btn1_click()));
connect(btn2,SIGNAL(clicked()),this,SLOT(btn2_click()));
edit1=new QLineEdit();
edit2=new QLineEdit();
edit3=new QLineEdit();
label1=new QLabel(tr("IP地址："));
label2=new QLabel(tr("端口号："));
label3=new QLabel(tr("消息内容："));
textb=new QTextBrowser();
QHBoxLayout *lay1=new QHBoxLayout();
lay1->addWidget(label1);
lay1->addWidget(edit1);
lay1->addWidget(label2);
lay1->addWidget(edit2);
lay1->addWidget(btn1);
QHBoxLayout *lay2=new QHBoxLayout();
lay2->addWidget(label3);
lay2->addWidget(edit3);
lay2->addWidget(btn2);
QHBoxLayout *lay4=new QHBoxLayout();
lay4->addWidget(textb);
QVBoxLayout * lay3=new QVBoxLayout(this);
lay3->addLayout(lay1);
lay3->addLayout(lay2);
lay3->addLayout(lay4);
/*接收数据*/
connect(tcpsocket,SIGNAL(readyRead()),this,SLOT(myrecvdata()));
}
//连接
void Widget::btn1_click()
{
/*禁用按钮,值是false*/
btn1->setEnabled(false);
//获取IP地址
QString ipaddr=edit1->text();
edit1->setEnabled(false);
//获取端口号
QString port=edit2->text();
edit2->setEnabled(false);
QHostAddress * serverip=new QHostAddress();
serverip->setAddress(ipaddr);
tcpsocket->connectToHost(*serverip,port.toInt());
//释放socket连接
delete serverip;
}
//发送
void Widget::btn2_click()
{
QString strtext=edit3->text();
if(!strtext.isEmpty())
tcpsocket->write(strtext.toStdString().data());
/*清空输入框*/
edit3->clear();
/*设置输入框重新获得焦点*/
edit3->setFocus();
}
//接收消息
void Widget::myrecvdata()
{
char buf[]={};
/*bytesAvailable()表示有效数据*/
while(tcpsocket->bytesAvailable()>)
{
memset(buf,,sizeof(buf));
tcpsocket->read(buf,sizeof(buf));
textb->append(buf);
//QMessageBox::information(this,"消息",buf);
}
}
//关闭
void Widget::closeEvent(QCloseEvent *event)
{
if(QMessageBox::question(this,"提示","你确定要退出程序吗？",QMessageBox::Yes|QMessageBox::No,QMessageBox::No)==QMessageBox::Yes)
{
event->accept();
}else
{
event->ignore();
}
}
Widget::~Widget()
{
}
//pub.c
#include "pub.h"
//create socket
int create_socket(int port)
{
if (port <= )
{
printf("create_socket() param not correct!\n");
return -;
}
int st = socket(AF_INET, SOCK_STREAM, );
if (st < )
{
printf("socket() failed ! error message:%s\n", strerror(errno));
return -;
}
//reuse address
int on = ;
if (setsockopt(st, SOL_SOCKET, SO_REUSEADDR, &on, sizeof(on)) < )
{
printf("socket() failed ! error message:%s\n", strerror(errno));
close(st);
return -;
}
//bind
struct sockaddr_in addr;
memset(&addr, , sizeof(addr));
addr.sin_family = AF_INET;
addr.sin_port = htons(port);
addr.sin_addr.s_addr = htonl(INADDR_ANY);
if (bind(st, (struct sockaddr *) &addr, sizeof(addr)) < )
{
printf("bind() failed ! error message:%s\n", strerror(errno));
close(st);
return -;
}
//listen
if (listen(st, ) < )
{
printf("listen() failed ! error message:%s\n", strerror(errno));
close(st);
return -;
}
return st;
}
//set noblock
int setnonblock(int st)
{
if (st < )
{
printf("setnonblock() param not correct!\n");
return -;
}
int opts = fcntl(st, F_GETFL);
if (opts < )
{
printf("fcntl() failed ! error message:%s\n", strerror(errno));
return -;
}
opts = opts | O_NONBLOCK;
if (fcntl(st, F_SETFL, opts))
{
printf("fcntl() failed ! error message:%s\n", strerror(errno));
return -;
}
return opts;
}
//net address to
int sockaddrtoa(const struct sockaddr_in *addr, char *ipaddr)
{
if (addr == NULL || ipaddr == NULL)
{
printf("sockaddrtoa() param not correct!\n");
return -;
}
unsigned char *p = (unsigned char *) &(addr->sin_addr.s_addr);
sprintf(ipaddr, "%u:%u:%u:%u", p[], p[], p[], p[]);
return ;
}
//accept clinet
int accept_client(int st)
{
if (st < )
{
printf("accept_client() param not correct!\n");
return -;
}
struct sockaddr_in addr;
memset(&addr, , sizeof(addr));
socklen_t len = sizeof(addr);
int client_st = accept(st, (struct sockaddr *) &addr, &len);
if (client_st < )
{
printf("accept() failed ! error message:%s\n", strerror(errno));
return -;
}
char ipstr[] = { };
sockaddrtoa(&addr, ipstr);
printf("accept by %s\n", ipstr);
return client_st;
}
//recv message
int recv_message(int st)
{
if (st < )
{
printf("accept_client() param not correct!\n");
return -;
}
char buf[]={};
int rc=recv(st,buf,sizeof(buf),);
if(rc<)
{
printf("recv() failed ! error message:%s\n",strerror(errno));
return -;
}else if(rc==)
{
/*这里需要判断客户端是不是关闭连接了*/
printf("client is closed!\n");
return -;
}
printf("%s\n",buf);
//发送接收信息
memset(buf,,sizeof(buf));
strcpy(buf,"server have recved !\n");
if(send(st,buf,sizeof(buf),)<)
{
printf("send() failed ! error message:%s\n",strerror(errno));
return -;
}
return ;
}
//select
int open_select(int st)
{
if (st < )
{
printf("open_select() param not correct!\n");
return -;
}
int i = ;
/*池子中只存放客户端socket*/
int socketpond[MAXPondNum] = { };
/*
* 注意int socketpond[MAXPondNum] = { -1 };
* 改写法并不能将所有元素初始化为-1，只有0这种唯一特例
* */
for(i=;i<MAXPondNum;i++)
{
socketpond[i]=-;
}
/*定义事件数组中socket最大的值*/
int maxfd = ;
/*准备监听事件数组结构*/
fd_set allsocket;
while ()
{
/*清空有消息的事件数组*/
FD_ZERO(&allsocket);
/*将服务器socket直接放入事件数组中*/
FD_SET(st, &allsocket);
/*假设服务器socket是事件数组中最大的*/
maxfd = st;
/*遍历现有socket池,找出最大的socket，将所有的socket添加到事件数组中*/
for (i = ; i < MAXPondNum; i++)
{
if (socketpond[i] != -)
{
FD_SET(socketpond[i], &allsocket);
if (socketpond[i] > maxfd)
{
maxfd = socketpond[i];
}
}
}
int events = select(maxfd + , &allsocket, NULL, NULL, NULL);
if (events < )
{
printf("select() failed ! error message:%s\n", strerror(errno));
break;
}
/*
* 猜想：当两个客户端同时连接到服务器时，select是如何处理服务器socket事件的呢
* 我认为一个服务器端socket只能处理一个客户端连接请求
* 事件数组中只有一个服务器socket，这个服务器socket只会处理一个客户端连接请求，
* 而此时另外一个客户端请求会在下次的select事件数组中被处理
* */
/*先处理服务器socket特殊处理*/
if (FD_ISSET(st, &allsocket))
{
printf("accept client!\n");
//accept
int client_st = accept_client(st);
if (client_st < )
{
break;
}
//set nonblock
if (setnonblock(client_st) < )
{
break;
}
for (i = ; i < MAXPondNum; i++)
{
if (socketpond[i] == -)
{
socketpond[i] = client_st;
break;
}
}
if (i == MAXPondNum)
{
printf("服务端连接池已满！\n");
close(client_st);
}
}
for (i = ; i < MAXPondNum; i++)
{
if (socketpond[i] < )
{
continue;
}
if (FD_ISSET(socketpond[i], &allsocket))
{
//接收客户端发送消息
if(recv_message(socketpond[i])<)
{
//接收客户端消息出错
//关闭客户端
close(socketpond[i]);
//从服务器端socket池子中将该socket清除
socketpond[i]=-;
}
/*需要处理的事件数减一*/
events--;
}
if(events<)
{
break;
}
}
}
return ;
}
