mirror of
https://github.com/OpenVPN/openvpn3.git
synced 2024-09-20 04:02:15 +02:00
93 lines
2.4 KiB
C++
93 lines
2.4 KiB
C++
//
|
|
// continuation.hpp
|
|
// OpenVPN
|
|
//
|
|
// Copyright (c) 2012 OpenVPN Technologies, Inc. All rights reserved.
|
|
//
|
|
|
|
// Handle pushed option list "continuations". This is where multiple
|
|
// option lists are pushed by the server, if an option list doesn't fit
|
|
// into the standard 1024 byte buffer. This class will aggregate the
|
|
// options.
|
|
|
|
#ifndef OPENVPN_OPTIONS_CONTINUATION_H
|
|
#define OPENVPN_OPTIONS_CONTINUATION_H
|
|
|
|
#include <openvpn/common/exception.hpp>
|
|
#include <openvpn/common/options.hpp>
|
|
|
|
namespace openvpn {
|
|
|
|
struct PushOptionsBase : public RC<thread_unsafe_refcount>
|
|
{
|
|
typedef boost::intrusive_ptr<PushOptionsBase> Ptr;
|
|
|
|
OptionList multi;
|
|
OptionList singleton;
|
|
};
|
|
|
|
// Aggregate pushed option continuations into a singular option list.
|
|
// Note that map is not updated until list is complete.
|
|
class OptionListContinuation : public OptionList
|
|
{
|
|
public:
|
|
OPENVPN_SIMPLE_EXCEPTION(olc_complete); // add called when object is already complete
|
|
|
|
OptionListContinuation(const PushOptionsBase::Ptr& push_base_arg)
|
|
: partial_(false),
|
|
complete_(false),
|
|
push_base(push_base_arg)
|
|
{
|
|
// Prepend from base where multiple options of the same type can aggregate,
|
|
// so that server-pushed options will be at the end of list.
|
|
if (push_base)
|
|
extend(push_base->multi, NULL);
|
|
}
|
|
|
|
// call with option list fragments
|
|
void add(const OptionList& other, OptionList::FilterBase* filt)
|
|
{
|
|
if (!complete_)
|
|
{
|
|
partial_ = true;
|
|
extend(other, filt);
|
|
if (!continuation(other))
|
|
{
|
|
if (push_base)
|
|
{
|
|
// Append from base where only a single instance of each option makes sense,
|
|
// provided that option wasn't already pushed by server.
|
|
update_map();
|
|
extend_nonexistent(push_base->singleton);
|
|
}
|
|
update_map();
|
|
complete_ = true;
|
|
}
|
|
}
|
|
else
|
|
throw olc_complete();
|
|
}
|
|
|
|
// returns true if add() was called at least once
|
|
bool partial() const { return partial_; }
|
|
|
|
// returns true if option list is complete
|
|
bool complete() const { return complete_; }
|
|
|
|
private:
|
|
static bool continuation(const OptionList& opt)
|
|
{
|
|
const Option *o = opt.get_ptr("push-continuation");
|
|
return o && o->size() >= 2 && o->ref(1) == "2";
|
|
}
|
|
|
|
bool partial_;
|
|
bool complete_;
|
|
|
|
PushOptionsBase::Ptr push_base;
|
|
};
|
|
|
|
} // namespace openvpn
|
|
|
|
#endif // OPENVPN_OPTIONS_CONTINUATION_H
|