即時通訊下數據粘包、斷包處理實例(基于CocoaAsyncSocket)

使用CocoaAsyncSocket這個框架進行數據封包和拆包。來解決頻繁的數據發送下,導致的數據粘包、以及較大數據(例如圖片、錄音等等)的發送,導致的數據斷包。一、什么是粘包?

經常我們發現,如果用客戶端同一時間發送幾條數據,而服務端只能收到一大條數據,類似下圖:
如圖

如圖,由于傳輸的過程為數據流,經過TCP傳輸后,三條數據被合并成了一條,這就是數據粘包了。

那么為什么會造成粘包呢?

原來這是因為TCP使用了優化方法(Nagle算法)。
它將多次間隔較小且數據量小的數據,合并成一個大的數據塊,然后進行封包。
這么做優點也很明顯,就是為了減少廣域網的小分組數目,從而減小網絡擁塞的出現。
http://www.cnblogs.com/wanpengcoder/p/5366156.html nagle算法
而UDP就不會有這種情況,它不會使用塊的合并優化算法。
這里說到了就順便提一下,由于它支持的是一對多的模式,所以接收端的skbuff(套接字緩沖區)采用了鏈式結構來記錄每一個到達的UDP包,在每個UDP包中就有了消息頭(消息來源地址,端口等信息)。

當然除了優化算法,TCP和UDP都會因為下面兩種情況造成粘包:

發送端需要等緩沖區滿才發送出去,造成粘包
接收方不及時接收緩沖區的包,造成多個包接收。
二、什么是斷包?

斷包應該還是比較好理解的,比如我們發送一條很大的數據包,類似圖片和錄音等等,很顯然一次發送或者讀取數據的緩沖區大小是有限的,所以我們會分段去發送或者讀取數據。
類似下圖:

7c632e30a6c60418e0504e19781d6d29.png

無論是粘包還是斷包,如果我們要正確解析數據,那么必須要使用一種合理的機制去解包。這個機制的思路其實很簡單:

我們在封包的時候給每個數據包加一個長度或者一個開始結束標記。
然后我們拆包的時候就能區分每個數據包了,再按照長度或者分解符去分拆成各個數據包。
Talk is cheap. Show me the code
三、實例:基于CocoaAsyncSocket的封包,拆包處理。

開始動手之前,我們需要去理解下面這幾個方法
<pre>//讀取數據,有數據就會觸發代理

  • (void)readDataWithTimeout:(NSTimeInterval)timeout tag:(long)tag;
    //直到讀到這個長度的數據,才會觸發代理
  • (void)readDataToLength:(NSUInteger)length withTimeout:(NSTimeInterval)timeout tag:(long)tag;
    //直到讀到data這個邊界,才會觸發代理
  • (void)readDataToData:(NSData *)data withTimeout:(NSTimeInterval)timeout tag:(long)tag;</pre>
    之前講:iOS即時通訊,從入門到“放棄”?中提到過,這個框架每次讀取數據,必須手動的去調用上述這些read方法,而我們之前的實現思路是,第一次連接成功的代理觸發后調用:<pre>- (void)readDataWithTimeout:(NSTimeInterval)timeout tag:(long)tag;</pre>

之后每次收到消息之后,都在去調用一次這個方法,超時為-1,即不超時。這樣我們每次收到消息,都會即時觸發我們讀取消息的代理:
<pre>- (void)socket:(GCDAsyncSocket *)sock didReadData:(NSData *)data withTag:(long)tag</pre>
然而這么做顯然沒有考慮數據的拆包,如果我們一條一條的發送文字信息,自然沒什么問題。如果我們一次發送數條,或者發送大圖片。那么問題就出來了,我們解析出來的數據顯然是不對的。

這時候我們就需要另外兩個read方法了,一個是讀取到指定長度,另一個是讀取到指定邊界。
我們通過自己定義的數據邊界,去調用這兩個方法,而觸發的讀取代理,得到的數據才是正確的一個包的數據。

所以我們的核心思路有了:

封包的時候給每個包的數據加一個標記,來標明數據的長度和類型(類型顯然是需要的,我們需要知道它是文本、圖片、還是錄音等等,來用正確的方式處理這個數據)。
拆包的時候,先獲取到我們給每個包的標記,然后根據標記的數據長度,去獲取數據。最后再根據標記的類型去處理數據。(文字輸出、圖片展示、錄音播放等等)。
接著我們可以開始動手了:
這里我們首先需要一個服務端,一個客戶端。為了簡單,我們都用OC來實現。

