diff options
Diffstat (limited to 'source4/libcli/dgram/dgramsocket.c')
| -rw-r--r-- | source4/libcli/dgram/dgramsocket.c | 12 |
1 files changed, 8 insertions, 4 deletions
diff --git a/source4/libcli/dgram/dgramsocket.c b/source4/libcli/dgram/dgramsocket.c index a3909df768..88eed5c10b 100644 --- a/source4/libcli/dgram/dgramsocket.c +++ b/source4/libcli/dgram/dgramsocket.c @@ -27,8 +27,6 @@ #include "libcli/dgram/libdgram.h" #include "lib/socket/socket.h" -#define DGRAM_MAX_PACKET_SIZE 2048 - /* handle recv events on a nbt dgram socket @@ -40,11 +38,17 @@ static void dgm_socket_recv(struct nbt_dgram_socket *dgmsock) const char *src_addr; int src_port; DATA_BLOB blob; - size_t nread; + size_t nread, dsize; struct nbt_dgram_packet *packet; const char *mailslot_name; - blob = data_blob_talloc(tmp_ctx, NULL, DGRAM_MAX_PACKET_SIZE); + status = socket_pending(dgmsock->sock, &dsize); + if (!NT_STATUS_IS_OK(status)) { + talloc_free(tmp_ctx); + return; + } + + blob = data_blob_talloc(tmp_ctx, NULL, dsize); if (blob.data == NULL) { talloc_free(tmp_ctx); return; |