iOS 使用 socket 即時通信（非第三方庫）

tianmaxk 7年前發布 | 20K 次閱讀 iOS開發 Socket 移動開發

其實寫這個socket一開始我是拒絕的。

因為大家學C 語言和linux基礎時肯定都有接觸，客戶端和服務端的通信也都了解過，加上現在很多開放的第三方庫都不需要我們來操作底層的通信。

但是來了！！！

但是！還是想寫。底層的東西最好了解下。

好了正經了！！！！

效果

效果.gif

由于5M的上傳限制GIF可能看不清我再截兩張圖吧

服務器

客戶端A

客戶端B

模型圖

做了個逗比模型圖:arrow_down:

模型

分析

由上圖可以了解到服務器和客戶端需要做哪些工作

服務器

抽象一點分為:

1.創建線程等待接收客戶端的連接
2.接收并解析客戶端發來的消息
3.給客戶端發送消息
具體一點:
1.創建socket. 綁定端口.開始監聽.
2.創建線程.等待接收客戶端連接.
3.接收客戶端發來的消息
4.解析消息內容
- a.設置用戶名
- b.發送消息給指定客戶端

客戶端

抽象一點分為:

1.連接服務器
2.給服務器發送消息
3.接收服務器消息
4.解析消息內容
具體一點:
1.創建socket.綁定端口.連接服務器
2.發送消息
- a.設置用戶名
- b.給指定用戶發消息:按服務器格式拼接字符串
3.接收消息
- a.普通消息
- b.用戶列表:保存至用戶列表

UI方面

服務器:其實不用什么UI放個控件展示下日志就是了

客戶端:比較簡單,一個俗套聊天室的界面,直接storyboard里拖拖控件設置約束了

DEMO而已別太當真

代碼部分

服務器

要使用scoket需要引用這三個頭文件

#include <netinet/in.h>
#include <sys/socket.h>
#include <arpa/inet.h>

只有一個model,用來綁定用戶名和socket

@interface ClientModel : NSObject
@property(nonatomic,assign)int clientSocket;
@property(nonatomic,copy)NSString *clientName;
@end

只有一個文件全給你

#import "ViewController.h"

#include <netinet/in.h>
#include <sys/socket.h>
#include <arpa/inet.h>
#import "ClientModel.h"
static int const kMaxConnectCount = 5;

@interface ViewController()
@property (weak) IBOutlet NSTextField *textField;
//@property (nonatomic,assign)int client_socket; //客戶端socket
@property (unsafe_unretained) IBOutlet NSTextView *textView;
    @property (nonatomic,strong)NSMutableArray *clientArray;
    @property (nonatomic,strong)NSMutableArray *clientNameArray;
@end

@implementation ViewController

- (NSMutableArray *)clientArray {
    if (!_clientArray) {
        _clientArray = [NSMutableArray array];
    }
    return _clientArray;
}
- (NSMutableArray *)clientNameArray {
    if (!_clientNameArray) {
        _clientNameArray = [NSMutableArray array];
    }
    return _clientNameArray;
}

- (void)viewDidLoad {
    [super viewDidLoad];
    // Do any additional setup after loading the view.
    //創建socket
    int server_socket = socket(AF_INET, SOCK_STREAM, 0);
    if (server_socket == -1) {
        NSLog(@"創建失敗");
        [self showLogsWithString:@"socket創建失敗"];

    }else{
        //綁定地址和端口
        struct sockaddr_in server_addr;
        server_addr.sin_len = sizeof(struct sockaddr_in);
        server_addr.sin_family = AF_INET;
        server_addr.sin_port = htons(1234);
        server_addr.sin_addr.s_addr = inet_addr("127.0.0.1");
        bzero(&(server_addr.sin_zero), 8);

        int bind_result = bind(server_socket, (struct sockaddr*)&server_addr, sizeof(server_addr));
        if (bind_result == -1) {
            NSLog(@"綁定端口失敗");
            [self showLogsWithString:@"綁定端口失敗"];

        }else{
            if (listen(server_socket, kMaxConnectCount)==-1) {
                NSLog(@"監聽失敗");
                [self showLogsWithString:@"監聽失敗"];

            }else{
                for (int i = 0; i < kMaxConnectCount; i++) {
                    //接受客戶端的鏈接
                    [self acceptClientWithServerSocket:server_socket];
                }
            }
        }
    }
}


- (void)setRepresentedObject:(id)representedObject {
    [super setRepresentedObject:representedObject];

    // Update the view, if already loaded.
}