其中我們客戶端用手機,服務端我們用Xcode模擬器。(由于Xcode只能同一時間運行一個模擬器…)

這里我們用客戶端封包發送數據,然后服務端拆包解析數據。

我們先來看看客戶端的代碼:
<pre>static NSString * Khost = @"10.10.100.48";
static const uint16_t Kport = 6969;
//建立連接

  • (BOOL)connect
    {
    return [gcdSocket connectToHost:Khost onPort:Kport error:nil];
    }</pre>
    這里需要說的是,為了連接上本機的服務端,我們這里的host為服務端的IP地址:
c89dce39acf938a5261d87ce4f60f5ab.png

端口為6969(只需和服務端accpet端口一致即可)。

注意:如果大家要運行github上的demo,只需修改這個host地址即可,把它改成你電腦(服務端)的IP地址。

接著我們來看看write方法,我們在該方法中進行封包:
<pre>//發送消息

  • (void)sendMsg
    {

    NSData *data = [@"你好" dataUsingEncoding:NSUTF8StringEncoding];
    NSData *data1 = [@"豬頭" dataUsingEncoding:NSUTF8StringEncoding];
    NSData *data2 = [@"先生" dataUsingEncoding:NSUTF8StringEncoding];

NSData *data3  = [@"今天天氣好" dataUsingEncoding:NSUTF8StringEncoding];
NSData *data4  = [@"吃飯了嗎" dataUsingEncoding:NSUTF8StringEncoding];

[self sendData:data :@"txt"];
[self sendData:data1 :@"txt"];
[self sendData:data2 :@"txt"];
[self sendData:data3 :@"txt"];
[self sendData:data4 :@"txt"];

NSString *filePath = [[NSBundle mainBundle]pathForResource:@"test1" ofType:@"jpg"];

NSData *data5 = [NSData dataWithContentsOfFile:filePath];

[self sendData:data5 :@"img"];

}

  • (void)sendData:(NSData *)data :(NSString *)type
    {
    NSUInteger size = data.length;

    NSMutableDictionary *headDic = [NSMutableDictionary dictionary];
    [headDic setObject:type forKey:@"type"];
    [headDic setObject:[NSString stringWithFormat:@"%ld",size] forKey:@"size"];
    NSString *jsonStr = [self dictionaryToJson:headDic];

NSData *lengthData = [jsonStr dataUsingEncoding:NSUTF8StringEncoding];


NSMutableData *mData = [NSMutableData dataWithData:lengthData];
//分界
[mData appendData:[GCDAsyncSocket CRLFData]];

[mData appendData:data];


//第二個參數,請求超時時間
[gcdSocket writeData:mData withTimeout:-1 tag:110];

}

  • (NSString *)dictionaryToJson:(NSDictionary *)dic
    {
    NSError *error = nil;
    NSData *jsonData = [NSJSONSerialization dataWithJSONObject:dic options:NSJSONWritingPrettyPrinted error:&error];
    return [[NSString alloc] initWithData:jsonData encoding:NSUTF8StringEncoding];
    }</pre>
    總共上述兩個方法,也很簡單,我們發送了6條數據,前5條為文本形式,最后一條是一個20多M的圖片。當我們點擊發送的時候會觸發這個方法,這6條數據會被同時發出。

這里我們來看看我們是如何封包的:

我們定義了一個headDic,這個是我們數據包的頭部,里面裝了這個數據包的大小和類型信息(當然,你可以裝更多的其他標識信息。)然后我們把它轉成了json,最后轉成data。
然后我們把這個head拼在最前面,接著拼了一個:
[GCDAsyncSocket CRLFData]

這個是什么呢?其實它就是一個\r\n。我們用它來做頭部的邊界。(又或者我們可以規定一個固定的頭部長度,來作為邊界,這里僅僅是提供給大家一個思路)。
最后我們把真正的數據包給拼接上。
注:如果你想的更遠的話,甚至可以在結尾,再拼一個包結束的標識符,后面我們會講到為什么可以這么做。這里暫時先這樣。

就這樣,我們完成了數據的封包和發送。

客戶端有了,接著我們來看看服務端是如何來拆包的:

首先我們需要監聽本機6969端口。
<pre>static const uint16_t Kport = 6969;

