extended_error_encode() for extended errors

diff --git a/daemon/worker.c b/daemon/worker.c
index 29c1961edeb5318bb0d7dbded3da41451b5fee05..c2a94be79216b767829ba6166912b489f2fb41b5 100644 (file)
--- a/daemon/worker.c
+++ b/daemon/worker.c
@@ -1439,8 +1439,6 @@ worker_handle_request(struct comm_point* c, void* arg, int error,
                log_addr(VERB_CLIENT,"from",&repinfo->addr, repinfo->addrlen);
                memset(&reply_edns, 0, sizeof(reply_edns));
                reply_edns.edns_present = 1;
-               reply_edns.udp_size = EDNS_ADVERTISED_SIZE;
-               LDNS_RCODE_SET(sldns_buffer_begin(c->buffer), ret);
                error_encode(c->buffer, ret, &qinfo,
                        *(uint16_t*)(void *)sldns_buffer_begin(c->buffer),
                        sldns_buffer_read_u16_at(c->buffer, 2), &reply_edns);
@@ -1449,22 +1447,14 @@ worker_handle_request(struct comm_point* c, void* arg, int error,
        }
        if(edns.edns_present) {
                if(edns.edns_version != 0) {
-                       edns.ext_rcode = (uint8_t)(EDNS_RCODE_BADVERS>>4);
-                       edns.edns_version = EDNS_ADVERTISED_VERSION;
-                       edns.udp_size = EDNS_ADVERTISED_SIZE;
-                       edns.bits &= EDNS_DO;
                        edns.opt_list_in = NULL;
                        edns.opt_list_out = NULL;
                        edns.opt_list_inplace_cb_out = NULL;
-                       edns.padding_block_size = 0;
                        verbose(VERB_ALGO, "query with bad edns version.");
                        log_addr(VERB_CLIENT,"from",&repinfo->addr, repinfo->addrlen);
-                       error_encode(c->buffer, EDNS_RCODE_BADVERS&0xf, &qinfo,
+                       extended_error_encode(c->buffer, EDNS_RCODE_BADVERS, &qinfo,
                                *(uint16_t*)(void *)sldns_buffer_begin(c->buffer),
-                               sldns_buffer_read_u16_at(c->buffer, 2), NULL);
-                       if(sldns_buffer_capacity(c->buffer) >=
-                          sldns_buffer_limit(c->buffer)+calc_edns_field_size(&edns))
-                               attach_edns_record(c->buffer, &edns);
+                               sldns_buffer_read_u16_at(c->buffer, 2), 0, &edns);
                        regional_free_all(worker->scratchpad);
                        goto send_reply;
                }

diff --git a/util/data/msgencode.c b/util/data/msgencode.c
index fe21cfb86bd11a46ef1b0a319f68d406bd340527..d832ddc91ddb02ae6931481051c156ce446125bb 100644 (file)
--- a/util/data/msgencode.c
+++ b/util/data/msgencode.c
@@ -959,14 +959,15 @@ qinfo_query_encode(sldns_buffer* pkt, struct query_info* qinfo)
 }
 
 void 
-error_encode(sldns_buffer* buf, int r, struct query_info* qinfo,
-       uint16_t qid, uint16_t qflags, struct edns_data* edns)
+extended_error_encode(sldns_buffer* buf, uint16_t rcode, struct query_info* qinfo,
+       uint16_t qid, uint16_t qflags, uint16_t xflags, struct edns_data* edns)
 {
        uint16_t flags;
 
        sldns_buffer_clear(buf);
        sldns_buffer_write(buf, &qid, sizeof(uint16_t));
-       flags = (uint16_t)(BIT_QR | BIT_RA | r); /* QR and retcode*/
+       flags = (uint16_t)(BIT_QR | BIT_RA | (rcode & 0xF)); /* QR and retcode*/
+       flags |= xflags;
        flags |= (qflags & (BIT_RD|BIT_CD)); /* copy RD and CD bit */
        sldns_buffer_write_u16(buf, flags);
        if(qinfo) flags = 1;
@@ -993,7 +994,7 @@ error_encode(sldns_buffer* buf, int r, struct query_info* qinfo,
                struct edns_data es = *edns;
                es.edns_version = EDNS_ADVERTISED_VERSION;
                es.udp_size = EDNS_ADVERTISED_SIZE;
-               es.ext_rcode = 0;
+               es.ext_rcode = (uint8_t)(rcode >> 4);
                es.bits &= EDNS_DO;
                if(sldns_buffer_limit(buf) + calc_edns_field_size(&es) >
                        edns->udp_size)
@@ -1001,3 +1002,12 @@ error_encode(sldns_buffer* buf, int r, struct query_info* qinfo,
                attach_edns_record(buf, &es);
        }
 }
+
+void 
+error_encode(sldns_buffer* buf, int r, struct query_info* qinfo,
+       uint16_t qid, uint16_t qflags, struct edns_data* edns)
+{
+       extended_error_encode(buf, (r & 0x000F), qinfo, qid, qflags
+                                , (r & 0xFFF0), edns);
+}
+

diff --git a/util/data/msgencode.h b/util/data/msgencode.h
index 30dc515cbe595eadcb0bf58b5ded21b32e55296c..1b37ca87092b8bb1a6778eb4dfdb793fe372287e 100644 (file)
--- a/util/data/msgencode.h
+++ b/util/data/msgencode.h
@@ -120,7 +120,7 @@ void attach_edns_record(struct sldns_buffer* pkt, struct edns_data* edns);
  * Encode an error. With QR and RA set.
  *
  * @param pkt: where to store the packet.
- * @param r: RCODE value to encode.
+ * @param r: RCODE value to encode (may contain extra flags).
  * @param qinfo: if not NULL, the query is included.
  * @param qid: query ID to set in packet. network order.
  * @param qflags: original query flags (to copy RD and CD bits). host order.
@@ -130,4 +130,21 @@ void attach_edns_record(struct sldns_buffer* pkt, struct edns_data* edns);
 void error_encode(struct sldns_buffer* pkt, int r, struct query_info* qinfo,
        uint16_t qid, uint16_t qflags, struct edns_data* edns);
 
+/** 
+ * Encode an extended error. With QR and RA set.
+ *
+ * @param pkt: where to store the packet.
+ * @param rcode: Extended RCODE value to encode.
+ * @param qinfo: if not NULL, the query is included.
+ * @param qid: query ID to set in packet. network order.
+ * @param qflags: original query flags (to copy RD and CD bits). host order.
+ * @param xflags: extra flags to set (such as for example BIT_AA and/or BIT_TC)
+ * @param edns: if not NULL, this is the query edns info,
+ *     and an edns reply is attached. Only attached if EDNS record fits reply.
+ *      Without edns extended errors (i.e. > 15 )will not be conveyed.
+ */
+void extended_error_encode(struct sldns_buffer* pkt, uint16_t rcode,
+       struct query_info* qinfo, uint16_t qid, uint16_t qflags,
+       uint16_t xflags, struct edns_data* edns);
+
 #endif /* UTIL_DATA_MSGENCODE_H */