66 lines
2.0 KiB
C++
66 lines
2.0 KiB
C++
/*
|
|
* IXDNSLookup.h
|
|
* Author: Benjamin Sergeant
|
|
* Copyright (c) 2018 Machine Zone, Inc. All rights reserved.
|
|
*
|
|
* Resolve a hostname+port to a struct addrinfo obtained with getaddrinfo
|
|
* Does this in a background thread so that it can be cancelled, since
|
|
* getaddrinfo is a blocking call, and we don't want to block the main thread on Mobile.
|
|
*/
|
|
|
|
#pragma once
|
|
|
|
#include "IXCancellationRequest.h"
|
|
#include <atomic>
|
|
#include <mutex>
|
|
#include <set>
|
|
#include <string>
|
|
#include <memory>
|
|
|
|
struct addrinfo;
|
|
|
|
namespace ix
|
|
{
|
|
class DNSLookup : public std::enable_shared_from_this<DNSLookup>
|
|
{
|
|
public:
|
|
DNSLookup(const std::string& hostname, int port, int64_t wait = DNSLookup::kDefaultWait);
|
|
~DNSLookup() = default;
|
|
|
|
struct addrinfo* resolve(std::string& errMsg,
|
|
const CancellationRequest& isCancellationRequested,
|
|
bool cancellable = true);
|
|
|
|
private:
|
|
struct addrinfo* resolveCancellable(std::string& errMsg,
|
|
const CancellationRequest& isCancellationRequested);
|
|
struct addrinfo* resolveUnCancellable(std::string& errMsg,
|
|
const CancellationRequest& isCancellationRequested);
|
|
|
|
static struct addrinfo* getAddrInfo(const std::string& hostname,
|
|
int port,
|
|
std::string& errMsg);
|
|
|
|
void run(std::weak_ptr<DNSLookup> self, std::string hostname, int port); // thread runner
|
|
|
|
void setErrMsg(const std::string& errMsg);
|
|
const std::string& getErrMsg();
|
|
|
|
void setRes(struct addrinfo* addr);
|
|
struct addrinfo* getRes();
|
|
|
|
std::string _hostname;
|
|
int _port;
|
|
int64_t _wait;
|
|
const static int64_t kDefaultWait;
|
|
|
|
struct addrinfo* _res;
|
|
std::mutex _resMutex;
|
|
|
|
std::string _errMsg;
|
|
std::mutex _errMsgMutex;
|
|
|
|
std::atomic<bool> _done;
|
|
};
|
|
} // namespace ix
|