• Home
  • Features
  • Pricing
  • Docs
  • Announcements
  • Sign In

PowerDNS / pdns / 12351442937

16 Dec 2024 11:13AM UTC coverage: 64.773% (+0.02%) from 64.751%
12351442937

Pull #14617

github

web-flow
Merge a5962f66f into 2c5c5b828
Pull Request #14617: rec: dedup records

37546 of 88848 branches covered (42.26%)

Branch coverage included in aggregate %.

77 of 77 new or added lines in 8 files covered. (100.0%)

42 existing lines in 12 files now uncovered.

125960 of 163583 relevant lines covered (77.0%)

4425415.16 hits per line

Source File
Press 'n' to go to next uncovered line, 'b' for previous

88.62
/modules/gpgsqlbackend/gpgsqlbackend.cc
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
#ifdef HAVE_CONFIG_H
23
#include "config.h"
24
#endif
25
#include <string>
26
#include <map>
27
#include "pdns/namespaces.hh"
28
#include "pdns/dns.hh"
29
#include "pdns/dnsbackend.hh"
30
#include "pdns/dnspacket.hh"
31
#include "pdns/pdnsexception.hh"
32
#include "pdns/logger.hh"
33
#include "pdns/arguments.hh"
34
#include "gpgsqlbackend.hh"
35
#include "spgsql.hh"
36
#include <sys/time.h>
37
#include <sstream>
38

39
gPgSQLBackend::gPgSQLBackend(const string& mode, const string& suffix) :
40
  GSQLBackend(mode, suffix)
41
{
1,796✔
42
  try {
1,796✔
43
    setDB(std::unique_ptr<SSql>(new SPgSQL(getArg("dbname"),
1,796✔
44
                                           getArg("host"),
1,796✔
45
                                           getArg("port"),
1,796✔
46
                                           getArg("user"),
1,796✔
47
                                           getArg("password"),
1,796✔
48
                                           getArg("extra-connection-parameters"),
1,796✔
49
                                           mustDo("prepared-statements"))));
1,796✔
50
  }
1,796✔
51

52
  catch (SSqlException& e) {
1,796✔
53
    g_log << Logger::Error << mode << " Connection failed: " << e.txtReason() << endl;
×
54
    throw PDNSException("Unable to launch " + mode + " connection: " + e.txtReason());
×
55
  }
×
56
  allocateStatements();
1,796✔
57
  g_log << Logger::Info << mode << " Connection successful. Connected to database '" << getArg("dbname") << "' on '" << getArg("host") << "'." << endl;
1,796✔
58
}
1,796✔
59

60
void gPgSQLBackend::reconnect()
61
{
×
62
  freeStatements();
×
63

64
  if (d_db) {
×
65
    d_db->reconnect();
×
66

67
    allocateStatements();
×
68
  }
×
69
}
×
70

71
bool gPgSQLBackend::inTransaction()
72
{
330,267✔
73
  const auto* db = dynamic_cast<SPgSQL*>(d_db.get());
330,267✔
74
  if (db) {
330,267!
75
    return db->in_trx();
330,267✔
76
  }
330,267✔
UNCOV
77
  return false;
×
78
}
330,267✔
79

