88 lines
2.7 KiB
C++
88 lines
2.7 KiB
C++
//
|
|
// Copyright (C) 2010 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
|
|
//
|
|
// http://www.apache.org/licenses/LICENSE-2.0
|
|
//
|
|
// Unless required by applicable law or agreed to in writing, software
|
|
// distributed under the License is distributed on an "AS IS" BASIS,
|
|
// WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
|
|
// See the License for the specific language governing permissions and
|
|
// limitations under the License.
|
|
//
|
|
|
|
#ifndef UPDATE_ENGINE_PROXY_RESOLVER_H_
|
|
#define UPDATE_ENGINE_PROXY_RESOLVER_H_
|
|
|
|
#include <deque>
|
|
#include <string>
|
|
|
|
#include <base/logging.h>
|
|
#include <brillo/message_loops/message_loop.h>
|
|
|
|
#include "update_engine/common/utils.h"
|
|
|
|
namespace chromeos_update_engine {
|
|
|
|
extern const char kNoProxy[];
|
|
|
|
// Callback for a call to GetProxiesForUrl().
|
|
// Resultant proxies are in |out_proxy|. Each will be in one of the
|
|
// following forms:
|
|
// http://<host>[:<port>] - HTTP proxy
|
|
// socks{4,5}://<host>[:<port>] - SOCKS4/5 proxy
|
|
// kNoProxy - no proxy
|
|
typedef void (*ProxiesResolvedFn)(const std::deque<std::string>& proxies,
|
|
void* data);
|
|
|
|
class ProxyResolver {
|
|
public:
|
|
ProxyResolver() {}
|
|
virtual ~ProxyResolver() {}
|
|
|
|
// Finds proxies for the given URL and returns them via the callback.
|
|
// |data| will be passed to the callback.
|
|
// Returns true on success.
|
|
virtual bool GetProxiesForUrl(const std::string& url,
|
|
ProxiesResolvedFn callback,
|
|
void* data) = 0;
|
|
|
|
private:
|
|
DISALLOW_COPY_AND_ASSIGN(ProxyResolver);
|
|
};
|
|
|
|
// Always says to not use a proxy
|
|
class DirectProxyResolver : public ProxyResolver {
|
|
public:
|
|
DirectProxyResolver() = default;
|
|
~DirectProxyResolver() override;
|
|
bool GetProxiesForUrl(const std::string& url,
|
|
ProxiesResolvedFn callback,
|
|
void* data) override;
|
|
|
|
// Set the number of direct (non-) proxies to be returned by resolver.
|
|
// The default value is 1; higher numbers are currently used in testing.
|
|
inline void set_num_proxies(size_t num_proxies) {
|
|
num_proxies_ = num_proxies;
|
|
}
|
|
|
|
private:
|
|
// The ID of the main loop callback.
|
|
brillo::MessageLoop::TaskId idle_callback_id_{
|
|
brillo::MessageLoop::kTaskIdNull};
|
|
|
|
// Number of direct proxies to return on resolved list; currently used for
|
|
// testing.
|
|
size_t num_proxies_{1};
|
|
|
|
// The MainLoop callback, from here we return to the client.
|
|
void ReturnCallback(ProxiesResolvedFn callback, void* data);
|
|
DISALLOW_COPY_AND_ASSIGN(DirectProxyResolver);
|
|
};
|
|
|
|
} // namespace chromeos_update_engine
|
|
|
|
#endif // UPDATE_ENGINE_PROXY_RESOLVER_H_
|