//創建線程接受客戶端
-(void)acceptClientWithServerSocket:(int)server_socket{
    struct sockaddr_in client_address;
    socklen_t address_len;
    dispatch_queue_t queue = dispatch_get_global_queue(DISPATCH_QUEUE_PRIORITY_DEFAULT, 0);
    dispatch_async(queue, ^{
        //創建新的socket
        while (1) {
            int client_socket = accept(server_socket, (struct sockaddr*)&client_address,&address_len );
            if (client_socket == -1) {
                [self showLogsWithString:@"接受客戶端鏈接失敗"];
                NSLog(@"接受客戶端鏈接失敗");
            }else{
                NSString *acceptInfo = [NSString stringWithFormat:@"客戶端 in,socket:%d",client_socket];
                [self showLogsWithString:acceptInfo];

                //接受客戶端數據
                [self recvFromClinetWithSocket:client_socket];
            }
        }
    });
}

//接受客戶端數據
- (void)recvFromClinetWithSocket:(int)client_socket{
    while (1) {
        //接受客戶端傳來的數據
        char buf[1024] = {0};
        long iReturn = recv(client_socket, buf, 1024, 0);
        if (iReturn>0) {
            NSLog(@"客戶端來消息了");
            NSString *str = [NSString stringWithCString:buf encoding:NSUTF8StringEncoding];
            [self showLogsWithString:[NSString stringWithFormat:@"客戶端來消息了:%@",str]];
            [self checkRecvStr:str andClientSocket:client_socket];
        }else if (iReturn == -1){
            NSLog(@"讀取消息失敗");
            [self showLogsWithString:@"讀取消息失敗"];
            break;
        }else if (iReturn == 0){
            NSLog(@"客戶端走了");
            [self showLogsWithString:[NSString stringWithFormat:@"客戶端 out socket:%d",client_socket]];
            NSMutableArray *array = [NSMutableArray arrayWithArray:self.clientArray];
            for (ClientModel *model in array) {
                if (model.clientSocket == client_socket) {
                    [self.clientNameArray removeObject:model.clientName];
                    [self.clientArray removeObject:model];
                }
            }

            close(client_socket);

            break;
        }
    }
}

    //檢查接受到的字符串
- (void)checkRecvStr:(NSString*)str andClientSocket:(int)socket{
    if ([str hasPrefix:@"name:"]) {
        NSString *name = [str substringFromIndex:5];

        ClientModel *model = [[ClientModel alloc] init];
        model.clientSocket = socket;
        model.clientName = name;


        if (self.clientArray.count > 0) {
            int flag = 999;
            //用戶名不能相同
            int i = 0;

            for (ClientModel *client in self.clientArray) {

                //改名
                if (client.clientSocket == socket) {
                    NSString *oldName = self.clientNameArray[i];
                    self.clientNameArray[i] = name;
                    self.clientArray[i] = model;

                    for (ClientModel *oldclient in self.clientArray) {
                        [self sendMsg:[NSString stringWithFormat:@"%@ 改名 %@",oldName,name] toClient:oldclient.clientSocket];
                        [self showLogsWithString:[NSString stringWithFormat:@"%@ 改名 %@",oldName,name]];
                        NSString *list = [self.clientNameArray componentsJoinedByString:@","];
                        //向客戶端推送當前在線列表
                        [self sendMsg:[NSString stringWithFormat:@"list:%@",list] toClient:oldclient.clientSocket];
                    }

                    flag = 2;

                }else{
                    if ([client.clientName isEqualToString:model.clientName]) {
                        //用戶名已存在
                        flag = 1;
                        break;
                    }
                }
                i++;

            }
            if (flag != 1 & flag != 2) {
                [self.clientArray addObject:model];
                [self.clientNameArray addObject:model.clientName];
                //向客戶端推送當前在線列表
                for (ClientModel *client in self.clientArray) {
                    [self sendMsg:[NSString stringWithFormat:@"%@,上線了",name] toClient:client.clientSocket];
                    NSString *list = [self.clientNameArray componentsJoinedByString:@","];
                    //向客戶端推送當前在線列表
                    [self sendMsg:[NSString stringWithFormat:@"list:%@",list] toClient:client.clientSocket];
                }

                //給當前客戶端發送一條歡迎信息
                NSString *msg = [NSString stringWithFormat:@"Welcome %@ !",name];
                [self sendMsg:msg toClient:socket];
                [self showLogsWithString:msg];

            }else if (flag == 1){
                [self sendMsg:@"注冊用戶名失敗，用戶名已經存在，請重新設置用戶名" toClient:socket];
                [self showLogsWithString:[NSString stringWithFormat:@"socket %d 注冊用戶名失敗，設置的用戶名已經存在",socket]];

                for (ClientModel *model in self.clientArray) {

                    [name isEqualToString:model.clientName];
                }



            }
        }else{
            [self.clientArray addObject:model];
            [self.clientNameArray addObject:model.clientName];
            //向客戶端推送當前在線列表
            //給當前客戶端發送一條歡迎信息
            NSString *msg = [NSString stringWithFormat:@"Welcome %@ !",name];
            [self sendMsg:msg toClient:socket];
            [self showLogsWithString:msg];

            NSString *list = [self.clientNameArray componentsJoinedByString:@","];
            //向客戶端推送當前在線列表
            [self sendMsg:[NSString stringWithFormat:@"list:%@",list] toClient:socket];

        }

    }
    //給某人發消息
    else if  ([str hasPrefix:@"to:"]){
        NSRange nameRange = [str rangeOfString:@"*"];
        NSString *name = [str substringWithRange:NSMakeRange(3, nameRange.location-3)];
        NSString *content = [str substringFromIndex:nameRange.location+1];
        NSString *fromClientName;
        //找出發送者
        for (ClientModel *model in self.clientArray) {
            if (socket == model.clientSocket) {
                fromClientName = model.clientName;
                break;
            }
        }

        //給目標發送信息
        for (ClientModel *model in self.clientArray) {
            if ([name isEqualToString:model.clientName]) {
                NSString *msg = [NSString stringWithFormat:@"%@ to you\n%@",fromClientName,content];
                [self sendMsg:msg toClient:model.clientSocket];

                [self showLogsWithString:[NSString stringWithFormat:@"%@ 發送給 %@ 內容是：%@",fromClientName,name,content]];
                break;

            }
        }

    }
}