80
class gPgSQLFactory : public BackendFactory
81
{
82
public:
83
  gPgSQLFactory(const string& mode) :
84
    BackendFactory(mode), d_mode(mode) {}
3,475✔
85

86
  void declareArguments(const string& suffix = "") override
87
  {
323✔
88
    declare(suffix, "dbname", "Backend database name to connect to", "");
323✔
89
    declare(suffix, "user", "Database backend user to connect as", "");
323✔
90
    declare(suffix, "host", "Database backend host to connect to", "");
323✔
91
    declare(suffix, "port", "Database backend port to connect to", "");
323✔
92
    declare(suffix, "password", "Database backend password to connect with", "");
323✔
93
    declare(suffix, "extra-connection-parameters", "Extra parameters to add to connection string", "");
323✔
94
    declare(suffix, "prepared-statements", "Use prepared statements instead of parameterized queries", "yes");
323✔
95

96
    declare(suffix, "dnssec", "Enable DNSSEC processing", "no");
323✔
97

98
    string record_query = "SELECT content,ttl,prio,type,domain_id,disabled::int,name,auth::int FROM records WHERE";
323✔
99

100
    declare(suffix, "basic-query", "Basic query", record_query + " disabled=false and type=$1 and name=$2");
323✔
101
    declare(suffix, "id-query", "Basic with ID query", record_query + " disabled=false and type=$1 and name=$2 and domain_id=$3");
323✔
102
    declare(suffix, "any-query", "Any query", record_query + " disabled=false and name=$1");
323✔
103
    declare(suffix, "any-id-query", "Any with ID query", record_query + " disabled=false and name=$1 and domain_id=$2");
323✔
104

105
    declare(suffix, "list-query", "AXFR query", "SELECT content,ttl,prio,type,domain_id,disabled::int,name,auth::int,ordername FROM records WHERE (disabled=false OR $1) and domain_id=$2 order by name, type");
323✔
106
    declare(suffix, "list-subzone-query", "Subzone listing", record_query + " disabled=false and (name=$1 OR name like $2) and domain_id=$3");
323✔
107

108
    declare(suffix, "remove-empty-non-terminals-from-zone-query", "remove all empty non-terminals from zone", "delete from records where domain_id=$1 and type is null");
323✔
109
    declare(suffix, "delete-empty-non-terminal-query", "delete empty non-terminal from zone", "delete from records where domain_id=$1 and name=$2 and type is null");
323✔
110

111
    declare(suffix, "info-zone-query", "", "select id,name,master,last_check,notified_serial,type,options,catalog,account from domains where name=$1");
323✔
112

113
    declare(suffix, "info-all-secondaries-query", "", "select domains.id, domains.name, domains.type, domains.master, domains.last_check, records.content from domains LEFT JOIN records ON records.domain_id=domains.id AND records.type='SOA' AND records.name=domains.name where domains.type in ('SLAVE', 'CONSUMER')");
323✔
114
    declare(suffix, "autoprimary-query", "", "select account from supermasters where ip=$1 and nameserver=$2");
323✔
115
    declare(suffix, "autoprimary-name-to-ips", "", "select ip,account from supermasters where nameserver=$1 and account=$2");
323✔
116
    declare(suffix, "autoprimary-add", "", "insert into supermasters (ip, nameserver, account) values ($1,$2,$3)");
323✔
117
    declare(suffix, "autoprimary-remove", "", "delete from supermasters where ip = $1 and nameserver = $2");
323✔
118
    declare(suffix, "list-autoprimaries", "", "select ip,nameserver,account from supermasters");
323✔
119

120
    declare(suffix, "insert-zone-query", "", "insert into domains (type,name,master,account,last_check, notified_serial) values($1,$2,$3,$4,null,null)");
323✔
121

122
    declare(suffix, "insert-record-query", "", "insert into records (content,ttl,prio,type,domain_id,disabled,name,ordername,auth) values ($1,$2,$3,$4,$5,$6,$7,$8,$9)");
323✔
123
    declare(suffix, "insert-empty-non-terminal-order-query", "insert empty non-terminal in zone", "insert into records (type,domain_id,disabled,name,ordername,auth,ttl,prio,content) values (null,$1,false,$2,$3,$4,null,null,null)");
323✔
124

125
    declare(suffix, "get-order-first-query", "DNSSEC Ordering Query, first", "select ordername from records where disabled=false and domain_id=$1 and ordername is not null order by 1 using ~<~ limit 1");
323✔
126
    declare(suffix, "get-order-before-query", "DNSSEC Ordering Query, before", "select ordername, name from records where disabled=false and ordername ~<=~ $1 and domain_id=$2 and ordername is not null order by 1 using ~>~ limit 1");
323✔
127
    declare(suffix, "get-order-after-query", "DNSSEC Ordering Query, after", "select ordername from records where disabled=false and ordername ~>~ $1 and domain_id=$2 and ordername is not null order by 1 using ~<~ limit 1");
323✔
128
    declare(suffix, "get-order-last-query", "DNSSEC Ordering Query, last", "select ordername, name from records where disabled=false and ordername != '' and domain_id=$1 and ordername is not null order by 1 using ~>~ limit 1");
323✔
129

130
    declare(suffix, "update-ordername-and-auth-query", "DNSSEC update ordername and auth for a qname query", "update records set ordername=$1,auth=$2 where domain_id=$3 and name=$4 and disabled=false");
323✔
131
    declare(suffix, "update-ordername-and-auth-type-query", "DNSSEC update ordername and auth for a rrset query", "update records set ordername=$1,auth=$2 where domain_id=$3 and name=$4 and type=$5 and disabled=false");
323✔
132
    declare(suffix, "nullify-ordername-and-update-auth-query", "DNSSEC nullify ordername and update auth for a qname query", "update records set ordername=NULL,auth=$1 where domain_id=$2 and name=$3 and disabled=false");
323✔
133
    declare(suffix, "nullify-ordername-and-update-auth-type-query", "DNSSEC nullify ordername and update auth for a rrset query", "update records set ordername=NULL,auth=$1 where domain_id=$2 and name=$3 and type=$4 and disabled=false");
323✔
134

135
    declare(suffix, "update-primary-query", "", "update domains set master=$1 where name=$2");
323✔
136
    declare(suffix, "update-kind-query", "", "update domains set type=$1 where name=$2");
323✔
137
    declare(suffix, "update-options-query", "", "update domains set options=$1 where name=$2");
323✔
138
    declare(suffix, "update-catalog-query", "", "update domains set catalog=$1 where name=$2");
323✔
139
    declare(suffix, "update-account-query", "", "update domains set account=$1 where name=$2");
323✔
140
    declare(suffix, "update-serial-query", "", "update domains set notified_serial=$1 where id=$2");
323✔
141
    declare(suffix, "update-lastcheck-query", "", "update domains set last_check=$1 where id=$2");
323✔
142
    declare(suffix, "info-all-primary-query", "", "select domains.id, domains.name, domains.type, domains.notified_serial, domains.options, domains.catalog, records.content from records join domains on records.domain_id=domains.id and records.name=domains.name where records.type='SOA' and records.disabled=false and domains.type in ('MASTER', 'PRODUCER') order by domains.id");
323✔
143
    declare(suffix, "info-producer-members-query", "", "select domains.id, domains.name, domains.options from records join domains on records.domain_id=domains.id and records.name=domains.name where domains.type='MASTER' and domains.catalog=$1 and records.type='SOA' and records.disabled=false");
323✔
144
    declare(suffix, "info-consumer-members-query", "", "select id, name, options, master from domains where type='SLAVE' and catalog=$1");
323✔
145
    declare(suffix, "delete-domain-query", "", "delete from domains where name=$1");
323✔
146
    declare(suffix, "delete-zone-query", "", "delete from records where domain_id=$1");
323✔
147
    declare(suffix, "delete-rrset-query", "", "delete from records where domain_id=$1 and name=$2 and type=$3");
323✔
148
    declare(suffix, "delete-names-query", "", "delete from records where domain_id=$1 and name=$2");
323✔
149

150
    declare(suffix, "add-domain-key-query", "", "insert into cryptokeys (domain_id, flags, active, published, content) select id, $1, $2, $3, $4 from domains where name=$5 returning id");
323✔
151
    declare(suffix, "get-last-inserted-key-id-query", "", "select pdns_bug_should_not_get_here('https://github.com/PowerDNS/pdns/pull/10392'), 1/0");
323✔
152
    declare(suffix, "list-domain-keys-query", "", "select cryptokeys.id, flags, case when active then 1 else 0 end as active, case when published then 1 else 0 end as published, content from domains, cryptokeys where cryptokeys.domain_id=domains.id and name=$1");
323✔
153
    declare(suffix, "get-all-domain-metadata-query", "", "select kind,content from domains, domainmetadata where domainmetadata.domain_id=domains.id and name=$1");
323✔
154
    declare(suffix, "get-domain-metadata-query", "", "select content from domains, domainmetadata where domainmetadata.domain_id=domains.id and name=$1 and domainmetadata.kind=$2");
323✔
155
    declare(suffix, "clear-domain-metadata-query", "", "delete from domainmetadata where domain_id=(select id from domains where name=$1) and domainmetadata.kind=$2");
323✔
156
    declare(suffix, "clear-domain-all-metadata-query", "", "delete from domainmetadata where domain_id=(select id from domains where name=$1)");
323✔
157
    declare(suffix, "set-domain-metadata-query", "", "insert into domainmetadata (domain_id, kind, content) select id, $1, $2 from domains where name=$3");
323✔
158
    declare(suffix, "activate-domain-key-query", "", "update cryptokeys set active=true where domain_id=(select id from domains where name=$1) and  cryptokeys.id=$2");
323✔
159
    declare(suffix, "deactivate-domain-key-query", "", "update cryptokeys set active=false where domain_id=(select id from domains where name=$1) and  cryptokeys.id=$2");
323✔
160
    declare(suffix, "publish-domain-key-query", "", "update cryptokeys set published=true where domain_id=(select id from domains where name=$1) and  cryptokeys.id=$2");
323✔
161
    declare(suffix, "unpublish-domain-key-query", "", "update cryptokeys set published=false where domain_id=(select id from domains where name=$1) and  cryptokeys.id=$2");
323✔
162
    declare(suffix, "remove-domain-key-query", "", "delete from cryptokeys where domain_id=(select id from domains where name=$1) and cryptokeys.id=$2");
323✔
163
    declare(suffix, "clear-domain-all-keys-query", "", "delete from cryptokeys where domain_id=(select id from domains where name=$1)");
323✔
164
    declare(suffix, "get-tsig-key-query", "", "select algorithm, secret from tsigkeys where name=$1");
323✔
165
    declare(suffix, "set-tsig-key-query", "", "insert into tsigkeys (name,algorithm,secret) values($1,$2,$3)");
323✔
166
    declare(suffix, "delete-tsig-key-query", "", "delete from tsigkeys where name=$1");
323✔
167
    declare(suffix, "get-tsig-keys-query", "", "select name,algorithm, secret from tsigkeys");
323✔
168

169
    declare(suffix, "get-all-domains-query", "Retrieve all domains", "select domains.id, domains.name, records.content, domains.type, domains.master, domains.notified_serial, domains.last_check, domains.account, domains.catalog from domains LEFT JOIN records ON records.domain_id=domains.id AND records.type='SOA' AND records.name=domains.name WHERE records.disabled=false OR $1");
323✔
170

171
    declare(suffix, "list-comments-query", "", "SELECT domain_id,name,type,modified_at,account,comment FROM comments WHERE domain_id=$1");
323✔
172
    declare(suffix, "insert-comment-query", "", "INSERT INTO comments (domain_id, name, type, modified_at, account, comment) VALUES ($1, $2, $3, $4, $5, $6)");
323✔
173
    declare(suffix, "delete-comment-rrset-query", "", "DELETE FROM comments WHERE domain_id=$1 AND name=$2 AND type=$3");
323✔
174
    declare(suffix, "delete-comments-query", "", "DELETE FROM comments WHERE domain_id=$1");
323✔
175
    declare(suffix, "search-records-query", "", record_query + " name ILIKE $1 OR content ILIKE $2 LIMIT $3");
323✔
176
    declare(suffix, "search-comments-query", "", "SELECT domain_id,name,type,modified_at,account,comment FROM comments WHERE name ILIKE $1 OR comment ILIKE $2 LIMIT $3");
323✔
177
  }
323✔
178

179
  DNSBackend* make(const string& suffix = "") override
180
  {
1,796✔
181
    return new gPgSQLBackend(d_mode, suffix);
1,796✔
182
  }
1,796✔
183

184
private:
185
  const string d_mode;
186
};
187

188
//! Magic class that is activated when the dynamic library is loaded
189
class gPgSQLLoader
190
{
191
public:
192
  //! This reports us to the main UeberBackend class
193
  gPgSQLLoader()
194
  {
3,475✔
195
    BackendMakers().report(std::make_unique<gPgSQLFactory>("gpgsql"));
3,475✔
196
    g_log << Logger::Info << "[gpgsqlbackend] This is the gpgsql backend version " VERSION
3,475✔
197
#ifndef REPRODUCIBLE
3,475✔
198
          << " (" __DATE__ " " __TIME__ ")"
3,475✔
199
#endif
3,475✔
200
          << " reporting" << endl;
3,475✔
201
  }
3,475✔
202
};
203
static gPgSQLLoader gpgsqlloader;
STATUS · Troubleshooting · Open an Issue · Sales · Support · CAREERS · ENTERPRISE · START FREE · SCHEDULE DEMO
ANNOUNCEMENTS · TWITTER · TOS & SLA · Supported CI Services · What's a CI service? · Automated Testing

© 2025 Coveralls, Inc