//等待連接

  • (BOOL)accept
    {
    NSError *error = nil;

    BOOL isSuccess = [gcdSocket acceptOnPort:Kport error:&error];
    if (isSuccess) {
    NSLog(@"監聽成功6969端口成功,等待連接");
    return YES;
    }else{
    NSLog(@"監聽失敗,原因:%@",error);
    return NO;
    }
    }</pre>
    當客戶端連接上來后,調用成功接收到客戶端連接的代理方法:
    <pre>- (void)socket:(GCDAsyncSocket *)sock didAcceptNewSocket:(GCDAsyncSocket *)newSocket
    {
    NSLog(@"接受到socket連接");

    [_sockets addObject:newSocket];
    [newSocket readDataToData:[GCDAsyncSocket CRLFData] withTimeout:-1 tag:110];
    }</pre>
    這里需要注意的是,成功接收到連接后,調用代理我們必須把新生成的這個newSocket保存起來,如果它被銷毀了,那么連接就斷開了,這里我們把它放到了一個數組中去了。
    這里需要注意的是,成功連接后,我們就調用了:
    <pre>[newSocket readDataToData:[GCDAsyncSocket CRLFData] withTimeout:-1 tag:110];</pre>
    還記得我們封包的時候,數據包頭部之后拼了這么一個分解符data。這樣,當有數據包傳輸過來我們就能獲取到這個數據包的頭部(后面的信息先不讀取)。

接著我們來看看服務端的read代理方法是如何拆包的:
<pre>- (void)socket:(GCDAsyncSocket *)sock didReadData:(NSData *)data withTag:(long)tag
{
//先讀取到當前數據包頭部信息
if (!currentPacketHead) {
currentPacketHead = [NSJSONSerialization
JSONObjectWithData:data
options:NSJSONReadingMutableContainers
error:nil];

    NSUInteger packetLength = [currentPacketHead[@"size"] integerValue];
    //讀到數據包的大小
    [sock readDataToLength:packetLength withTimeout:-1 tag:110];

    return;
}
if (!currentPacketHead) {
    NSLog(@"error:當前數據包的頭為空");
    //斷開連接
    [self disConnect];
    return;
}

//正式的包處理
NSUInteger packetLength = [currentPacketHead[@"size"] integerValue];
//說明數據有問題
if (packetLength <= 0 || data.length != packetLength) {
    NSLog(@"error:當前數據包數據大小不正確");
    [self disConnect];
    return;
}

NSString *type = currentPacketHead[@"type"];

if ([type isEqualToString:@"img"]) {
    NSLog(@"圖片設置成功");
    self.recvImg.image = [UIImage imageWithData:data];
}else{

    NSString *msg = [[NSString alloc]initWithData:data encoding:NSUTF8StringEncoding];
    NSLog(@"收到消息:%@",msg);
}

currentPacketHead = nil;

[sock readDataToData:[GCDAsyncSocket CRLFData] withTimeout:-1 tag:110];

}
</pre>

這個方法也很簡單,我們判斷,如果currentPacketHead(當前數據包的頭部)為空,則說明這次讀取,是一個頭部信息,我們去獲取到該數據包的頭部信息。并且調用下一次讀取,讀取長度為從頭部信息中取出來的數據包長度:
<pre>
[sock readDataToLength:packetLength withTimeout:-1 tag:110];</pre>
這樣當GCDAsyncSocket中數據緩沖區長度達到我們需要讀取的length就能觸發代理方法的第二次回調。(具體原理實現會在樓主的GCDAsyncSocket解析的后續系列Read篇中去講,敬請期待)。
這時候因為currentPacketHead不為空,所以我們就知道是去獲取一個數據包,我們從頭部信息中拿到數據包的類型,如果是文本或者圖片,則分別輸出或展示到屏幕上。讀取完成后我們再次調用:
<pre>[sock readDataToData:[GCDAsyncSocket CRLFData] withTimeout:-1 tag:110];</pre>
接著我們來講講我們之前所說的為什么可以在數據包之后加一個結束標識符。我們數據很可能在傳輸的過程中,丟失了一部分,或者頭部信息不可讀,導致我們無法正常讀取這個數據包。
可能我們會有一個應用場景,當出現錯誤包的時候,我們就直接拋棄掉它,直接開始下一個數據包的讀取(當然現實中,我們往往是需要重新發送,這里僅僅是舉一個應用場景)。這樣這個結束標識符就起作用了,我們可以直接把數據讀取到這個錯誤包的結束標識處,不做任何處理,這樣相當于丟棄掉這個錯誤包了。

