blob: c0fbaef05a052a4d87a52d169c1ecc7aee5886e9 [file] [log] [blame]
* Copyright (C) 2017 The Android Open Source Project
* Licensed under the Apache License, Version 2.0 (the "License");
* you may not use this file except in compliance with the License.
* You may obtain a copy of the License at
* Unless required by applicable law or agreed to in writing, software
* distributed under the License is distributed on an "AS IS" BASIS,
* See the License for the specific language governing permissions and
* limitations under the License.
#include <future>
#include <map>
#include <mutex>
#include <vector>
#include <android-base/thread_annotations.h>
#include <android-base/unique_fd.h>
#include <netdutils/Slice.h>
#include "DnsTlsQueryMap.h"
#include "DnsTlsServer.h"
#include "DnsTlsSessionCache.h"
#include "IDnsTlsSocket.h"
#include "IDnsTlsSocketObserver.h"
namespace android {
namespace net {
class IDnsTlsSocketFactory;
// Manages at most one DnsTlsSocket at a time. This class handles socket lifetime issues,
// such as reopening the socket and reissuing pending queries.
class DnsTlsTransport : public IDnsTlsSocketObserver {
DnsTlsTransport(const DnsTlsServer& server, unsigned mark,
IDnsTlsSocketFactory* _Nonnull factory)
: mMark(mark), mServer(server), mFactory(factory) {}
using Response = DnsTlsQueryMap::Response;
using Result = DnsTlsQueryMap::Result;
// Given a |query|, this method sends it to the server and returns the result asynchronously.
std::future<Result> query(const netdutils::Slice query) EXCLUDES(mLock);
// Check that a given TLS server is fully working on the specified netid.
// This function is used in ResolverController to ensure that we don't enable DNS over TLS
// on networks where it doesn't actually work.
static bool validate(const DnsTlsServer& server, unsigned netid, uint32_t mark);
int getConnectCounter() const EXCLUDES(mLock);
// Implement IDnsTlsSocketObserver
void onResponse(std::vector<uint8_t> response) override;
void onClosed() override EXCLUDES(mLock);
mutable std::mutex mLock;
DnsTlsSessionCache mCache;
DnsTlsQueryMap mQueries;
const unsigned mMark; // Socket mark
const DnsTlsServer mServer;
IDnsTlsSocketFactory* _Nonnull const mFactory;
void doConnect() REQUIRES(mLock);
// doReconnect is used by onClosed. It runs on the reconnect thread.
void doReconnect() EXCLUDES(mLock);
std::unique_ptr<std::thread> mReconnectThread GUARDED_BY(mLock);
// Used to prevent onClosed from starting a reconnect during the destructor.
bool mClosing GUARDED_BY(mLock) = false;
// Sending queries on the socket is thread-safe, but construction/destruction is not.
std::unique_ptr<IDnsTlsSocket> mSocket GUARDED_BY(mLock);
// Send a query to the socket.
bool sendQuery(const DnsTlsQueryMap::Query& q) REQUIRES(mLock);
// The number of times an attempt to connect the nameserver.
int mConnectCounter GUARDED_BY(mLock) = 0;
} // end of namespace net
} // end of namespace android