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
*/