io_fetch.cc 16.7 KB
Newer Older
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16
// Copyright (C) 2011  Internet Systems Consortium, Inc. ("ISC")
//
// Permission to use, copy, modify, and/or distribute this software for any
// purpose with or without fee is hereby granted, provided that the above
// copyright notice and this permission notice appear in all copies.
//
// THE SOFTWARE IS PROVIDED "AS IS" AND ISC DISCLAIMS ALL WARRANTIES WITH
// REGARD TO THIS SOFTWARE INCLUDING ALL IMPLIED WARRANTIES OF MERCHANTABILITY
// AND FITNESS.  IN NO EVENT SHALL ISC BE LIABLE FOR ANY SPECIAL, DIRECT,
// INDIRECT, OR CONSEQUENTIAL DAMAGES OR ANY DAMAGES WHATSOEVER RESULTING FROM
// LOSS OF USE, DATA OR PROFITS, WHETHER IN AN ACTION OF CONTRACT, NEGLIGENCE
// OR OTHER TORTIOUS ACTION, ARISING OUT OF OR IN CONNECTION WITH THE USE OR
// PERFORMANCE OF THIS SOFTWARE.

#include <config.h>

17
#include <unistd.h>             // for some IPC/network system calls
18
#include <netinet/in.h>
19 20
#include <stdint.h>
#include <sys/socket.h>
21 22

#include <boost/bind.hpp>
23
#include <boost/scoped_ptr.hpp>
24
#include <boost/date_time/posix_time/posix_time_types.hpp>
25

Stephen Morris's avatar
Stephen Morris committed
26
#include <asio.hpp>
27 28
#include <asio/deadline_timer.hpp>

29
#include <asiolink/io_address.h>
30
#include <asiolink/io_asio_socket.h>
31
#include <asiolink/io_endpoint.h>
32 33 34 35 36
#include <asiolink/io_service.h>
#include <asiolink/tcp_endpoint.h>
#include <asiolink/tcp_socket.h>
#include <asiolink/udp_endpoint.h>
#include <asiolink/udp_socket.h>
37

38 39 40 41 42
#include <dns/messagerenderer.h>
#include <dns/opcode.h>
#include <dns/rcode.h>

#include <asiodns/io_fetch.h>
43 44 45

#include <util/buffer.h>
#include <util/random/qid_gen.h>
46

Michal 'vorner' Vaner's avatar
Michal 'vorner' Vaner committed
47
#include <asiodns/logger.h>
48 49

using namespace asio;
50
using namespace isc::asiolink;
51
using namespace isc::dns;
52
using namespace isc::util;
53
using namespace isc::util::random;
Stephen Morris's avatar
Stephen Morris committed
54 55
using namespace isc::log;
using namespace std;
56

57 58
namespace isc {
namespace asiodns {
59

60 61
// Log debug verbosity

62 63 64
const int DBG_IMPORTANT = DBGLVL_TRACE_BASIC;
const int DBG_COMMON = DBGLVL_TRACE_DETAIL;
const int DBG_ALL = DBGLVL_TRACE_DETAIL + 20;
65

66 67
/// \brief IOFetch Data
///
68 69 70 71 72
/// The data for IOFetch is held in a separate struct pointed to by a shared_ptr
/// object.  This is because the IOFetch object will be copied often (it is used
/// as a coroutine and passed as callback to many async_*() functions) and we
/// want keep the same data).  Organising the data in this way keeps copying to
/// a minimum.
73 74 75 76 77
struct IOFetchData {

