]>
Commit | Line | Data |
---|---|---|
bde73d5b RG |
1 | /* |
2 | * This file is part of PowerDNS or dnsdist. | |
3 | * Copyright -- PowerDNS.COM B.V. and its contributors | |
4 | * | |
5 | * This program is free software; you can redistribute it and/or modify | |
6 | * it under the terms of version 2 of the GNU General Public License as | |
7 | * published by the Free Software Foundation. | |
8 | * | |
9 | * In addition, for the avoidance of any doubt, permission is granted to | |
10 | * link this program with OpenSSL and to (re)distribute the binaries | |
11 | * produced as the result of such linking. | |
12 | * | |
13 | * This program is distributed in the hope that it will be useful, | |
14 | * but WITHOUT ANY WARRANTY; without even the implied warranty of | |
15 | * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the | |
16 | * GNU General Public License for more details. | |
17 | * | |
18 | * You should have received a copy of the GNU General Public License | |
19 | * along with this program; if not, write to the Free Software | |
20 | * Foundation, Inc., 51 Franklin Street, Fifth Floor, Boston, MA 02110-1301 USA. | |
21 | */ | |
22 | ||
23 | #include "dnsdist-proxy-protocol.hh" | |
24 | ||
25 | bool addProxyProtocol(DNSQuestion& dq) | |
26 | { | |
27 | auto payload = makeProxyHeader(dq.tcp, *dq.remote, *dq.local, dq.proxyProtocolValues ? *dq.proxyProtocolValues : std::vector<ProxyProtocolValue>()); | |
28 | if ((dq.size - dq.len) < payload.size()) { | |
29 | return false; | |
30 | } | |
31 | ||
32 | memmove(reinterpret_cast<char*>(dq.dh) + payload.size(), dq.dh, dq.len); | |
33 | memcpy(dq.dh, payload.c_str(), payload.size()); | |
34 | dq.len += payload.size(); | |
35 | ||
36 | return true; | |
37 | } | |
fbfb8fb8 RG |
38 | |
39 | bool addProxyProtocol(std::vector<uint8_t>& buffer, bool tcp, const ComboAddress& source, const ComboAddress& destination, const std::vector<ProxyProtocolValue>& values) | |
40 | { | |
41 | auto payload = makeProxyHeader(tcp, source, destination, values); | |
42 | ||
43 | auto previousSize = buffer.size(); | |
44 | buffer.resize(previousSize + payload.size()); | |
45 | std::copy_backward(buffer.begin(), buffer.begin() + previousSize, buffer.end()); | |
46 | std::copy(payload.begin(), payload.end(), buffer.begin()); | |
47 | ||
48 | return true; | |
49 | } |