48 	dns_tcpmsg_t *tcpmsg = ev_in->ev_arg;  in recv_length()  local
52 	INSIST(VALID_TCPMSG(tcpmsg));  in recv_length()
54 	dev = &tcpmsg->event;  in recv_length()
55 	tcpmsg->address = ev->address;  in recv_length()
58 		tcpmsg->result = ev->result;  in recv_length()
65 	tcpmsg->size = ntohs(tcpmsg->size);  in recv_length()
66 	if (tcpmsg->size == 0) {  in recv_length()
67 		tcpmsg->result = ISC_R_UNEXPECTEDEND;  in recv_length()
70 	if (tcpmsg->size > tcpmsg->maxsize) {  in recv_length()
71 		tcpmsg->result = ISC_R_RANGE;  in recv_length()
75 	region.base = isc_mem_get(tcpmsg->mctx, tcpmsg->size);  in recv_length()
76 	region.length = tcpmsg->size;  in recv_length()
78 		tcpmsg->result = ISC_R_NOMEMORY;  in recv_length()
81 	XDEBUG(("Allocated %d bytes\n", tcpmsg->size));  in recv_length()
83 	isc_buffer_init(&tcpmsg->buffer, region.base, region.length);  in recv_length()
84 	result = isc_socket_recv(tcpmsg->sock, &region, 0, task, recv_message,  in recv_length()
85 				 tcpmsg);  in recv_length()
87 		tcpmsg->result = result;  in recv_length()
95 	isc_task_send(tcpmsg->task, &dev);  in recv_length()
96 	tcpmsg->task = NULL;  in recv_length()
105 	dns_tcpmsg_t *tcpmsg = ev_in->ev_arg;  in recv_message()  local
109 	INSIST(VALID_TCPMSG(tcpmsg));  in recv_message()
111 	dev = &tcpmsg->event;  in recv_message()
112 	tcpmsg->address = ev->address;  in recv_message()
115 		tcpmsg->result = ev->result;  in recv_message()
119 	tcpmsg->result = ISC_R_SUCCESS;  in recv_message()
120 	isc_buffer_add(&tcpmsg->buffer, ev->n);  in recv_message()
122 	XDEBUG(("Received %u bytes (of %d)\n", ev->n, tcpmsg->size));  in recv_message()
125 	isc_task_send(tcpmsg->task, &dev);  in recv_message()
126 	tcpmsg->task = NULL;  in recv_message()
131 dns_tcpmsg_init(isc_mem_t *mctx, isc_socket_t *sock, dns_tcpmsg_t *tcpmsg) {  in dns_tcpmsg_init()  argument
134 	REQUIRE(tcpmsg != NULL);  in dns_tcpmsg_init()
136 	tcpmsg->magic = TCPMSG_MAGIC;  in dns_tcpmsg_init()
137 	tcpmsg->size = 0;  in dns_tcpmsg_init()
138 	tcpmsg->buffer.base = NULL;  in dns_tcpmsg_init()
139 	tcpmsg->buffer.length = 0;  in dns_tcpmsg_init()
140 	tcpmsg->maxsize = 65535; /* Largest message possible. */  in dns_tcpmsg_init()
141 	tcpmsg->mctx = mctx;  in dns_tcpmsg_init()
142 	tcpmsg->sock = sock;  in dns_tcpmsg_init()
143 	tcpmsg->task = NULL;		   /* None yet. */  in dns_tcpmsg_init()
144 	tcpmsg->result = ISC_R_UNEXPECTED; /* None yet. */  in dns_tcpmsg_init()
150 dns_tcpmsg_setmaxsize(dns_tcpmsg_t *tcpmsg, unsigned int maxsize) {  in dns_tcpmsg_setmaxsize()  argument
151 	REQUIRE(VALID_TCPMSG(tcpmsg));  in dns_tcpmsg_setmaxsize()
154 	tcpmsg->maxsize = maxsize;  in dns_tcpmsg_setmaxsize()
158 dns_tcpmsg_readmessage(dns_tcpmsg_t *tcpmsg, isc_task_t *task,  in dns_tcpmsg_readmessage()  argument
163 	REQUIRE(VALID_TCPMSG(tcpmsg));  in dns_tcpmsg_readmessage()
165 	REQUIRE(tcpmsg->task == NULL); /* not currently in use */  in dns_tcpmsg_readmessage()
167 	if (tcpmsg->buffer.base != NULL) {  in dns_tcpmsg_readmessage()
168 		isc_mem_put(tcpmsg->mctx, tcpmsg->buffer.base,  in dns_tcpmsg_readmessage()
169 			    tcpmsg->buffer.length);  in dns_tcpmsg_readmessage()
170 		tcpmsg->buffer.base = NULL;  in dns_tcpmsg_readmessage()
171 		tcpmsg->buffer.length = 0;  in dns_tcpmsg_readmessage()
174 	tcpmsg->task = task;  in dns_tcpmsg_readmessage()
175 	tcpmsg->action = action;  in dns_tcpmsg_readmessage()
176 	tcpmsg->arg = arg;  in dns_tcpmsg_readmessage()
177 	tcpmsg->result = ISC_R_UNEXPECTED; /* unknown right now */  in dns_tcpmsg_readmessage()
179 	ISC_EVENT_INIT(&tcpmsg->event, sizeof(isc_event_t), 0, 0,  in dns_tcpmsg_readmessage()
180 		       DNS_EVENT_TCPMSG, action, arg, tcpmsg, NULL, NULL);  in dns_tcpmsg_readmessage()
182 	region.base = (unsigned char *)&tcpmsg->size;  in dns_tcpmsg_readmessage()
184 	result = isc_socket_recv(tcpmsg->sock, &region, 0, tcpmsg->task,  in dns_tcpmsg_readmessage()
185 				 recv_length, tcpmsg);  in dns_tcpmsg_readmessage()
188 		tcpmsg->task = NULL;  in dns_tcpmsg_readmessage()
195 dns_tcpmsg_cancelread(dns_tcpmsg_t *tcpmsg) {  in dns_tcpmsg_cancelread()  argument
196 	REQUIRE(VALID_TCPMSG(tcpmsg));  in dns_tcpmsg_cancelread()
198 	isc_socket_cancel(tcpmsg->sock, NULL, ISC_SOCKCANCEL_RECV);  in dns_tcpmsg_cancelread()
202 dns_tcpmsg_keepbuffer(dns_tcpmsg_t *tcpmsg, isc_buffer_t *buffer) {  in dns_tcpmsg_keepbuffer()  argument
203 	REQUIRE(VALID_TCPMSG(tcpmsg));  in dns_tcpmsg_keepbuffer()
206 	*buffer = tcpmsg->buffer;  in dns_tcpmsg_keepbuffer()
207 	tcpmsg->buffer.base = NULL;  in dns_tcpmsg_keepbuffer()
208 	tcpmsg->buffer.length = 0;  in dns_tcpmsg_keepbuffer()
213 dns_tcpmsg_freebuffer(dns_tcpmsg_t *tcpmsg) {
214 	REQUIRE(VALID_TCPMSG(tcpmsg));
216 	if (tcpmsg->buffer.base == NULL) {
220 	isc_mem_put(tcpmsg->mctx, tcpmsg->buffer.base, tcpmsg->buffer.length);
221 	tcpmsg->buffer.base = NULL;
222 	tcpmsg->buffer.length = 0;
227 dns_tcpmsg_invalidate(dns_tcpmsg_t *tcpmsg) {  in dns_tcpmsg_invalidate()  argument
228 	REQUIRE(VALID_TCPMSG(tcpmsg));  in dns_tcpmsg_invalidate()
230 	tcpmsg->magic = 0;  in dns_tcpmsg_invalidate()
232 	if (tcpmsg->buffer.base != NULL) {  in dns_tcpmsg_invalidate()
233 		isc_mem_put(tcpmsg->mctx, tcpmsg->buffer.base,  in dns_tcpmsg_invalidate()
234 			    tcpmsg->buffer.length);  in dns_tcpmsg_invalidate()
235 		tcpmsg->buffer.base = NULL;  in dns_tcpmsg_invalidate()
236 		tcpmsg->buffer.length = 0;  in dns_tcpmsg_invalidate()