Changes in uspace/srv/net/dnsrsrv/dnsrsrv.c [5d1cb8a:959d2ec] in mainline

File:

• uspace/srv/net/dnsrsrv/dnsrsrv.c (modified) (2 diffs)

uspace/srv/net/dnsrsrv/dnsrsrv.c

@@ -89 +89 @@
         char *name;
         dns_host_info_t *hinfo;
+        ipc_callid_t rcallid;
+        size_t size;
+        sysarg_t retval;
+        size_t act_size;
         int rc;
 
@@ -100 +104 @@
         }
 
+        if (!async_data_read_receive(&rcallid, &size)) {
+                async_answer_0(rcallid, EREFUSED);
+                async_answer_0(callid, EREFUSED);
+                return;
+        }
+
         rc = dns_name2host(name, &hinfo);
         if (rc != EOK) {
+                async_answer_0(rcallid, rc);
                 async_answer_0(callid, rc);
                 return;
         }
 
-        async_answer_1(callid, EOK, hinfo->addr.ipv4);
+        act_size = str_size(hinfo->cname);
+        if (act_size > size) {
+                async_answer_0(rcallid, EOVERFLOW);
+                async_answer_0(callid, EOVERFLOW);
+                return;
+        }
+
+        retval = async_data_read_finalize(rcallid, hinfo->cname, act_size);
+        async_answer_1(callid, retval, hinfo->addr.ipv4);
 
         dns_hostinfo_destroy(hinfo);