    // The first two members are shared pointers to a base class because what is
    // actually instantiated depends on whether the fetch is over UDP or TCP,
    // which is not known until construction of the IOFetch.  Use of a shared
78
    // pointer here is merely to ensure deletion when the data object is deleted.
79
    boost::scoped_ptr<IOAsioSocket<IOFetch> > socket;
80 81 82
                                             ///< Socket to use for I/O
    boost::scoped_ptr<IOEndpoint> remote_snd;///< Where the fetch is sent
    boost::scoped_ptr<IOEndpoint> remote_rcv;///< Where the response came from
83 84
    OutputBufferPtr   msgbuf;      ///< Wire buffer for question
    OutputBufferPtr   received;    ///< Received data put here
85 86 87 88 89 90 91 92
    IOFetch::Callback*          callback;    ///< Called on I/O Completion
    asio::deadline_timer        timer;       ///< Timer to measure timeouts
    IOFetch::Protocol           protocol;    ///< Protocol being used
    size_t                      cumulative;  ///< Cumulative received amount
    size_t                      expected;    ///< Expected amount of data
    size_t                      offset;      ///< Offset to receive data
    bool                        stopped;     ///< Have we stopped running?
    int                         timeout;     ///< Timeout in ms
Stephen Morris's avatar
Stephen Morris committed
93
    bool                        packet;      ///< true if packet was supplied
94 95 96 97 98 99 100

    // In case we need to log an error, the origin of the last asynchronous
    // I/O is recorded.  To save time and simplify the code, this is recorded
    // as the ID of the error message that would be generated if the I/O failed.
    // This means that we must make sure that all possible "origins" take the
    // same arguments in their message in the same order.
    isc::log::MessageID         origin;     ///< Origin of last asynchronous I/O
101 102
    uint8_t                     staging[IOFetch::STAGING_LENGTH];
                                            ///< Temporary array for received data
103
    isc::dns::qid_t             qid;         ///< The QID set in the query
104 105 106 107 108

    /// \brief Constructor
    ///
    /// Just fills in the data members of the IOFetchData structure
    ///
109
    /// \param proto Either IOFetch::TCP or IOFetch::UDP.
110
    /// \param service I/O Service object to handle the asynchronous
111
    ///        operations.
112 113 114
    /// \param address IP address of upstream server
    /// \param port Port to use for the query
    /// \param buff Output buffer into which the response (in wire format)
115
    ///        is written (if a response is received).
116
    /// \param cb Callback object containing the callback to be called
117 118
    ///        when we terminate.  The caller is responsible for managing this
    ///        object and deleting it if necessary.
119 120 121
    /// \param wait Timeout for the fetch (in ms).
    ///
    /// TODO: May need to alter constructor (see comment 4 in Trac ticket #554)
122
    IOFetchData(IOFetch::Protocol proto, IOService& service,
123 124
        const IOAddress& address, uint16_t port, OutputBufferPtr& buff,
        IOFetch::Callback* cb, int wait)
125
        :
126
        socket((proto == IOFetch::UDP) ?
127 128 129 130 131
            static_cast<IOAsioSocket<IOFetch>*>(
                new UDPSocket<IOFetch>(service)) :
            static_cast<IOAsioSocket<IOFetch>*>(
                new TCPSocket<IOFetch>(service))
            ),
132 133 134 135 136
        remote_snd((proto == IOFetch::UDP) ?
            static_cast<IOEndpoint*>(new UDPEndpoint(address, port)) :
            static_cast<IOEndpoint*>(new TCPEndpoint(address, port))
            ),
        remote_rcv((proto == IOFetch::UDP) ?
137 138 139
            static_cast<IOEndpoint*>(new UDPEndpoint(address, port)) :
            static_cast<IOEndpoint*>(new TCPEndpoint(address, port))
            ),
140
        msgbuf(new OutputBuffer(512)),
141
        received(buff),
142
        callback(cb),
143 144
        timer(service.get_io_service()),
        protocol(proto),
145
        cumulative(0),
146 147
        expected(0),
        offset(0),
148 149
        stopped(false),
        timeout(wait),
Stephen Morris's avatar
Stephen Morris committed
150
        packet(false),
151
        origin(ASIODNS_UNKNOWN_ORIGIN),
152 153
        staging(),
        qid(QidGenerator::getInstance().generateQid())
154
    {}
155 156 157 158 159 160 161 162 163 164 165 166

