From 37a3a03e3347e5f25068201b3150183bca148da3 Mon Sep 17 00:00:00 2001
From: chenyong <1521761801@qq.com>
Date: Thu, 29 Nov 2018 18:55:14 +0800
Subject: [PATCH] [net][at] Fix at_recvfrom() select event issue

Signed-off-by: chenyong <1521761801@qq.com>
---
 components/net/at/at_socket/at_socket.c | 3 +++
 1 file changed, 3 insertions(+)

diff --git a/components/net/at/at_socket/at_socket.c b/components/net/at/at_socket/at_socket.c
index 01646f0d89..84325c1501 100644
--- a/components/net/at/at_socket/at_socket.c
+++ b/components/net/at/at_socket/at_socket.c
@@ -622,6 +622,9 @@ int at_recvfrom(int socket, void *mem, size_t len, int flags, struct sockaddr *f
             goto __exit;
         }
         sock->state = AT_SOCKET_CONNECT;
+        /* set AT socket receive data callback function */
+        at_dev_ops->at_set_event_cb(AT_SOCKET_EVT_RECV, at_recv_notice_cb);
+        at_dev_ops->at_set_event_cb(AT_SOCKET_EVT_CLOSED, at_closed_notice_cb);
     }
 
     /* socket passively closed, receive function return 0 */