Imported Upstream version 7.53.1
[platform/upstream/curl.git] / docs / examples / externalsocket.c
1 /***************************************************************************
2  *                                  _   _ ____  _
3  *  Project                     ___| | | |  _ \| |
4  *                             / __| | | | |_) | |
5  *                            | (__| |_| |  _ <| |___
6  *                             \___|\___/|_| \_\_____|
7  *
8  * Copyright (C) 1998 - 2016, Daniel Stenberg, <daniel@haxx.se>, et al.
9  *
10  * This software is licensed as described in the file COPYING, which
11  * you should have received as part of this distribution. The terms
12  * are also available at https://curl.haxx.se/docs/copyright.html.
13  *
14  * You may opt to use, copy, modify, merge, publish, distribute and/or sell
15  * copies of the Software, and permit persons to whom the Software is
16  * furnished to do so, under the terms of the COPYING file.
17  *
18  * This software is distributed on an "AS IS" basis, WITHOUT WARRANTY OF ANY
19  * KIND, either express or implied.
20  *
21  ***************************************************************************/
22 /* <DESC>
23  * An example demonstrating how an application can pass in a custom
24  * socket to libcurl to use. This example also handles the connect itself.
25  * </DESC>
26  */
27 #include <stdio.h>
28 #include <string.h>
29 #include <stdlib.h>
30 #include <curl/curl.h>
31
32 #ifdef WIN32
33 #include <windows.h>
34 #include <winsock2.h>
35 #include <ws2tcpip.h>
36 #define close closesocket
37 #else
38 #include <sys/types.h>        /*  socket types              */
39 #include <sys/socket.h>       /*  socket definitions        */
40 #include <netinet/in.h>
41 #include <arpa/inet.h>        /*  inet (3) funtions         */
42 #include <unistd.h>           /*  misc. Unix functions      */
43 #endif
44
45 #include <errno.h>
46
47 /* The IP address and port number to connect to */
48 #define IPADDR "127.0.0.1"
49 #define PORTNUM 80
50
51 #ifndef INADDR_NONE
52 #define INADDR_NONE 0xffffffff
53 #endif
54
55 static size_t write_data(void *ptr, size_t size, size_t nmemb, void *stream)
56 {
57   size_t written = fwrite(ptr, size, nmemb, (FILE *)stream);
58   return written;
59 }
60
61 static curl_socket_t opensocket(void *clientp,
62                                 curlsocktype purpose,
63                                 struct curl_sockaddr *address)
64 {
65   curl_socket_t sockfd;
66   (void)purpose;
67   (void)address;
68   sockfd = *(curl_socket_t *)clientp;
69   /* the actual externally set socket is passed in via the OPENSOCKETDATA
70      option */
71   return sockfd;
72 }
73
74 static int sockopt_callback(void *clientp, curl_socket_t curlfd,
75                             curlsocktype purpose)
76 {
77   (void)clientp;
78   (void)curlfd;
79   (void)purpose;
80   /* This return code was added in libcurl 7.21.5 */
81   return CURL_SOCKOPT_ALREADY_CONNECTED;
82 }
83
84 int main(void)
85 {
86   CURL *curl;
87   CURLcode res;
88   struct sockaddr_in servaddr;  /*  socket address structure  */
89   curl_socket_t sockfd;
90
91 #ifdef WIN32
92   WSADATA wsaData;
93   int initwsa = WSAStartup(MAKEWORD(2, 0), &wsaData);
94   if(initwsa != 0) {
95     printf("WSAStartup failed: %d\n", initwsa);
96     return 1;
97   }
98 #endif
99
100   curl = curl_easy_init();
101   if(curl) {
102     /*
103      * Note that libcurl will internally think that you connect to the host
104      * and port that you specify in the URL option.
105      */
106     curl_easy_setopt(curl, CURLOPT_URL, "http://99.99.99.99:9999");
107
108     /* Create the socket "manually" */
109     sockfd = socket(AF_INET, SOCK_STREAM, 0);
110     if(sockfd == CURL_SOCKET_BAD) {
111       printf("Error creating listening socket.\n");
112       return 3;
113     }
114
115     memset(&servaddr, 0, sizeof(servaddr));
116     servaddr.sin_family = AF_INET;
117     servaddr.sin_port   = htons(PORTNUM);
118
119     servaddr.sin_addr.s_addr = inet_addr(IPADDR);
120     if(INADDR_NONE == servaddr.sin_addr.s_addr)
121       return 2;
122
123     if(connect(sockfd, (struct sockaddr *) &servaddr, sizeof(servaddr)) ==
124        -1) {
125       close(sockfd);
126       printf("client error: connect: %s\n", strerror(errno));
127       return 1;
128     }
129
130     /* no progress meter please */
131     curl_easy_setopt(curl, CURLOPT_NOPROGRESS, 1L);
132
133     /* send all data to this function  */
134     curl_easy_setopt(curl, CURLOPT_WRITEFUNCTION, write_data);
135
136     /* call this function to get a socket */
137     curl_easy_setopt(curl, CURLOPT_OPENSOCKETFUNCTION, opensocket);
138     curl_easy_setopt(curl, CURLOPT_OPENSOCKETDATA, &sockfd);
139
140     /* call this function to set options for the socket */
141     curl_easy_setopt(curl, CURLOPT_SOCKOPTFUNCTION, sockopt_callback);
142
143     curl_easy_setopt(curl, CURLOPT_VERBOSE, 1);
144
145     res = curl_easy_perform(curl);
146
147     curl_easy_cleanup(curl);
148
149     if(res) {
150       printf("libcurl error: %d\n", res);
151       return 4;
152     }
153   }
154   return 0;
155 }