Socket/TCP 原理這里就不闡述了,網上一搜一大堆,直接上關鍵代碼。
【注】iOS 目前有非常著名的第三方庫 CocoaAsyncSocket 可以使用,但是我們項目當時做大數據上報要求直接發送 NSString 格式數據,所以自己寫了一個簡易版 TCP 連接,可以實現發送和接收數據,經過時間考驗,不存在閃退情況。
- 1、TCPClient.h 代碼
#import <Foundation/Foundation.h>
@interface TCPClient : NSObject
@property (nonatomic, assign) int clientSocket;
@property (nonatomic, assign) int result;
+ (TCPClient *)shareTCPClient;
//建立連接
- (BOOL)connection:(NSString *)hostText port:(int)port;
//發送字符串數據
- (void)sendStringToServerAndReceived:(NSString *)message;
//斷開連接
- (void)disConnection;
@end
- 2、TCPClient.m 代碼
#import "TCPClient.h"
#import <sys/socket.h>
#import <netinet/in.h>
#import <arpa/inet.h>
#define TCPHOSTADDNUM **ip地址**
@implementation TCPClient
static TCPClient * tcpClient = nil;
+ (TCPClient *)shareTCPClient {
@synchronized(self) {
if (!tcpClient) {
tcpClient = [[TCPClient alloc]init];
}
}
return tcpClient;
}
- 2.1、通過域名建立連接,hostText:域名地址 port:端口號
- (BOOL)connection:(NSString *)hostText port:(int)port {
/**
socket 參數
domain: 協議域,AF_INET(IPV4的網絡開發)
type: Socket 類型,SOCK_STREAM(TCP)/SOCK_DGRAM(UDP,報文)
protocol:IPPROTO_TCP,協議,如果輸入0,可以根據第二個參數,自動選擇協議
return: if > 0 就表示成功
*/
self.clientSocket = - 1;
self.clientSocket = socket(AF_INET, SOCK_STREAM, 0);
if (self.clientSocket > 0) {
NSLog(@"socket 連接成功: %d", self.clientSocket);
} else {
NSLog(@"socket 連接失敗");
return NO;
}
//通過域名獲取Ip地址
NSString * tcpIp = [self obtainTCPIpAddressWithHost:hostText];
//Connect
struct sockaddr_in serverAddress;
serverAddress.sin_family = AF_INET;
serverAddress.sin_addr.s_addr = inet_addr(tcpIp.UTF8String);
serverAddress.sin_port = htons(port);
self.result = connect(self.clientSocket, (const struct sockaddr *)&serverAddress, sizeof(serverAddress));
if (self.clientSocket > 0 && self.result >= 0) {
NSLog(@"connect 連接成功");
return YES;
}else {
NSLog(@"connect 連接失敗");
[[TCPClient shareTCPClient] disConnection];
return NO;
}
}
- 2.2、 通過域名獲取 IP 地址
- (NSString *)obtainTCPIpAddressWithHost:(NSString *)hostAdd {
NSString * tcpIpStr;
struct hostent * host_entry = gethostbyname([hostAdd UTF8String]);
char IPStr[64] = {0};
if(host_entry != 0) {
sprintf(IPStr, "%d.%d.%d.%d",
(host_entry->h_addr_list[0][0]&0x00ff),
(host_entry->h_addr_list[0][1]&0x00ff),
(host_entry->h_addr_list[0][2]&0x00ff),
(host_entry->h_addr_list[0][3]&0x00ff));
char * ip = inet_ntoa(*((struct in_addr *)host_entry->h_addr));
tcpIpStr = [NSString stringWithFormat:@"%s", ip];
NSLog(@"通過域名得到:%@", tcpIpStr);
}else {
tcpIpStr = TCPHOSTADDNUM;
NSLog(@"通過IP得到:%@", tcpIpStr);
}
return tcpIpStr;
}
- 2.3、發送 NSString 數據
//發送和接收字符串
- (void)sendStringToServerAndReceived:(NSString *)message {
if (self.clientSocket > 0 && self.result >= 0) {
//不加下面的代碼,如果在發送數據的途中服務器斷開連接,會閃退。
sigset_t set;
sigemptyset(&set);
sigaddset(&set, SIGPIPE);
sigprocmask(SIG_BLOCK, &set, NULL);
ssize_t sendLen = send(self.clientSocket, message.UTF8String, strlen(message.UTF8String), 0);
NSLog(@"發送的TCP數據長度 == %ld", sendLen);
if (sendLen > 0) {
[self performSelectorInBackground:@selector(readStream) withObject:nil];
}
}else {
//發送的時候如果連接失敗,重新連接。
}
}
- 2.4、接收數據
//接收數據
- (void)readStream {
/**
第一個int:創建的socket
void *: 接收內容的地址
size_t: 接收內容的長度
第二個int:接收數據的標記 0,就是阻塞式,一直等待服務器的數據
return: 接收到的數據長度
*/
char readBuffer[1024] = {0};
long OrgBr = 0;
OrgBr = recv(self.clientSocket, readBuffer, sizeof(readBuffer), 0) < sizeof(readBuffer);
NSLog(@"\nbr = %ld\nReceived Data:%s\n", OrgBr, readBuffer);
memset(readBuffer, 0, sizeof(readBuffer));
NSString * readString = [NSString stringWithUTF8String:readBuffer];
if (readString && ![readString isKindOfClass:[NSNull class]] && readString.length > 0) {
//接收到的數據 NSString 可以自己做相關的操作
}else {
//重新連接
}
}
- 2.5、 斷開 Socket 連接
//斷開連接
- (void)disConnection {
if (self.clientSocket > 0) {
close(self.clientSocket);
self.clientSocket = -1;
}
}
GitHub地址
原創文章,轉載請注明出處。