//板凳——————————————————c++(99)

Linux C与C++ 一线开发p512
#include <sys/socket.h>
#include <stdio.h>
#include <arpa/inet.h>
#include <assert.h>
#include <stdlib.h>
#include <fcntl.h>
#include <sys/time.h>
#include <string.h>
#include <unistd.h>
#include <errno.h>
#include
#define BUFFER_SIZE 1023

int setnonblocking(int fd){
int old_option = fcntl(fd, F_GETFL);
int new_option = old_option | O_NONBLOCK;
fcntl(fd, F_SETFL, new_option);
return old_option;
}

int unblock_connect(const char* ip, int port, int time){
int ret = 0;
struct sockaddr_in address;
bzero(&address, sizeof(address));
address.sin_family = AF_INET;
inet_pton (AF_INET, ip, &address.sin_addr);
address.sin_port = htons(port);

int sockfd = socket(PF_INET, SOCK_STREAM, 0);
int fdopt = setnonblocking (sockfd);
ret = connect (sockfd, (struct sockaddr*)& address, sizeof(address));
printf("connect ret code = %d\n", ret);
if(ret == 0){
     printf("connect with server immediately\n");
     fcntl(sockfd, F_SETFL, fdopt);
     return sockfd;
}
else if (errno != EINPROGRESS){
     printf("ret = %d\n", ret);
     printf("unblock connect failed!\n");
     return -1;
}
else if (errno == EINPROGRESS){
     printf("unblock mode socket is connecting ... \n");
}

fd_set  readfds;
fd_set  writefds;
struct timeval timeout;

FD_ZERO(&readfds);
FD_SET(sockfd, &writefds);
timeout.tv_sec = time;
timeout.tv_usec = 0;

ret = select(sockfd + 1, NULL, &writefds, NULL, &timeout);
if(ret <= 0){
    printf("connection time out\n");
    close(sockfd);
    return -1;
}
if(! FD_ISSET(sockfd, &writefds)){
    printf("no events on sockfd found\n");
    close(sockfd);
    return -1;
}
int error = 0;
socklen_t length = sizeof(error);
if(getsockopt(sockfd, SOL_SOCKET, SO_ERROR, &error, &length) < 0){
     printf("get socket option failed\n");
     close (sockfd);
     return -1;
}
if(error != 0){
     printf("connection failed after select with the error: %d\n", error);
    close(sockfd);
    return -1;
}
printf("connection ready after select with the socket: %d\n", sockfd);
fcntl(sockfd, F_SETFL, fdopt);
return sockfd;

}

int main(int argc, char* argv[]){
int sock = socket(PF_INET, SOCK_STREAM, 0);
assert(sock >= 0);

  int old_option = fcntl(sock, F_GETFL);
  if(0 ==(old_option & O_NONBLOCK))
       printf("now socket is BLOCK mode\n");
  else
       printf("now socket is NOT BLOCK mode \n");
       
  setnonblocking(sock);
  old_option = fcntl(sock, F_GETFL);
  if(0 == (old_option & O_NONBLOCK))
       printf("now socket is BLOCK mode\n");
  else
       printf("now socket is NOT BLOCK mode \n");

Linux C与C++ 一线开发p515
if(argc <= 2){
printf(“usage: %s ip_address port_number\n”, basename(argv[0]));
return -1;
}
const char* ip = argv[1];
int port = atoi(argv[2]);
int sockfd = unblock_connect(ip, port, 1);
if(sockfd < 0){
printf(“sockfd error! return -1\n”);
return 1;
}
printf(“sent data out \n”);
send(sockfd, “abc”, 3, 0);
shutdown(sockfd, SHUT_WR);
close(sockfd);

       return 0; 

}
/*
wannian07@wannian07-PC:~$ g++ -std=c++17 -o c13 c13.cpp
wannian07@wannian07-PC:~$ ./c13
now socket is BLOCK mode
now socket is NOT BLOCK mode

wannian07@wannian07-PC:~$ ./c13 172.16.2.88 11133 kkk
connect ret code = -1
unblock mode socket is connecting …
connection time out
sockfd error! return -1
wannian07@wannian07-PC:~$ ./c13 192.168.0.11 11133 kkk
now socket is BLOCK mode
now socket is NOT BLOCK mode
connect ret code = -1
unblock mode socket is connecting …
connection failed after select with the error: 111
sockfd error! return -1

*/

  • 0
    点赞
  • 0
    收藏
    觉得还不错? 一键收藏
  • 0
    评论
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值