//給客戶端發送信息
- (void)sendMsg:(NSString*)msg toClient:(int)socket{
    char *buf[1024] = {0};
    const char *p1 = (char*)buf;
    p1 = [msg cStringUsingEncoding:NSUTF8StringEncoding];
    send(socket, p1, 1024, 0);
}

    //在界面上顯示日志
- (void)showLogsWithString:(NSString*)str {
    dispatch_async(dispatch_get_main_queue(), ^{
        NSString *newStr = [NSString stringWithFormat:@"\n%@",str];
        self.textView.string = [self.textView.string stringByAppendingString:newStr];
    });
}

@end

客戶端

由于客戶端設計的就比較簡單,所以代碼量也很少,全給你.

#import "ViewController.h"
#include <netinet/in.h>
#include <sys/socket.h>
#include <arpa/inet.h>
@interface ViewController ()<UITableViewDelegate,UITableViewDataSource>
//服務器socket
@property (nonatomic,assign)int server_socket;

//UI
@property (weak, nonatomic) IBOutlet UITextField *userNameField;
@property (weak, nonatomic) IBOutlet UITextView *chatView;
@property (weak, nonatomic) IBOutlet UITextField *msgField;
@property (weak, nonatomic) IBOutlet UILabel *toName;
@property (weak, nonatomic) IBOutlet UIView *onlineUserView;
@property (nonatomic,strong)UITableView *onlineTable;

//user列表
@property (nonatomic,strong)NSMutableArray *userArray;

@end

@implementation ViewController
- (NSMutableArray *)userArray {
    if (!_userArray) {
        _userArray = [NSMutableArray array];
    }
    return _userArray;
}
- (void)viewDidLoad {
    [super viewDidLoad];
    [self.userNameField becomeFirstResponder];
    self.userNameField.text = @"";
    self.msgField.text = @"";
    //添加table用戶列表
    self.onlineTable = [[UITableView alloc] initWithFrame:self.onlineUserView.frame style:UITableViewStylePlain];
    self.onlineTable.delegate = self;
    self.onlineTable.dataSource = self;
    [self.view addSubview:self.onlineTable];

    int server_socket = socket(AF_INET, SOCK_STREAM, 0);
    if (server_socket == -1) {
        NSLog(@"創建失敗");
    }else{
        //綁定地址和端口
        struct sockaddr_in server_addr;
        server_addr.sin_len = sizeof(struct sockaddr_in);
        server_addr.sin_family = AF_INET;
        server_addr.sin_port = htons(1234);
        server_addr.sin_addr.s_addr = inet_addr("127.0.0.1");
        bzero(&(server_addr.sin_zero), 8);

        //接受客戶端的鏈接
        dispatch_queue_t queue = dispatch_get_global_queue(DISPATCH_QUEUE_PRIORITY_DEFAULT, 0);
        dispatch_async(queue, ^{
            //創建新的socket
            int aResult = connect(server_socket, (struct sockaddr*)&server_addr, sizeof(struct sockaddr_in));
            if (aResult == -1) {
                NSLog(@"鏈接失敗");
            }else{
                self.server_socket = server_socket;
                [self acceptFromServer];
            }
        });
    }
}

