C++?socket通信遇到的問(wèn)題及解決方法
為了解決這兩個(gè)問(wèn)題,你可以對(duì)代碼進(jìn)行以下修改:
1.處理socket斷開(kāi)的情況:
在recv函數(shù)返回0的判斷分支中,你可以在斷開(kāi)連接時(shí)進(jìn)行一些處理,例如打印提示信息或進(jìn)行重連操作。
if (ret == 0) {
printf("Remote socket close!!!\n");
// 處理socket斷開(kāi),例如進(jìn)行重連操作
Reconnect();
break;
}你可以在Reconnect函數(shù)中實(shí)現(xiàn)重新連接的邏輯。
2.處理socket連接沒(méi)有發(fā)送消息的情況:
在接收數(shù)據(jù)失敗的判斷分支中,你可以添加一個(gè)超時(shí)機(jī)制,如果在一定時(shí)間內(nèi)沒(méi)有接收到數(shù)據(jù),就認(rèn)為連接沒(méi)有發(fā)送消息。
if (ret < 0) {
if ((errno == EAGAIN) || (errno == EWOULDBLOCK) || (errno == EINTR)) {
std::this_thread::sleep_for(std::chrono::microseconds(10));
continue;
}
if (errno == EWOULDBLOCK || errno == EAGAIN) {
// 沒(méi)有數(shù)據(jù)可讀,可以繼續(xù)執(zhí)行其他任務(wù)
printf("Failed to receive data!!!\n");
// 處理連接沒(méi)有發(fā)送消息的情況,例如重新發(fā)送消息或斷開(kāi)連接
HandleNoMessage();
continue;
} else {
printf("Remote socket error!!!\n");
break;
}
}你可以在HandleNoMessage函數(shù)中實(shí)現(xiàn)處理連接沒(méi)有發(fā)送消息的邏輯。
請(qǐng)注意,以上只是一種修改代碼的思路,具體的實(shí)現(xiàn)方式可能需要根據(jù)你的需求和業(yè)務(wù)邏輯來(lái)進(jìn)行調(diào)整。
要將這段C++代碼改為非阻塞執(zhí)行,需要使用非阻塞的套接字操作和事件循環(huán)機(jī)制。下面是修改后的代碼:
void NtripServer::ThreadHandler(void) {
int ret;
char recv_buffer[1024] = {};
thread_is_running_ = true;
// 將socket_fd_設(shè)置為非阻塞模式
if (SetNonBlocking(socket_fd_) == -1) {
printf("Failed to set socket to non-blocking mode\n");
return;
}
while (thread_is_running_) {
ret = recv(socket_fd_, recv_buffer, sizeof(recv_buffer), 0);
if (ret == 0) {
printf("Remote socket close!!!\n");
break;
} else if (ret < 0) {
if (errno != EAGAIN && errno != EWOULDBLOCK && errno != EINTR) {
printf("Remote socket error!!!\n");
break;
}
} else {
// 處理接收到的數(shù)據(jù)
}
std::this_thread::sleep_for(std::chrono::milliseconds(10));
}
close(socket_fd_);
socket_fd_ = -1;
thread_is_running_ = false;
service_is_running_ = false;
}在修改后的代碼中,我們首先調(diào)用SetNonBlocking函數(shù)將socket_fd_設(shè)置為非阻塞模式。然后,在while循環(huán)中,我們使用recv函數(shù)接收數(shù)據(jù),并根據(jù)返回值進(jìn)行處理。
如果recv返回值為0,表示遠(yuǎn)程socket關(guān)閉,我們打印相應(yīng)的信息并跳出循環(huán)。
如果recv返回值小于0,我們首先判斷errno是否為EAGAIN、EWOULDBLOCK或EINTR。如果不是這些錯(cuò)誤碼,表示發(fā)生了其他錯(cuò)誤,我們打印相應(yīng)的信息并跳出循環(huán)。
如果recv返回值大于0,表示成功接收到數(shù)據(jù),我們可以在代碼中添加處理接收到的數(shù)據(jù)的邏輯。
在每次循環(huán)結(jié)束后,我們使用std::this_thread::sleep_for函數(shù)暫停10毫秒,以避免忙等待。
最后,在循環(huán)結(jié)束后,我們關(guān)閉socket_fd_并將其置為-1,同時(shí)將thread_is_running_和service_is_running_標(biāo)志置為false,表示線(xiàn)程已停止。
是的,SetNonBlocking函數(shù)需要自定義。下面是一個(gè)示例的SetNonBlocking函數(shù)的實(shí)現(xiàn):
int SetNonBlocking(int socket_fd) {
int flags = fcntl(socket_fd, F_GETFL, 0);
if (flags == -1) {
return -1;
}
if (fcntl(socket_fd, F_SETFL, flags | O_NONBLOCK) == -1) {
return -1;
}
return 0;
}這個(gè)函數(shù)的實(shí)現(xiàn)與之前提到的設(shè)置socket為非阻塞模式的代碼相同。它首先使用fcntl函數(shù)獲取socket_fd的文件描述符標(biāo)志,并將返回值保存在變量flags中。如果獲取失敗,函數(shù)返回-1。
然后,它使用fcntl函數(shù)的F_SETFL命令來(lái)修改socket_fd的文件描述符標(biāo)志。修改的方式是將O_NONBLOCK標(biāo)志添加到原有標(biāo)志中,使用位運(yùn)算符|進(jìn)行操作。如果修改失敗,函數(shù)返回-1。
如果設(shè)置成功,函數(shù)返回0。
下面是修改后的完整代碼示例:
#include <iostream>
#include <thread>
#include <chrono>
#include <cstring>
#include <unistd.h>
#include <fcntl.h>
#include <sys/socket.h>
#include <netinet/in.h>
#include <arpa/inet.h>
#include <errno.h>
class NtripServer {
public:
NtripServer() : socket_fd_(-1), thread_is_running_(false), service_is_running_(false) {}
~NtripServer() {}
void Start() {
if (service_is_running_) {
std::cout << "Service is already running!" << std::endl;
return;
}
// 創(chuàng)建socket
socket_fd_ = socket(AF_INET, SOCK_STREAM, 0);
if (socket_fd_ == -1) {
std::cout << "Failed to create socket!" << std::endl;
return;
}
// 設(shè)置socket選項(xiàng)
int optval = 1;
setsockopt(socket_fd_, SOL_SOCKET, SO_REUSEADDR, &optval, sizeof(optval));
// 綁定地址和端口
sockaddr_in server_addr{};
server_addr.sin_family = AF_INET;
server_addr.sin_port = htons(1234);
server_addr.sin_addr.s_addr = INADDR_ANY;
if (bind(socket_fd_, (struct sockaddr*)&server_addr, sizeof(server_addr)) == -1) {
std::cout << "Failed to bind address!" << std::endl;
close(socket_fd_);
socket_fd_ = -1;
return;
}
// 監(jiān)聽(tīng)連接
if (listen(socket_fd_, 5) == -1) {
std::cout << "Failed to listen!" << std::endl;
close(socket_fd_);
socket_fd_ = -1;
return;
}
// 啟動(dòng)線(xiàn)程處理連接
service_is_running_ = true;
std::thread thread(&NtripServer::ThreadHandler, this);
thread.detach();
}
void Stop() {
if (!service_is_running_) {
std::cout << "Service is not running!" << std::endl;
return;
}
service_is_running_ = false;
}
private:
int socket_fd_;
bool thread_is_running_;
bool service_is_running_;
void ThreadHandler() {
int ret;
char recv_buffer[1024] = {};
thread_is_running_ = true;
// 將socket_fd_設(shè)置為非阻塞模式
if (SetNonBlocking(socket_fd_) == -1) {
std::cout << "Failed to set socket to non-blocking mode" << std::endl;
return;
}
while (thread_is_running_) {
ret = recv(socket_fd_, recv_buffer, sizeof(recv_buffer), 0);
if (ret == 0) {
std::cout << "Remote socket close!!!" << std::endl;
break;
} else if (ret < 0) {
if (errno != EAGAIN && errno != EWOULDBLOCK && errno != EINTR) {
std::cout << "Remote socket error!!!" << std::endl;
break;
}
} else {
// 處理接收到的數(shù)據(jù)
std::cout << "Received data: " << recv_buffer << std::endl;
}
std::this_thread::sleep_for(std::chrono::milliseconds(10));
}
close(socket_fd_);
socket_fd_ = -1;
thread_is_running_ = false;
service_is_running_ = false;
}
int SetNonBlocking(int socket_fd) {
int flags = fcntl(socket_fd, F_GETFL, 0);
if (flags == -1) {
return -1;
}
if (fcntl(socket_fd, F_SETFL, flags | O_NONBLOCK) == -1) {
return -1;
}
return 0;
}
};
int main() {
NtripServer server;
server.Start();
// 服務(wù)運(yùn)行一段時(shí)間后停止
std::this_thread::sleep_for(std::chrono::seconds(10));
server.Stop();
return 0;
}在這個(gè)示例代碼中,我們創(chuàng)建了一個(gè)NtripServer類(lèi)來(lái)管理Ntrip服務(wù)。在Start函數(shù)中,我們創(chuàng)建了一個(gè)socket,并設(shè)置選項(xiàng)、綁定地址和端口,然后開(kāi)始監(jiān)聽(tīng)連接。接著,我們啟動(dòng)一個(gè)線(xiàn)程來(lái)處理連接,線(xiàn)程的入口函數(shù)是ThreadHandler。
在ThreadHandler函數(shù)中,我們將socket_fd_設(shè)置為非阻塞模式,并使用recv函數(shù)接收數(shù)據(jù)。根據(jù)返回值進(jìn)行處理,如果返回值為0,表示遠(yuǎn)程socket關(guān)閉;如果返回值小于0,并且errno不是EAGAIN、EWOULDBLOCK或EINTR,表示發(fā)生了其他錯(cuò)誤;如果返回值大于0,表示成功接收到數(shù)據(jù),我們可以在代碼中添加處理接收到的數(shù)據(jù)的邏輯。
在主函數(shù)中,我們創(chuàng)建了一個(gè)NtripServer對(duì)象,并調(diào)用Start函數(shù)啟動(dòng)服務(wù)。然后,讓服務(wù)運(yùn)行一段時(shí)間后調(diào)用Stop函數(shù)停止服務(wù)。
請(qǐng)注意,這只是一個(gè)簡(jiǎn)單的示例代碼,實(shí)際應(yīng)用中可能需要進(jìn)行更多的錯(cuò)誤處理和邏輯處理。對(duì)于Linux系統(tǒng),可以使用fcntl函數(shù)來(lái)設(shè)置socket為非阻塞模式。在上述代碼中,SetNonBlocking函數(shù)就是用來(lái)設(shè)置socket為非阻塞的。具體來(lái)說(shuō),它通過(guò)調(diào)用fcntl函數(shù),使用F_GETFL命令獲取socket的當(dāng)前標(biāo)志位,然后將O_NONBLOCK標(biāo)志位加入到標(biāo)志位中,再次調(diào)用fcntl函數(shù),使用F_SETFL命令將新的標(biāo)志位設(shè)置回socket。
這樣設(shè)置之后,socket的recv函數(shù)將變?yōu)榉亲枞J?,即不?huì)一直等待數(shù)據(jù)到達(dá),而是立即返回。如果沒(méi)有數(shù)據(jù)到達(dá),recv函數(shù)將返回-1,并且errno會(huì)被設(shè)置為EAGAIN或EWOULDBLOCK。這樣,我們可以在代碼中根據(jù)返回值和errno來(lái)判斷是否有數(shù)據(jù)到達(dá),并進(jìn)行相應(yīng)的處理。如果在SetNonBlocking函數(shù)中設(shè)置socket為非阻塞模式失敗,可以在調(diào)用該函數(shù)的地方進(jìn)行錯(cuò)誤處理??梢愿鶕?jù)具體情況選擇合適的錯(cuò)誤處理方式,例如打印錯(cuò)誤信息、記錄日志、關(guān)閉socket等。以下是一個(gè)示例的錯(cuò)誤處理代碼:
if (SetNonBlocking(socket_fd) == -1) {
std::cout << "Failed to set socket to non-blocking mode" << std::endl;
// 處理錯(cuò)誤,例如關(guān)閉socket
close(socket_fd);
return;
}在上述代碼中,如果SetNonBlocking函數(shù)返回-1,表示設(shè)置socket為非阻塞模式失敗。我們可以打印錯(cuò)誤信息,然后進(jìn)行相應(yīng)的錯(cuò)誤處理,例如關(guān)閉socket。這樣可以確保在設(shè)置失敗的情況下,不會(huì)繼續(xù)使用這個(gè)socket進(jìn)行后續(xù)操作。
相關(guān)文章
C語(yǔ)言簡(jiǎn)明講解隊(duì)列的實(shí)現(xiàn)方法
隊(duì)列(Queue)與棧一樣,是一種線(xiàn)性存儲(chǔ)結(jié)構(gòu),它具有如下特點(diǎn):隊(duì)列中的數(shù)據(jù)元素遵循“先進(jìn)先出”(First?In?First?Out)的原則,簡(jiǎn)稱(chēng)FIFO結(jié)構(gòu)。在隊(duì)尾添加元素,在隊(duì)頭刪除元素2022-04-04
C語(yǔ)言使用openSSL庫(kù)DES模塊實(shí)現(xiàn)加密功能詳解
這篇文章主要介紹了C語(yǔ)言使用openSSL庫(kù)DES模塊實(shí)現(xiàn)加密功能,簡(jiǎn)單講解了DES加密的相關(guān)概念,并結(jié)合實(shí)例形式分析了DES加密的具體實(shí)現(xiàn)技巧,需要的朋友可以參考下2017-05-05
使用UDP協(xié)議實(shí)現(xiàn)單詞翻譯服務(wù)器
這篇文章主要為大家詳細(xì)介紹了如何使用UDP協(xié)議實(shí)現(xiàn)英文單詞翻譯服務(wù)器,文中的示例代碼講解詳細(xì),具有一定的學(xué)習(xí)價(jià)值,感興趣的小伙伴可以了解下2023-08-08
C++11中的智能指針shared_ptr、weak_ptr源碼解析
本文是基于gcc-4.9.0的源代碼進(jìn)行分析,shared_ptr和weak_ptr是C++11才加入標(biāo)準(zhǔn)的,僅對(duì)C++智能指針shared_ptr、weak_ptr源碼進(jìn)行解析,需要讀者有一定的C++基礎(chǔ)并且對(duì)智能指針有所了解2021-09-09

