iOS 建立socket连接

rtmp协议,是基于tcp的协议,首先我们需要建立起tcp连接.socket是对tcp协议的封装,因此我们用socket来建立起连接.幸运的是,苹果给我们提供了简单的接口,方便我们使用.
代码如下:

//.h文件
 #import <Foundation/Foundation.h>

typedef NSStreamEvent SGStreamStatus;

@class SGStreamSession;
@protocol SGStreamSessionDelegate <NSObject>

- (void)streamSession:(SGStreamSession *)session didChangeStatus:(SGStreamStatus)streamStatus;

@end

@interface SGStreamSession : NSObject

@property (nonatomic,assign,readonly) SGStreamStatus streamStatus;

@property (nonatomic,weak) id<SGStreamSessionDelegate> delegate;
//连接到服务器
- (void)connectToServer:(NSString *)host port:(UInt32)port;
//断开连接
- (void)disConnect;
//读取数据
- (NSData *)readData;
//写数据
- (NSInteger)writeData:(NSData *)data;

@end

//.m文件
 #import "SGStreamSession.h"
 #define SGLog NSLog

@interface SGStreamSession()<NSStreamDelegate>
{
    NSInputStream  *_inputStream;
    NSOutputStream *_outputStream;
}
@end


@implementation SGStreamSession
- (instancetype)init{
    if (self = [super init]) {
        _streamStatus = NSStreamEventNone;
    }
    return self;
}
- (void)connectToServer:(NSString *)host port:(UInt32)port;{
    if (self.streamStatus > 0) {
        [self close];
    }
    
    //输入流,读取数据
    CFReadStreamRef  readStream;
    //输出流,负责写数据
    CFWriteStreamRef writeStream;
    
   //rtmp协议默认端口为1935
    if (port <= 0) {
        port = 1935;
    }
    //建立连接
    CFStreamCreatePairWithSocketToHost(NULL, (__bridge CFStringRef)host, port, &readStream, &writeStream);
    //注意_bridge_transfer,转移内存管理所有权
    _inputStream           = (__bridge_transfer  NSInputStream  *)readStream;
    _outputStream          = (__bridge_transfer  NSOutputStream *)writeStream;

    _outputStream.delegate = self;
    _inputStream.delegate  = self;

    [_outputStream scheduleInRunLoop:[NSRunLoop mainRunLoop] forMode:NSDefaultRunLoopMode];
    [_inputStream  scheduleInRunLoop:[NSRunLoop mainRunLoop] forMode:NSDefaultRunLoopMode];

    //打开输入流和输出流
    [_inputStream open];
    [_outputStream open];
}
#pragma mark -------delegate---------
- (void)stream:(NSStream *)aStream handleEvent:(NSStreamEvent)eventCode {
    
    switch (eventCode) {
        case NSStreamEventNone: {
            return;//此处是return
        }
        case NSStreamEventOpenCompleted: {
            if (_inputStream == aStream ) {
                SGLog(@"连接成功");
                _streamStatus = NSStreamEventOpenCompleted;
            }
            break;
        }
        case NSStreamEventHasBytesAvailable: {
            SGLog(@"有字节可读");
            _streamStatus |= NSStreamEventHasBytesAvailable;
            break;
        }
        case NSStreamEventHasSpaceAvailable: {
            SGLog(@"可以发送字节");
            _streamStatus |= NSStreamEventHasSpaceAvailable;
            break;
        }
        case NSStreamEventErrorOccurred: {
            SGLog(@"连接出现错误");
            _streamStatus = NSStreamEventErrorOccurred;
            break;
        }
        case NSStreamEventEndEncountered: {
            SGLog(@"连接结束");
            _streamStatus = NSStreamEventEndEncountered;
            break;
        }
    }
    
    if ([self.delegate respondsToSelector:@selector(streamSession:didChangeStatus:)]) {
        [self.delegate streamSession:self didChangeStatus:_streamStatus];
    }
}

- (void)disConnect{
    [self close];
}

- (void)close{
    
    [_inputStream  close];
    [_outputStream close];
    [_inputStream  removeFromRunLoop:[NSRunLoop currentRunLoop] forMode:NSDefaultRunLoopMode];
    [_outputStream removeFromRunLoop:[NSRunLoop currentRunLoop] forMode:NSDefaultRunLoopMode];
    
    _streamStatus = NSStreamEventNone;
    _outputStream = nil;
    _inputStream  = nil;
    _inputStream.delegate = nil;
    _outputStream.delegate = nil;
}

- (void)dealloc{
    NSLog(@"%s",__func__);
    [self close];
}

- (NSData *)readData{
    
    uint8_t buff[4096];//最大支持4K
    NSUInteger len = [_inputStream read:buff maxLength:sizeof(buff)];
    
   
    NSData *data = nil;
    
    if(len < sizeof(buff) && (_streamStatus & NSStreamEventHasBytesAvailable)){
        _streamStatus ^= NSStreamEventHasBytesAvailable;
        data = [NSData dataWithBytes:buff length:len];
    }
    
    return data;
}

- (NSInteger)writeData:(NSData *)data{
    if (data.length == 0) {
        return 0;
    }
    
    NSInteger ret = 0;
    if (_outputStream.hasSpaceAvailable) {
        ret = [_outputStream write:data.bytes maxLength:data.length];
    }
    
    if (ret > 0 && (_streamStatus & NSStreamEventHasSpaceAvailable)) {
        //移除标志位
        _streamStatus ^= NSStreamEventHasSpaceAvailable;
    }
    
    return ret;
}
@end


