为了实现TCP客户端采用RAW API 可以没有操作系统。我们将TCP客户端分成4个函数来实现。首先依然是实现TCP客户端的初始化:
/* TCP客户端初始化 */
void Tcp_Client_Initialization(void)
{
struct tcp_pcb *tcp_client_pcb;
ip_addr_t ipaddr;
/* 将目标服务器的IP写入一个结构体,为pc机本地连接IP地址 */
IP4_ADDR(&ipaddr,serverIP[0],serverIP[1],serverIP[2],serverIP[3]);
/* 为tcp客户端分配一个tcp_pcb结构体 */
tcp_client_pcb = tcp_new();
/* 绑定本地端号和IP地址 */
tcp_bind(tcp_client_pcb, IP_ADDR_ANY, TCP_CLIENT_PORT);
if (tcp_client_pcb != NULL)
{
/* 与目标服务器进行连接,参数包括了目标端口和目标IP */
tcp_connect(tcp_client_pcb, &ipaddr, TCP_SERVER_PORT, TCPClientConnected);
tcp_err(tcp_client_pcb, TCPClientConnectError);
}
}
上述初始化的代码很简单,有两个地方需要说一下:一是使用tcp_connect注册连接完成的处理回调函数;二是使用tcp_err注册了连接错误处理回调函数。很明显接下来我们需要实现这两个函数。
连接到服务器成功后的回调函数是tcp_connected_fn类型。在客户端建立一个连接后,内核会调用这个函数。在这个函数中,客户端回想服务器发送最初的操作请求,并且会在这个函数中注册数据接收处理回调函数。
/* TCP客户端连接到服务器回调函数 */
static err_t TCPClientConnected(void *arg, struct tcp_pcb *pcb, err_t err)
{
char clientString[]="This is a new client connection.\r\n";
/* 配置接收回调函数 */
tcp_recv(pcb, TCPClientCallback);
/* 发送一个建立连接的问候字符串*/
tcp_write(pcb,clientString, strlen(clientString),0);
return ERR_OK;
}
对于TCP客户端连接服务器错误回调函数,它是tcp_err_fn类型,在这个程序中主要完成连接异常结束时的一些处理,可以释放一些必要的资源。在这个函数被内核调用时,连接实际上已经断开,相关控制块也已经被删除。所以在这个函数中我们可以重新初始化连接及其资源。在这里额我们就是使用它来重新初始化TCP客户端。
1 /* TCP客户端连接服务器错误回调函数 */
2 static void TCPClientConnectError(void *arg, err_t err)
3 {
4 /* 重新启动连接 */
5 Tcp_Client_Initialization();
6 }
最后我们需要实现的是TCP客户端接收到数据后的数据处理回调函数。这个函数其实就是我们前面连接成功时,注册过的TCP客户端数据接收处理函数。这个函数是tcp_recv_fn类型。这是使用RAW API实现TCP客户端功能最重要的一个函数,因为它决定TCP客户端的具体功能。
/* TCP客户端接收到数据后的数据处理回调函数 */
static err_t TCPClientCallback(void *arg, struct tcp_pcb *pcb, struct pbuf *tcp_recv_pbuf, err_t err)
{
struct pbuf *tcp_send_pbuf;
char echoString[]="This is the server content echo:\r\n";
if (tcp_recv_pbuf != NULL)
{
/* 更新接收窗口 */
tcp_recved(pcb, tcp_recv_pbuf->tot_len);
/* 将接收到的服务器内容回显*/
tcp_write(pcb,echoString, strlen(echoString), 1);
tcp_send_pbuf = tcp_recv_pbuf;
tcp_write(pcb, tcp_send_pbuf->payload, tcp_send_pbuf->len, 1);
pbuf_free(tcp_recv_pbuf);
}
else if (err == ERR_OK)
{
tcp_close(pcb);
Tcp_Client_Initialization();
return ERR_OK;
}
return ERR_OK;
}
到这里,我们就实现了一个简单的TCP客户端。对于TCP客户端的具体功能就在于就收处理回调函数的实现了。具体的应用只是功能上的复杂程度不一样,结构上是一样的。 |