1 /*- 2 * SPDX-License-Identifier: BSD-3-Clause 3 * 4 * Copyright (c) 1998-2014 Dag-Erling Smørgrav 5 * All rights reserved. 6 * 7 * Redistribution and use in source and binary forms, with or without 8 * modification, are permitted provided that the following conditions 9 * are met: 10 * 1. Redistributions of source code must retain the above copyright 11 * notice, this list of conditions and the following disclaimer 12 * in this position and unchanged. 13 * 2. Redistributions in binary form must reproduce the above copyright 14 * notice, this list of conditions and the following disclaimer in the 15 * documentation and/or other materials provided with the distribution. 16 * 3. The name of the author may not be used to endorse or promote products 17 * derived from this software without specific prior written permission 18 * 19 * THIS SOFTWARE IS PROVIDED BY THE AUTHOR ``AS IS'' AND ANY EXPRESS OR 20 * IMPLIED WARRANTIES, INCLUDING, BUT NOT LIMITED TO, THE IMPLIED WARRANTIES 21 * OF MERCHANTABILITY AND FITNESS FOR A PARTICULAR PURPOSE ARE DISCLAIMED. 22 * IN NO EVENT SHALL THE AUTHOR BE LIABLE FOR ANY DIRECT, INDIRECT, 23 * INCIDENTAL, SPECIAL, EXEMPLARY, OR CONSEQUENTIAL DAMAGES (INCLUDING, BUT 24 * NOT LIMITED TO, PROCUREMENT OF SUBSTITUTE GOODS OR SERVICES; LOSS OF USE, 25 * DATA, OR PROFITS; OR BUSINESS INTERRUPTION) HOWEVER CAUSED AND ON ANY 26 * THEORY OF LIABILITY, WHETHER IN CONTRACT, STRICT LIABILITY, OR TORT 27 * (INCLUDING NEGLIGENCE OR OTHERWISE) ARISING IN ANY WAY OUT OF THE USE OF 28 * THIS SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF SUCH DAMAGE. 29 */ 30 31 #ifndef _COMMON_H_INCLUDED 32 #define _COMMON_H_INCLUDED 33 34 #define FTP_DEFAULT_PORT 21 35 #define HTTP_DEFAULT_PORT 80 36 #define FTP_DEFAULT_PROXY_PORT 21 37 #define HTTP_DEFAULT_PROXY_PORT 3128 38 39 #ifdef WITH_SSL 40 #include <openssl/crypto.h> 41 #include <openssl/x509.h> 42 #include <openssl/pem.h> 43 #include <openssl/ssl.h> 44 #include <openssl/err.h> 45 #endif 46 47 /* Connection */ 48 typedef struct fetchconn conn_t; 49 struct fetchconn { 50 int sd; /* socket descriptor */ 51 char *buf; /* buffer */ 52 size_t bufsize; /* buffer size */ 53 size_t buflen; /* length of buffer contents */ 54 size_t pos; /* current position in buffer */ 55 int err; /* last protocol reply code */ 56 #ifdef WITH_SSL 57 SSL *ssl; /* SSL handle */ 58 SSL_CTX *ssl_ctx; /* SSL context */ 59 X509 *ssl_cert; /* server certificate */ 60 #endif 61 int ref; /* reference count */ 62 }; 63 64 static inline conn_t *fetch_ref(conn_t *conn) 65 { 66 ++conn->ref; 67 return (conn); 68 } 69 70 static inline conn_t *fetch_deref(conn_t *conn) 71 { 72 --conn->ref; 73 return (conn); 74 } 75 76 /* Structure used for error message lists */ 77 struct fetcherr { 78 const int num; 79 const int cat; 80 const char *string; 81 }; 82 83 /* For SOCKS header size */ 84 #define HEAD_SIZE 4 85 #define FQDN_SIZE 256 86 #define PACK_SIZE 1 87 #define PORT_SIZE 2 88 #define BUFF_SIZE HEAD_SIZE + FQDN_SIZE + PACK_SIZE + PORT_SIZE 89 90 /* SOCKS5 Request Header */ 91 #define SOCKS_VERSION_5 0x05 92 /* SOCKS5 CMD */ 93 #define SOCKS_CONNECTION 0x01 94 #define SOCKS_BIND 0x02 95 #define SOCKS_UDP 0x03 96 #define SOCKS_NOMETHODS 0xFF 97 #define SOCKS5_NOTIMPLEMENTED 0x00 98 /* SOCKS5 Reserved */ 99 #define SOCKS_RSV 0x00 100 /* SOCKS5 Address Type */ 101 #define SOCKS_ATYP_IPV4 0x01 102 #define SOCKS_ATYP_DOMAINNAME 0x03 103 #define SOCKS_ATYP_IPV6 0x04 104 /* SOCKS5 Reply Field */ 105 #define SOCKS_SUCCESS 0x00 106 #define SOCKS_GENERAL_FAILURE 0x01 107 #define SOCKS_CONNECTION_NOT_ALLOWED 0x02 108 #define SOCKS_NETWORK_UNREACHABLE 0x03 109 #define SOCKS_HOST_UNREACHABLE 0x04 110 #define SOCKS_CONNECTION_REFUSED 0x05 111 #define SOCKS_TTL_EXPIRED 0x06 112 #define SOCKS_COMMAND_NOT_SUPPORTED 0x07 113 #define SOCKS_ADDRESS_NOT_SUPPORTED 0x08 114 115 /* for fetch_writev */ 116 struct iovec; 117 118 void fetch_seterr(struct fetcherr *, int); 119 void fetch_syserr(void); 120 void fetch_info(const char *, ...) __printflike(1, 2); 121 int fetch_socks5_getenv(char **host, int *port); 122 int fetch_socks5_init(conn_t *conn, const char *host, 123 int port, int verbose); 124 int fetch_default_port(const char *); 125 int fetch_default_proxy_port(const char *); 126 struct addrinfo *fetch_resolve(const char *, int, int); 127 int fetch_bind(int, int, const char *); 128 conn_t *fetch_connect(const char *, int, int, int); 129 conn_t *fetch_reopen(int); 130 #ifdef WITH_SSL 131 int fetch_ssl_cb_verify_crt(int, X509_STORE_CTX*); 132 #endif 133 int fetch_ssl(conn_t *, const struct url *, int); 134 ssize_t fetch_read(conn_t *, void *, size_t); 135 ssize_t fetch_getln(conn_t *); 136 ssize_t fetch_bufread(conn_t *, void *, size_t); 137 ssize_t fetch_write(conn_t *, const void *, size_t); 138 ssize_t fetch_writev(conn_t *, struct iovec *, int); 139 int fetch_putln(conn_t *, const char *, size_t); 140 int fetch_close(conn_t *); 141 int fetch_add_entry(struct url_ent **, int *, int *, 142 const char *, struct url_stat *); 143 int fetch_netrc_auth(struct url *url); 144 int fetch_no_proxy_match(const char *); 145 146 #define ftp_seterr(n) fetch_seterr(ftp_errlist, n) 147 #define http_seterr(n) fetch_seterr(http_errlist, n) 148 #define netdb_seterr(n) fetch_seterr(netdb_errlist, n) 149 #define url_seterr(n) fetch_seterr(url_errlist, n) 150 #define socks5_seterr(n) fetch_seterr(socks5_errlist, n) 151 152 #ifndef NDEBUG 153 #define DEBUGF(...) \ 154 do { \ 155 if (fetchDebug) \ 156 fprintf(stderr, __VA_ARGS__); \ 157 } while (0) 158 #else 159 #define DEBUGF(...) \ 160 do { \ 161 /* nothing */ \ 162 } while (0) 163 #endif 164 165 /* 166 * I don't really like exporting http_request() and ftp_request(), 167 * but the HTTP and FTP code occasionally needs to cross-call 168 * eachother, and this saves me from adding a lot of special-case code 169 * to handle those cases. 170 * 171 * Note that _*_request() free purl, which is way ugly but saves us a 172 * whole lot of trouble. 173 */ 174 FILE *http_request(struct url *, const char *, 175 struct url_stat *, struct url *, const char *); 176 FILE *http_request_body(struct url *, const char *, 177 struct url_stat *, struct url *, const char *, 178 const char *, const char *); 179 FILE *ftp_request(struct url *, const char *, 180 struct url_stat *, struct url *, const char *); 181 182 /* 183 * Check whether a particular flag is set 184 */ 185 #define CHECK_FLAG(x) (flags && strchr(flags, (x))) 186 187 #endif 188