netbasef.c

来自「Linux环境登陆源码」· C语言 代码 · 共 36 行

#include <sys/types.h>
#include <arpa/inet.h>
#include <netdb.h>
#include <sys/socket.h>
#include <netinet/in.h>
#include <unistd.h>
#define putinfo(str) write(STDERR_FILENO, str, strlen(str))
int socket_init(char *ipbuf, int port)
{
   int sockfd;
   struct sockaddr_in name;
   if((sockfd=socket(AF_INET, SOCK_STREAM, 0)) < 0)
      {putinfo("creat socket faild!");return(-1);}
   if(inet_addr(ipbuf) < 0) /* Bad ip address */
      name.sin_addr.s_addr = htonl(INADDR_ANY);
   else
      name.sin_addr.s_addr = inet_addr(ipbuf);
   name.sin_port = htons(port);
   name.sin_family = AF_INET;
   if(bind(sockfd, (struct sockaddr *)&name, sizeof(name)) < 0)
      {char buf[255];
       putinfo("bind socket error!");
       strcpy(buf, inet_ntoa(name.sin_addr));
       putinfo("ip:\t");
       putinfo(buf);
       return(-1);}
   return(sockfd);
}

int socket_listen(int sockfd, int queue_n)
{
   if(listen(sockfd, queue_n) < 0)
      return(-1);
   return(0);
}