//從服務端接受消息
- (void)acceptFromServer{
    while (1) {
        //接受服務器傳來的數據
        char buf[1024];
        long iReturn = recv(self.server_socket, buf, 1024, 0);
        if (iReturn>0) {
            NSString *str = [NSString stringWithCString:buf encoding:NSUTF8StringEncoding];

            //篩選前綴
            if ([str hasPrefix:@"list:"]) {
                NSString *arrayStr = [str substringFromIndex:5];
                NSArray *list = [arrayStr componentsSeparatedByString:@","];
                self.userArray = [NSMutableArray arrayWithArray:list];
                dispatch_async(dispatch_get_main_queue(), ^{
                    [self.onlineTable reloadData];
                });
                NSLog(@"當前在線用戶列表：%@",arrayStr);
            }else{
                //回到主線程 界面上顯示內容
                [self showLogsWithString:str];
            }

        }else if (iReturn == -1){
            NSLog(@"接受失敗-1");
            break;
        }
    }
}

//在界面上顯示日志
- (void)showLogsWithString:(NSString*)str {
    dispatch_async(dispatch_get_main_queue(), ^{
        NSString *newStr = [NSString stringWithFormat:@"\n%@",str];
        self.chatView.text = [self.chatView.text stringByAppendingString:newStr];
    });
}

- (void)didReceiveMemoryWarning {
    [super didReceiveMemoryWarning];
    // Dispose of any resources that can be recreated.
}

//設置用戶名
- (IBAction)clickSetUserName:(id)sender {
    NSString *msg = [NSString stringWithFormat:@"name:%@",self.userNameField.text] ;
    [self sendMsg:msg];
//    [self showLogsWithString:msg];
    [self.msgField becomeFirstResponder];
}

//發送信息
- (IBAction)clickSendMsg:(id)sender {
    if ([self.msgField.text isEqualToString:@""] || ![self.userArray containsObject:self.userNameField.text] || [self.toName.text isEqualToString:self.userNameField.text]) {
        [self showLogsWithString:@"請設置用戶名、檢查發送對象、消息不能為空"];
        return;
    }
    NSString *msg = [NSString stringWithFormat:@"to:%@*%@",self.toName.text,self.msgField.text];
    [self sendMsg:msg];
    NSString *displayMsg = [NSString stringWithFormat:@"to:%@\n%@",self.toName.text,self.msgField.text];
    [self showLogsWithString:displayMsg];
    self.msgField.text = @"";

}

//給客戶端發送信息
- (void)sendMsg:(NSString*)msg {
    char *buf[1024] = {0};
    const char *p1 = (char*)buf;
    p1 = [msg cStringUsingEncoding:NSUTF8StringEncoding];
    send(self.server_socket, p1, 1024, 0);
}

#pragma mark - TableViewDelegate & dataSource
- (NSInteger)tableView:(UITableView *)tableView numberOfRowsInSection:(NSInteger)section{
    return self.userArray.count;
}
- (UITableViewCell *)tableView:(UITableView *)tableView cellForRowAtIndexPath:(NSIndexPath *)indexPath{
    static NSString *cellId = @"onlinetableviewcellid";
    UITableViewCell *cell = [tableView dequeueReusableCellWithIdentifier:cellId];
    if (cell == nil) {
        cell = [[UITableViewCell alloc] initWithStyle:UITableViewCellStyleDefault reuseIdentifier:cellId];
    }else{
        NSLog(@"cell重用了");
    }
    cell.textLabel.text = self.userArray[indexPath.row];
    return cell;
}

//點擊cell
- (void)tableView:(UITableView *)tableView didSelectRowAtIndexPath:(NSIndexPath *)indexPath{
    self.toName.text = self.userArray[indexPath.row];
    [self.msgField becomeFirstResponder];
}
@end

本文由用戶 tianmaxk 自行上傳分享，僅供網友學習交流。所有權歸原作者，若您的權利被侵害，請聯系管理員。

轉載本站原創文章，請注明出處，并保留原始鏈接、圖片水印。

本站是一個以用戶分享為主的開源技術平臺，歡迎各類分享！

本文地址：http://www.baiduhome.net/lib/view/open1484824361920.html

iOS開發 Socket 移動開發

iOS 使用 socket 即時通信（非第三方庫）

效果

模型圖

分析

服務器

客戶端

UI方面

代碼部分

服務器

客戶端

相關經驗

相關資訊

相關文檔

目錄