免费注册 查看新帖 |

ChinaUnix.net

  平台 论坛 博客 文库
最近访问板块 发新帖
查看: 4365 | 回复: 0

一段网络处理的代码 [复制链接]

论坛徽章:
0
发表于 2016-12-21 15:18 来自手机 |显示全部楼层
本帖最后由 001xiaofei 于 2016-12-21 15:30 编辑

#include <stdio.h>
#include <unistd.h>
#include <stdlib.h> /* malloc */
#include <string.h> /* memset */
#include <errno.h>
#include <assert.h>
#include <sys/types.h>
#include <sys/socket.h>
#include <netinet/in.h>
#include <arpa/inet.h>
#include <sys/ioctl.h>
#include <net/if.h>
#include "common.h"

int net_socket(int domain, int type, int protocol)
{
    int sockfd;
    int err;

#if defined(SOCK_NONBLOCK) && defined(SOCK_CLOEXEC)
    sockfd = socket(domain, type | SOCK_NONBLOCK | SOCK_CLOEXEC, protocol);
    if (sockfd != -1)
        return sockfd;

    if (errno != EINVAL)
        return -errno;
#endif

    sockfd = socket(domain, type, protocol);
    if (sockfd == -1) {
        MLOG_ERROR("error socket !\n");
        return -errno;
    }

    err = cm_nonblock(sockfd, 1);
    if (err < 0) {
        cm_close(sockfd);
        return err;
    }
   
    err = cm_cloexec(sockfd, 1);
    if (err < 0) {
        cm_close(sockfd);
        return err;
    }

#if defined(SO_NOSIGPIPE)
    int on = 1;
    setsockopt(sockfd, SOL_SOCKET, SO_NOSIGPIPE, (void*)&on, sizeof(on));
#endif

    return sockfd;
}


/* ifname:"eth0" */
unsigned long net_ip4_ifaddr(char *ifname)
{
        struct ifreq ifr;
        int skfd;
        struct sockaddr_in *saddr;

        if ( (skfd = socket(AF_INET, SOCK_DGRAM, 0)) < 0 ) {
                return -1;
        }

        strncpy(ifr.ifr_name, ifname, IFNAMSIZ);
        if (ioctl(skfd, SIOCGIFADDR, &ifr) < 0) {
                close(skfd);
                return -1;
        }
        close(skfd);

        saddr = (struct sockaddr_in *) &ifr.ifr_addr;
        return saddr->sin_addr.s_addr;
}

int net_ip4_addr(const char* ip, int port, struct sockaddr_in* addr)
{   
    memset(addr, 0, sizeof(struct sockaddr_in));
    addr->sin_family = AF_INET;
    addr->sin_port = htons(port);
    addr->sin_addr.s_addr = inet_addr(ip);
   
    return 0;
}

char *net_ip4_inet_ntoa(struct sockaddr_in* addr)
{
    return inet_ntoa(addr->sin_addr);
}
您需要登录后才可以回帖 登录 | 注册

本版积分规则 发表回复

基于案例的 SQL 优化实战训练营

讲师:中电福富特级专家梁敬彬,参与本次课程培训,你将收获:
1. 能编写出较为高效的 SQL;
2. 能解决70%以上的数据库常见优化问题;
3. 能得到老师提供的高效的相关工具和解决方案;
4. 能举一反三,收获不仅仅是 SQL 优化。
现在购票享受8.8折优惠!
----------------------------------------
优惠时间:2019年3月20日前

大会官网>>
  

北京盛拓优讯信息技术有限公司. 版权所有 16024965号-6 北京市公安局海淀分局网监中心备案编号:11010802020122
中国互联网协会会员  联系我们:huangweiwei@it168.com
感谢所有关心和支持过ChinaUnix的朋友们 转载本站内容请注明原作者名及出处

清除 Cookies - ChinaUnix - Archiver - WAP - TOP