最后編輯于
?著作權歸作者所有,轉載或內容合作請聯系作者
平臺聲明:文章內容(如有圖片或視頻亦包括在內)由作者上傳并發布,文章內容僅代表作者本人觀點,簡書系信息發布平臺,僅提供信息存儲服務。
  • 序言:七十年代末,一起剝皮案震驚了整個濱河市,隨后出現的幾起案子,更是在濱河造成了極大的恐慌,老刑警劉巖,帶你破解...
    沈念sama閱讀 227,797評論 6 531
  • 序言:濱河連續發生了三起死亡事件,死亡現場離奇詭異,居然都是意外死亡,警方通過查閱死者的電腦和手機,發現死者居然都...
    沈念sama閱讀 98,179評論 3 414
  • 文/潘曉璐 我一進店門,熙熙樓的掌柜王于貴愁眉苦臉地迎上來,“玉大人,你說我怎么就攤上這事。” “怎么了?”我有些...
    開封第一講書人閱讀 175,628評論 0 373
  • 文/不壞的土叔 我叫張陵,是天一觀的道長。 經常有香客問我,道長,這世上最難降的妖魔是什么? 我笑而不...
    開封第一講書人閱讀 62,642評論 1 309
  • 正文 為了忘掉前任,我火速辦了婚禮,結果婚禮上,老公的妹妹穿的比我還像新娘。我一直安慰自己,他們只是感情好,可當我...
    茶點故事閱讀 71,444評論 6 405
  • 文/花漫 我一把揭開白布。 她就那樣靜靜地躺著,像睡著了一般。 火紅的嫁衣襯著肌膚如雪。 梳的紋絲不亂的頭發上,一...
    開封第一講書人閱讀 54,948評論 1 321
  • 那天,我揣著相機與錄音,去河邊找鬼。 笑死,一個胖子當著我的面吹牛,可吹牛的內容都是我干的。 我是一名探鬼主播,決...
    沈念sama閱讀 43,040評論 3 440
  • 文/蒼蘭香墨 我猛地睜開眼,長吁一口氣:“原來是場噩夢啊……” “哼!你這毒婦竟也來了?” 一聲冷哼從身側響起,我...
    開封第一講書人閱讀 42,185評論 0 287
  • 序言:老撾萬榮一對情侶失蹤,失蹤者是張志新(化名)和其女友劉穎,沒想到半個月后,有當地人在樹林里發現了一具尸體,經...
    沈念sama閱讀 48,717評論 1 333
  • 正文 獨居荒郊野嶺守林人離奇死亡,尸身上長有42處帶血的膿包…… 初始之章·張勛 以下內容為張勛視角 年9月15日...
    茶點故事閱讀 40,602評論 3 354
  • 正文 我和宋清朗相戀三年,在試婚紗的時候發現自己被綠了。 大學時的朋友給我發了我未婚夫和他白月光在一起吃飯的照片。...
    茶點故事閱讀 42,794評論 1 369
  • 序言:一個原本活蹦亂跳的男人離奇死亡,死狀恐怖,靈堂內的尸體忽然破棺而出,到底是詐尸還是另有隱情,我是刑警寧澤,帶...
    沈念sama閱讀 38,316評論 5 358
  • 正文 年R本政府宣布,位于F島的核電站,受9級特大地震影響,放射性物質發生泄漏。R本人自食惡果不足惜,卻給世界環境...
    茶點故事閱讀 44,045評論 3 347
  • 文/蒙蒙 一、第九天 我趴在偏房一處隱蔽的房頂上張望。 院中可真熱鬧,春花似錦、人聲如沸。這莊子的主人今日做“春日...
    開封第一講書人閱讀 34,418評論 0 26
  • 文/蒼蘭香墨 我抬頭看了看天上的太陽。三九已至,卻和暖如春,著一層夾襖步出監牢的瞬間,已是汗流浹背。 一陣腳步聲響...
    開封第一講書人閱讀 35,671評論 1 281
  • 我被黑心中介騙來泰國打工, 沒想到剛下飛機就差點兒被人妖公主榨干…… 1. 我叫王不留,地道東北人。 一個月前我還...
    沈念sama閱讀 51,414評論 3 390
  • 正文 我出身青樓,卻偏偏與公主長得像,于是被迫代替她去往敵國和親。 傳聞我的和親對象是個殘疾皇子,可洞房花燭夜當晚...
    茶點故事閱讀 47,750評論 2 370

推薦閱讀更多精彩內容