注释已经写得很详细了.这里提醒一下,里面的状态处理如果看不懂,不要纠结,|= 和 ^= 可以认为是 = .如果想深入了解,可以看看位枚举.

最后编辑于
©著作权归作者所有,转载或内容合作请联系作者
  • 序言:七十年代末,一起剥皮案震惊了整个滨河市,随后出现的几起案子,更是在滨河造成了极大的恐慌,老刑警刘岩,带你破解...
    沈念sama阅读 205,033评论 6 478
  • 序言:滨河连续发生了三起死亡事件,死亡现场离奇诡异,居然都是意外死亡,警方通过查阅死者的电脑和手机,发现死者居然都...
    沈念sama阅读 87,725评论 2 381
  • 文/潘晓璐 我一进店门,熙熙楼的掌柜王于贵愁眉苦脸地迎上来,“玉大人,你说我怎么就摊上这事。” “怎么了?”我有些...
    开封第一讲书人阅读 151,473评论 0 338
  • 文/不坏的土叔 我叫张陵,是天一观的道长。 经常有香客问我,道长,这世上最难降的妖魔是什么? 我笑而不...
    开封第一讲书人阅读 54,846评论 1 277
  • 正文 为了忘掉前任,我火速办了婚礼,结果婚礼上,老公的妹妹穿的比我还像新娘。我一直安慰自己,他们只是感情好,可当我...
    茶点故事阅读 63,848评论 5 368
  • 文/花漫 我一把揭开白布。 她就那样静静地躺着,像睡着了一般。 火红的嫁衣衬着肌肤如雪。 梳的纹丝不乱的头发上,一...
    开封第一讲书人阅读 48,691评论 1 282
  • 那天,我揣着相机与录音,去河边找鬼。 笑死,一个胖子当着我的面吹牛,可吹牛的内容都是我干的。 我是一名探鬼主播,决...
    沈念sama阅读 38,053评论 3 399
  • 文/苍兰香墨 我猛地睁开眼,长吁一口气:“原来是场噩梦啊……” “哼!你这毒妇竟也来了?” 一声冷哼从身侧响起,我...
    开封第一讲书人阅读 36,700评论 0 258
  • 序言:老挝万荣一对情侣失踪,失踪者是张志新(化名)和其女友刘颖,没想到半个月后,有当地人在树林里发现了一具尸体,经...
    沈念sama阅读 42,856评论 1 300
  • 正文 独居荒郊野岭守林人离奇死亡,尸身上长有42处带血的脓包…… 初始之章·张勋 以下内容为张勋视角 年9月15日...
    茶点故事阅读 35,676评论 2 323
  • 正文 我和宋清朗相恋三年,在试婚纱的时候发现自己被绿了。 大学时的朋友给我发了我未婚夫和他白月光在一起吃饭的照片。...
    茶点故事阅读 37,787评论 1 333
  • 序言:一个原本活蹦乱跳的男人离奇死亡,死状恐怖,灵堂内的尸体忽然破棺而出,到底是诈尸还是另有隐情,我是刑警宁泽,带...
    沈念sama阅读 33,430评论 4 321
  • 正文 年R本政府宣布,位于F岛的核电站,受9级特大地震影响,放射性物质发生泄漏。R本人自食恶果不足惜,却给世界环境...
    茶点故事阅读 39,034评论 3 307
  • 文/蒙蒙 一、第九天 我趴在偏房一处隐蔽的房顶上张望。 院中可真热闹,春花似锦、人声如沸。这庄子的主人今日做“春日...
    开封第一讲书人阅读 29,990评论 0 19
  • 文/苍兰香墨 我抬头看了看天上的太阳。三九已至,却和暖如春,着一层夹袄步出监牢的瞬间,已是汗流浃背。 一阵脚步声响...
    开封第一讲书人阅读 31,218评论 1 260
  • 我被黑心中介骗来泰国打工, 没想到刚下飞机就差点儿被人妖公主榨干…… 1. 我叫王不留,地道东北人。 一个月前我还...
    沈念sama阅读 45,174评论 2 352
  • 正文 我出身青楼,却偏偏与公主长得像,于是被迫代替她去往敌国和亲。 传闻我的和亲对象是个残疾皇子,可洞房花烛夜当晚...
    茶点故事阅读 42,526评论 2 343

推荐阅读更多精彩内容

  • 一、网络各个协议:TCP/IP、SOCKET、HTTP等 网络七层由下往上分别为物理层、数据链路层、网络层、传输层...
    杯水救车薪阅读 2,258评论 0 17
  • 1.项目经验 2.基础问题 3.指南认识 4.解决思路 ios开发三大块: 1.Oc基础 2.CocoaTouch...
    阳光的大男孩儿阅读 4,969评论 0 13
  • 第一部分、概念的理解1、什么是Socket?Socket又称之为“套接字”,是系统提供的用于网络通信的方法。它的实...
    Hevin_Chen阅读 2,436评论 0 5
  • 1.1 TCP/IP协议组 TCP/IP协议(传输控制协议)由网络层的IP协议和传输层的TCP协议组成 IP层负责...
    F麦子阅读 2,781评论 0 25
  • 我个人的读书历史:我读了大概几百本书,虽然不算很多,但也勉强可以指导入门级的了,使他们少走些弯路,我的第...
    张洁小确幸阅读 433评论 2 3