    // Checks if the response we received was ok;
    // - data contains the buffer we read, as well as the address
    // we sent to and the address we received from.
    // length is provided by the operator() in IOFetch.
    // Addresses must match, number of octets read must be at least
    // 2, and the first two octets must match the qid of the message
    // we sent.
    bool responseOK() {
        return (*remote_snd == *remote_rcv && cumulative >= 2 &&
                readUint16(received->getData()) == qid);
    }
167 168
};

169
/// IOFetch Constructor - just initialize the private data
Stephen Morris's avatar
Stephen Morris committed
170

171
IOFetch::IOFetch(Protocol protocol, IOService& service,
Dima Volodin's avatar
Dima Volodin committed
172 173
    const isc::dns::Question& question, const IOAddress& address,
    uint16_t port, OutputBufferPtr& buff, Callback* cb, int wait, bool edns)
174
{
175 176
    MessagePtr query_msg(new Message(Message::RENDER));
    initIOFetch(query_msg, protocol, service, question, address, port, buff,
Dima Volodin's avatar
Dima Volodin committed
177
                cb, wait, edns);
178 179
}

Stephen Morris's avatar
Stephen Morris committed
180 181 182 183 184 185 186 187 188 189 190
IOFetch::IOFetch(Protocol protocol, IOService& service,
    OutputBufferPtr& outpkt, const IOAddress& address, uint16_t port,
    OutputBufferPtr& buff, Callback* cb, int wait)
    :
    data_(new IOFetchData(protocol, service,
          address, port, buff, cb, wait))
{
    data_->msgbuf = outpkt;
    data_->packet = true;
}

191 192 193 194 195
IOFetch::IOFetch(Protocol protocol, IOService& service,
    ConstMessagePtr query_message, const IOAddress& address, uint16_t port,
    OutputBufferPtr& buff, Callback* cb, int wait)
{
    MessagePtr msg(new Message(Message::RENDER));
196 197 198 199 200

    msg->setHeaderFlag(Message::HEADERFLAG_RD,
                       query_message->getHeaderFlag(Message::HEADERFLAG_RD));
    msg->setHeaderFlag(Message::HEADERFLAG_CD,
                       query_message->getHeaderFlag(Message::HEADERFLAG_CD));
201 202 203 204 205 206 207

    initIOFetch(msg, protocol, service,
                **(query_message->beginQuestion()),
                address, port, buff, cb, wait);
}

void
208 209
IOFetch::initIOFetch(MessagePtr& query_msg, Protocol protocol,
                     IOService& service,
210 211
                     const isc::dns::Question& question,
                     const IOAddress& address, uint16_t port,
Dima Volodin's avatar
Dima Volodin committed
212
                     OutputBufferPtr& buff, Callback* cb, int wait, bool edns)
213 214 215 216 217 218 219 220 221
{
    data_ = boost::shared_ptr<IOFetchData>(new IOFetchData(
        protocol, service, address, port, buff, cb, wait));

    query_msg->setQid(data_->qid);
    query_msg->setOpcode(Opcode::QUERY());
    query_msg->setRcode(Rcode::NOERROR());
    query_msg->setHeaderFlag(Message::HEADERFLAG_RD);
    query_msg->addQuestion(question);
Dima Volodin's avatar
Dima Volodin committed
222 223 224 225 226 227 228

    if (edns) {
        EDNSPtr edns_query(new EDNS());
        edns_query->setUDPSize(Message::DEFAULT_MAX_EDNS0_UDPSIZE);
        query_msg->setEDNS(edns_query);
    }

229 230
    MessageRenderer renderer;
    renderer.setBuffer(data_->msgbuf.get());
231
    query_msg->toWire(renderer);
232
    renderer.setBuffer(NULL);
233 234
}

235 236 237 238 239 240 241
// Return protocol in use.

IOFetch::Protocol
IOFetch::getProtocol() const {
    return (data_->protocol);
}

242 243
/// The function operator is implemented with the "stackless coroutine"
/// pattern; see internal/coroutine.h for details.
Stephen Morris's avatar
Stephen Morris committed
244

245
void
246
IOFetch::operator()(asio::error_code ec, size_t length) {
247

248 249 250 251
    if (data_->stopped) {
        return;
    } else if (ec) {
        logIOFailure(ec);
252 253 254 255
        return;
    }

    CORO_REENTER (this) {
Stephen Morris's avatar
Stephen Morris committed
256

257 258 259 260
        /// Generate the upstream query and render it to wire format
        /// This is done in a different scope to allow inline variable
        /// declarations.
        {
Stephen Morris's avatar
Stephen Morris committed
261 262 263 264 265
            if (data_->packet) {
                // A packet was given, overwrite the QID (which is in the
                // first two bytes of the packet).
                data_->msgbuf->writeUint16At(data_->qid, 0);

266
            } 
267 268
        }

269 270
        // If we timeout, we stop, which will can cancel outstanding I/Os and
        // shutdown everything.
271 272 273 274 275 276 277 278
        if (data_->timeout != -1) {
            data_->timer.expires_from_now(boost::posix_time::milliseconds(
                data_->timeout));
            data_->timer.async_wait(boost::bind(&IOFetch::stop, *this,
                TIME_OUT));
        }

        // Open a connection to the target system.  For speed, if the operation
279
        // is synchronous (i.e. UDP operation) we bypass the yield.
280
        data_->origin = ASIODNS_OPEN_SOCKET;
281
        if (data_->socket->isOpenSynchronous()) {
282
            data_->socket->open(data_->remote_snd.get(), *this);
283
        } else {
284
            CORO_YIELD data_->socket->open(data_->remote_snd.get(), *this);
285 286
        }

Stephen Morris's avatar
Stephen Morris committed
287
        do {
288 289
            // Begin an asynchronous send, and then yield.  When the send completes,
            // we will resume immediately after this point.
290
            data_->origin = ASIODNS_SEND_DATA;
291 292 293 294 295 296 297 298 299 300 301 302 303 304 305 306 307 308 309 310 311 312
            CORO_YIELD data_->socket->asyncSend(data_->msgbuf->getData(),
                data_->msgbuf->getLength(), data_->remote_snd.get(), *this);
    
            // Now receive the response.  Since TCP may not receive the entire
            // message in one operation, we need to loop until we have received
            // it. (This can't be done within the asyncReceive() method because
            // each I/O operation will be done asynchronously and between each one
            // we need to yield ... and we *really* don't want to set up another
            // coroutine within that method.)  So after each receive (and yield),
            // we check if the operation is complete and if not, loop to read again.
            //
            // Another concession to TCP is that the amount of is contained in the
            // first two bytes.  This leads to two problems:
            //
            // a) We don't want those bytes in the return buffer.
            // b) They may not both arrive in the first I/O.
            //
            // So... we need to loop until we have at least two bytes, then store
            // the expected amount of data.  Then we need to loop until we have
            // received all the data before copying it back to the user's buffer.
            // And we want to minimise the amount of copying...
    
313
            data_->origin = ASIODNS_READ_DATA;
314 315
            data_->cumulative = 0;          // No data yet received
            data_->offset = 0;              // First data into start of buffer
316
            data_->received->clear();       // Clear the receive buffer
317 318 319 320 321 322 323 324 325
            do {
                CORO_YIELD data_->socket->asyncReceive(data_->staging,
                                                       static_cast<size_t>(STAGING_LENGTH),
                                                       data_->offset,
                                                       data_->remote_rcv.get(), *this);
            } while (!data_->socket->processReceivedData(data_->staging, length,
                                                         data_->cumulative, data_->offset,
                                                         data_->expected, data_->received));
        } while (!data_->responseOK());
326

327 328
        // Finished with this socket, so close it.  This will not generate an
        // I/O error, but reset the origin to unknown in case we change this.
329
        data_->origin = ASIODNS_UNKNOWN_ORIGIN;
330 331
        data_->socket->close();

332 333 334 335 336 337 338 339 340
        /// We are done
        stop(SUCCESS);
    }
}

// Function that stops the coroutine sequence.  It is called either when the
// query finishes or when the timer times out.  Either way, it sets the
// "stopped_" flag and cancels anything that is in progress.
//
341 342
// As the function may be entered multiple times as things wind down, it checks
// if the stopped_ flag is already set.  If it is, the call is a no-op.
Stephen Morris's avatar
Stephen Morris committed
343

344 345
void
IOFetch::stop(Result result) {
346

347
    if (!data_->stopped) {
348 349 350 351 352 353 354 355 356 357 358 359 360

        // Mark the fetch as stopped to prevent other completion callbacks
        // (invoked because of the calls to cancel()) from executing the
        // cancel calls again.
        //
        // In a single threaded environment, the callbacks won't be invoked
        // until this one completes. In a multi-threaded environment, they may
        // well be, in which case the testing (and setting) of the stopped_
        // variable should be done inside a mutex (and the stopped_ variable
        // declared as "volatile").
        //
        // TODO: Update testing of stopped_ if threads are used.
        data_->stopped = true;
361 362
        switch (result) {
            case TIME_OUT:
363
                LOG_DEBUG(logger, DBG_COMMON, ASIODNS_READ_TIMEOUT).
364 365
                    arg(data_->remote_snd->getAddress().toText()).
                    arg(data_->remote_snd->getPort());
366 367
                break;

368
            case SUCCESS:
369
                LOG_DEBUG(logger, DBG_ALL, ASIODNS_FETCH_COMPLETED).
370 371
                    arg(data_->remote_rcv->getAddress().toText()).
                    arg(data_->remote_rcv->getPort());
372 373
                break;

374 375 376 377
            case STOPPED:
                // Fetch has been stopped for some other reason.  This is
                // allowed but as it is unusual it is logged, but with a lower
                // debug level than a timeout (which is totally normal).
378
                LOG_DEBUG(logger, DBG_IMPORTANT, ASIODNS_FETCH_STOPPED).
379 380
                    arg(data_->remote_snd->getAddress().toText()).
                    arg(data_->remote_snd->getPort());
381 382
                break;

383
            default:
384
                LOG_ERROR(logger, ASIODNS_UNKNOWN_RESULT).
385 386
                    arg(data_->remote_snd->getAddress().toText()).
                    arg(data_->remote_snd->getPort());
387 388
        }

Stephen Morris's avatar
Stephen Morris committed
389 390 391 392 393 394 395 396
        // Stop requested, cancel and I/O's on the socket and shut it down,
        // and cancel the timer.
        data_->socket->cancel();
        data_->socket->close();

        data_->timer.cancel();

        // Execute the I/O completion callback (if present).
397
        if (data_->callback) {
Stephen Morris's avatar
Stephen Morris committed
398
            (*(data_->callback))(result);
399 400 401 402
        }
    }
}

403 404
// Log an error - called on I/O failure

405
void IOFetch::logIOFailure(asio::error_code ec) {
406

407
    // Should only get here with a known error code.
408 409 410
    assert((data_->origin == ASIODNS_OPEN_SOCKET) ||
           (data_->origin == ASIODNS_SEND_DATA) ||
           (data_->origin == ASIODNS_READ_DATA) ||
411
           (data_->origin == ASIODNS_UNKNOWN_ORIGIN));
412

413
    static const char* PROTOCOL[2] = {"TCP", "UDP"};
414 415 416 417 418
    LOG_ERROR(logger, data_->origin).arg(ec.value()).
        arg((data_->remote_snd->getProtocol() == IPPROTO_TCP) ?
                     PROTOCOL[0] : PROTOCOL[1]).
        arg(data_->remote_snd->getAddress().toText()).
        arg(data_->remote_snd->getPort());
419 420
}

421
} // namespace asiodns
422
} // namespace isc {