Compiler.h: move to util/
[ncmpc-debian.git] / src / net / async_rconnect.cxx
1 /* ncmpc (Ncurses MPD Client)
2    (c) 2004-2018 The Music Player Daemon Project
3    Project homepage: http://musicpd.org
4
5    Redistribution and use in source and binary forms, with or without
6    modification, are permitted provided that the following conditions
7    are met:
8
9    - Redistributions of source code must retain the above copyright
10    notice, this list of conditions and the following disclaimer.
11
12    - Redistributions in binary form must reproduce the above copyright
13    notice, this list of conditions and the following disclaimer in the
14    documentation and/or other materials provided with the distribution.
15
16    THIS SOFTWARE IS PROVIDED BY THE COPYRIGHT HOLDERS AND CONTRIBUTORS
17    ``AS IS'' AND ANY EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT
18    LIMITED TO, THE IMPLIED WARRANTIES OF MERCHANTABILITY AND FITNESS FOR
19    A PARTICULAR PURPOSE ARE DISCLAIMED.  IN NO EVENT SHALL THE FOUNDATION OR
20    CONTRIBUTORS BE LIABLE FOR ANY DIRECT, INDIRECT, INCIDENTAL, SPECIAL,
21    EXEMPLARY, OR CONSEQUENTIAL DAMAGES (INCLUDING, BUT NOT LIMITED TO,
22    PROCUREMENT OF SUBSTITUTE GOODS OR SERVICES; LOSS OF USE, DATA, OR
23    PROFITS; OR BUSINESS INTERRUPTION) HOWEVER CAUSED AND ON ANY THEORY OF
24    LIABILITY, WHETHER IN CONTRACT, STRICT LIABILITY, OR TORT (INCLUDING
25    NEGLIGENCE OR OTHERWISE) ARISING IN ANY WAY OUT OF THE USE OF THIS
26    SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF SUCH DAMAGE.
27 */
28
29 #include "async_rconnect.hxx"
30 #include "async_connect.hxx"
31 #include "resolver.hxx"
32 #include "util/Compiler.h"
33
34 #include <string>
35
36 #include <assert.h>
37 #include <stdio.h>
38 #include <string.h>
39
40 struct async_rconnect {
41         const struct async_rconnect_handler *handler;
42         void *handler_ctx;
43
44         const char *host;
45         struct resolver *resolver;
46
47         struct async_connect *connect = nullptr;
48
49         std::string last_error;
50
51         async_rconnect(const char *_host, struct resolver *_resolver,
52                        const struct async_rconnect_handler &_handler,
53                        void *_ctx)
54                 :handler(&_handler), handler_ctx(_ctx),
55                  host(_host), resolver(_resolver) {}
56
57         ~async_rconnect() {
58                 if (connect != nullptr)
59                         async_connect_cancel(connect);
60                 resolver_free(resolver);
61         }
62 };
63
64 static void
65 async_rconnect_next(struct async_rconnect *rc);
66
67 static void
68 async_rconnect_success(socket_t fd, void *ctx)
69 {
70         auto *rc = (struct async_rconnect *)ctx;
71         rc->connect = nullptr;
72
73         rc->handler->success(fd, rc->handler_ctx);
74
75         delete rc;
76 }
77
78 static void
79 async_rconnect_error(const char *message, void *ctx)
80 {
81         auto *rc = (struct async_rconnect *)ctx;
82         rc->connect = nullptr;
83
84         rc->last_error = message;
85
86         async_rconnect_next(rc);
87 }
88
89 static const struct async_connect_handler async_rconnect_connect_handler = {
90         .success = async_rconnect_success,
91         .error = async_rconnect_error,
92 };
93
94 static void
95 async_rconnect_next(struct async_rconnect *rc)
96 {
97         assert(rc->connect == nullptr);
98
99         const struct resolver_address *a = resolver_next(rc->resolver);
100         if (a == nullptr) {
101                 char msg[256];
102
103                 if (rc->last_error.empty()) {
104                         snprintf(msg, sizeof(msg),
105                                  "Host '%s' has no address",
106                                  rc->host);
107                 } else {
108                         snprintf(msg, sizeof(msg),
109                                  "Failed to connect to host '%s': %s",
110                                  rc->host, rc->last_error.c_str());
111                 }
112
113                 rc->handler->error(msg, rc->handler_ctx);
114                 delete rc;
115                 return;
116         }
117
118         async_connect_start(&rc->connect, a->addr, a->addrlen,
119                             &async_rconnect_connect_handler, rc);
120 }
121
122 void
123 async_rconnect_start(struct async_rconnect **rcp,
124                      const char *host, unsigned port,
125                      const struct async_rconnect_handler *handler, void *ctx)
126 {
127         struct resolver *r = resolver_new(host, port);
128         if (host == nullptr)
129                 host = "[default]";
130
131         if (r == nullptr) {
132                 char msg[256];
133                 snprintf(msg, sizeof(msg), "Failed to resolve host '%s'",
134                          host);
135                 handler->error(msg, ctx);
136                 return;
137         }
138
139         auto *rc = new async_rconnect(host, r, *handler, ctx);
140         *rcp = rc;
141
142         async_rconnect_next(rc);
143 }
144
145 void
146 async_rconnect_cancel(struct async_rconnect *rc)
147 {
148         delete rc;
149 }