00001
00002
00003
00004
00005
00006
00007
00008
00009
00010 #include <ldns/config.h>
00011
00012 #include <ldns/ldns.h>
00013
00014 #include <strings.h>
00015 #include <stdlib.h>
00016 #include <limits.h>
00017
00018
00019
00020
00021
00022
00023
00024
00025
00026 ldns_pkt *
00027 ldns_update_pkt_new(ldns_rdf *zone_rdf, ldns_rr_class class,
00028 ldns_rr_list *pr_rrlist, ldns_rr_list *up_rrlist, ldns_rr_list *ad_rrlist)
00029 {
00030 ldns_pkt *p;
00031
00032 if (!zone_rdf || !up_rrlist) {
00033 return NULL;
00034 }
00035
00036 if (class == 0) {
00037 class = LDNS_RR_CLASS_IN;
00038 }
00039
00040
00041 p = ldns_pkt_query_new(zone_rdf, LDNS_RR_TYPE_SOA, class, LDNS_RD);
00042 if (!p) {
00043 return NULL;
00044 }
00045 zone_rdf = NULL;
00046
00047 ldns_pkt_set_opcode(p, LDNS_PACKET_UPDATE);
00048
00049 ldns_rr_list_deep_free(p->_authority);
00050
00051 ldns_pkt_set_authority(p, ldns_rr_list_clone(up_rrlist));
00052
00053 ldns_update_set_upcount(p, ldns_rr_list_rr_count(up_rrlist));
00054
00055 if (pr_rrlist) {
00056 ldns_rr_list_deep_free(p->_answer);
00057 ldns_pkt_set_answer(p, ldns_rr_list_clone(pr_rrlist));
00058 ldns_update_set_prcount(p, ldns_rr_list_rr_count(pr_rrlist));
00059 }
00060
00061 if (ad_rrlist) {
00062 ldns_rr_list_deep_free(p->_additional);
00063 ldns_pkt_set_additional(p, ldns_rr_list_clone(ad_rrlist));
00064 ldns_update_set_adcount(p, ldns_rr_list_rr_count(ad_rrlist));
00065 }
00066 return p;
00067 }
00068
00069 ldns_status
00070 ldns_update_pkt_tsig_add(ldns_pkt *p, ldns_resolver *r)
00071 {
00072 #ifdef HAVE_SSL
00073 uint16_t fudge = 300;
00074 if (ldns_resolver_tsig_keyname(r) && ldns_resolver_tsig_keydata(r))
00075 return ldns_pkt_tsig_sign(p, ldns_resolver_tsig_keyname(r),
00076 ldns_resolver_tsig_keydata(r), fudge,
00077 ldns_resolver_tsig_algorithm(r), NULL);
00078 #else
00079
00080 p = p;
00081 r = r;
00082 #endif
00083
00084 return LDNS_STATUS_OK;
00085 }
00086
00087
00088
00089 ldns_status
00090 ldns_update_soa_mname(ldns_rdf *zone, ldns_resolver *r,
00091 ldns_rr_class class, ldns_rdf **mname)
00092 {
00093 ldns_rr *soa_rr;
00094 ldns_pkt *query, *resp;
00095
00096
00097 query = ldns_pkt_query_new(ldns_rdf_clone(zone), LDNS_RR_TYPE_SOA,
00098 class, LDNS_RD);
00099 if (!query) {
00100 return LDNS_STATUS_ERR;
00101 }
00102
00103 ldns_pkt_set_random_id(query);
00104 if (ldns_resolver_send_pkt(&resp, r, query) != LDNS_STATUS_OK) {
00105 ldns_pkt_free(query);
00106 return LDNS_STATUS_ERR;
00107 }
00108 ldns_pkt_free(query);
00109 if (!resp) {
00110 return LDNS_STATUS_ERR;
00111 }
00112
00113
00114 *mname = NULL;
00115 while ((soa_rr = ldns_rr_list_pop_rr(ldns_pkt_answer(resp)))) {
00116 if (ldns_rr_get_type(soa_rr) != LDNS_RR_TYPE_SOA)
00117 continue;
00118
00119 *mname = ldns_rdf_clone(ldns_rr_rdf(soa_rr, 0));
00120 break;
00121 }
00122 ldns_pkt_free(resp);
00123
00124 return *mname ? LDNS_STATUS_OK : LDNS_STATUS_ERR;
00125 }
00126
00127
00128 ldns_status
00129 ldns_update_soa_zone_mname(const char *fqdn, ldns_resolver *r,
00130 ldns_rr_class class, ldns_rdf **zone_rdf, ldns_rdf **mname_rdf)
00131 {
00132 ldns_rr *soa_rr, *rr;
00133 ldns_rdf *soa_zone = NULL, *soa_mname = NULL;
00134 ldns_rdf *ipaddr, *fqdn_rdf, *tmp;
00135 ldns_rdf **nslist;
00136 ldns_pkt *query, *resp;
00137 size_t i;
00138
00139
00140
00141
00142
00143
00144
00145 fqdn_rdf = ldns_dname_new_frm_str(fqdn);
00146 query = ldns_pkt_query_new(fqdn_rdf, LDNS_RR_TYPE_SOA, class, LDNS_RD);
00147 if (!query) {
00148 return LDNS_STATUS_ERR;
00149 }
00150 fqdn_rdf = NULL;
00151
00152 ldns_pkt_set_random_id(query);
00153 if (ldns_resolver_send_pkt(&resp, r, query) != LDNS_STATUS_OK) {
00154 ldns_pkt_free(query);
00155 return LDNS_STATUS_ERR;
00156 }
00157 ldns_pkt_free(query);
00158 if (!resp) {
00159 return LDNS_STATUS_ERR;
00160 }
00161
00162
00163 while ((soa_rr = ldns_rr_list_pop_rr(ldns_pkt_authority(resp)))) {
00164 if (ldns_rr_get_type(soa_rr) != LDNS_RR_TYPE_SOA)
00165 continue;
00166
00167 soa_mname = ldns_rdf_clone(ldns_rr_rdf(soa_rr, 0));
00168 break;
00169 }
00170 ldns_pkt_free(resp);
00171 if (!soa_rr) {
00172 return LDNS_STATUS_ERR;
00173 }
00174
00175
00176 query = ldns_pkt_query_new(soa_mname, LDNS_RR_TYPE_A, class, LDNS_RD);
00177 if (!query) {
00178 return LDNS_STATUS_ERR;
00179 }
00180 soa_mname = NULL;
00181
00182 ldns_pkt_set_random_id(query);
00183 if (ldns_resolver_send_pkt(&resp, r, query) != LDNS_STATUS_OK) {
00184 ldns_pkt_free(query);
00185 return LDNS_STATUS_ERR;
00186 }
00187 ldns_pkt_free(query);
00188 if (!resp) {
00189 return LDNS_STATUS_ERR;
00190 }
00191
00192 if (ldns_pkt_ancount(resp) == 0) {
00193 ldns_pkt_free(resp);
00194 return LDNS_STATUS_ERR;
00195 }
00196
00197
00198 rr = ldns_rr_list_pop_rr(ldns_pkt_answer(resp));
00199 ipaddr = ldns_rr_rdf(rr, 0);
00200
00201
00202 nslist = ldns_resolver_nameservers(r);
00203 for (i = 0; i < ldns_resolver_nameserver_count(r); i++) {
00204 if (ldns_rdf_compare(ipaddr, nslist[i]) == 0) {
00205 if (i) {
00206 tmp = nslist[0];
00207 nslist[0] = nslist[i];
00208 nslist[i] = tmp;
00209 }
00210 break;
00211 }
00212 }
00213 if (i >= ldns_resolver_nameserver_count(r)) {
00214
00215 (void) ldns_resolver_push_nameserver(r, ipaddr);
00216 nslist = ldns_resolver_nameservers(r);
00217 i = ldns_resolver_nameserver_count(r) - 1;
00218 tmp = nslist[0];
00219 nslist[0] = nslist[i];
00220 nslist[i] = tmp;
00221 }
00222 ldns_pkt_free(resp);
00223
00224
00225 ldns_resolver_set_random(r, false);
00226
00227
00228 fqdn_rdf = ldns_dname_new_frm_str(fqdn);
00229 query = ldns_pkt_query_new(fqdn_rdf, LDNS_RR_TYPE_SOA, class, LDNS_RD);
00230 if (!query) {
00231 return LDNS_STATUS_ERR;
00232 }
00233 fqdn_rdf = NULL;
00234
00235 ldns_pkt_set_random_id(query);
00236 if (ldns_resolver_send_pkt(&resp, r, query) != LDNS_STATUS_OK) {
00237 ldns_pkt_free(query);
00238 return LDNS_STATUS_ERR;
00239 }
00240 ldns_pkt_free(query);
00241 if (!resp) {
00242 return LDNS_STATUS_ERR;
00243 }
00244
00245
00246 while ((soa_rr = ldns_rr_list_pop_rr(ldns_pkt_authority(resp)))) {
00247 if (ldns_rr_get_type(soa_rr) != LDNS_RR_TYPE_SOA)
00248 continue;
00249
00250 soa_mname = ldns_rdf_clone(ldns_rr_rdf(soa_rr, 0));
00251 soa_zone = ldns_rdf_clone(ldns_rr_owner(soa_rr));
00252 break;
00253 }
00254 ldns_pkt_free(resp);
00255 if (!soa_rr) {
00256 return LDNS_STATUS_ERR;
00257 }
00258
00259
00260 *zone_rdf = soa_zone;
00261 *mname_rdf = soa_mname;
00262 return LDNS_STATUS_OK;
00263 }
00264
00265
00266
00267
00268
00269 uint16_t
00270 ldns_update_zocount(const ldns_pkt *p)
00271 {
00272 return ldns_pkt_qdcount(p);
00273 }
00274
00275 uint16_t
00276 ldns_update_prcount(const ldns_pkt *p)
00277 {
00278 return ldns_pkt_ancount(p);
00279 }
00280
00281 uint16_t
00282 ldns_update_upcount(const ldns_pkt *p)
00283 {
00284 return ldns_pkt_nscount(p);
00285 }
00286
00287 uint16_t
00288 ldns_update_ad(const ldns_pkt *p)
00289 {
00290 return ldns_pkt_arcount(p);
00291 }
00292
00293 void
00294 ldns_update_set_zo(ldns_pkt *p, uint16_t v)
00295 {
00296 ldns_pkt_set_qdcount(p, v);
00297 }
00298
00299 void
00300 ldns_update_set_prcount(ldns_pkt *p, uint16_t v)
00301 {
00302 ldns_pkt_set_ancount(p, v);
00303 }
00304
00305 void
00306 ldns_update_set_upcount(ldns_pkt *p, uint16_t v)
00307 {
00308 ldns_pkt_set_nscount(p, v);
00309 }
00310
00311 void
00312 ldns_update_set_adcount(ldns_pkt *p, uint16_t v)
00313 {
00314 ldns_pkt_set_arcount(p, v);
00315 }