linux_im/utils.cpp at master · superlikex/linux_im · GitHub

1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
#include "utils.h"
#include "errno.h"
#include <cstdio>
/*
*params: ip,port
*return: socket fd
*/
int Utils::create_socket(char const *ip,int port)
{
    int sfd;
    struct sockaddr_in servaddr;
    sfd = socket(AF_INET,SOCK_STREAM,0);
    if(sfd < 0)
    {
        perror("socket");
	printf("------\n");
        return -1;
    }

    bzero(&servaddr,sizeof(servaddr));

    servaddr.sin_family = AF_INET;
    servaddr.sin_port = htons(port);
    if(!ip)
    {
        servaddr.sin_addr.s_addr = htonl(INADDR_ANY);
        if ( bind(sfd,(struct sockaddr*)&servaddr,sizeof(servaddr)) < 0 )
        {
        perror("socket");
	    printf("------\n");
            return -1;
        }
    }
    else
    {
        inet_pton(AF_INET, ip,  &servaddr.sin_addr);
        if( connect(sfd,(struct sockaddr*)&servaddr,sizeof(servaddr)) <0 )
        {
            return -1;
        }
    }
    return sfd;
}