1 /******************************************************************************
3 * Copyright (C) 2009-2012 Broadcom Corporation
5 * Licensed under the Apache License, Version 2.0 (the "License");
6 * you may not use this file except in compliance with the License.
7 * You may obtain a copy of the License at:
9 * http://www.apache.org/licenses/LICENSE-2.0
11 * Unless required by applicable law or agreed to in writing, software
12 * distributed under the License is distributed on an "AS IS" BASIS,
13 * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
14 * See the License for the specific language governing permissions and
15 * limitations under the License.
17 ******************************************************************************/
19 /************************************************************************************
21 * Filename: btif_sock_thread.c
23 * Description: socket select thread
26 ***********************************************************************************/
28 #include <hardware/bluetooth.h>
29 #include <hardware/bt_sock.h>
37 #include <sys/types.h>
38 #include <sys/socket.h>
47 #include <sys/select.h>
49 #include <cutils/sockets.h>
52 #define LOG_TAG "bt_btif_sock"
53 #include "btif_common.h"
54 #include "btif_util.h"
58 #include "btif_sock.h"
59 #include "btif_sock_thread.h"
60 #include "btif_sock_util.h"
62 #define asrt(s) if(!(s)) APPL_TRACE_ERROR("## %s assert %s failed at line:%d ##",__FUNCTION__, #s, __LINE__)
63 #define print_events(events) do { \
64 APPL_TRACE_DEBUG("print poll event:%x", events); \
65 if (events & POLLIN) APPL_TRACE_DEBUG( " POLLIN "); \
66 if (events & POLLPRI) APPL_TRACE_DEBUG( " POLLPRI "); \
67 if (events & POLLOUT) APPL_TRACE_DEBUG( " POLLOUT "); \
68 if (events & POLLERR) APPL_TRACE_DEBUG( " POLLERR "); \
69 if (events & POLLHUP) APPL_TRACE_DEBUG( " POLLHUP "); \
70 if (events & POLLNVAL) APPL_TRACE_DEBUG(" POLLNVAL "); \
71 if (events & POLLRDHUP) APPL_TRACE_DEBUG(" POLLRDHUP"); \
76 #define POLL_EXCEPTION_EVENTS (POLLHUP | POLLRDHUP | POLLERR | POLLNVAL)
77 #define IS_EXCEPTION(e) ((e) & POLL_EXCEPTION_EVENTS)
78 #define IS_READ(e) ((e) & POLLIN)
79 #define IS_WRITE(e) ((e) & POLLOUT)
80 /*cmd executes in socket poll thread */
84 #define CMD_REMOVE_FD 4
85 #define CMD_USER_PRIVATE 5
96 poll_slot_t ps[MAX_POLL];
97 int psi[MAX_POLL]; //index of poll slot
98 volatile pthread_t thread_id;
99 btsock_signaled_cb callback;
100 btsock_cmd_cb cmd_callback;
103 static thread_slot_t ts[MAX_THREAD];
107 static void *sock_poll_thread(void *arg);
108 static inline void close_cmd_fd(int h);
110 static inline void add_poll(int h, int fd, int type, int flags, uint32_t user_id);
112 static pthread_mutex_t thread_slot_lock = PTHREAD_RECURSIVE_MUTEX_INITIALIZER_NP;
114 static inline void set_socket_blocking(int s, int blocking)
117 opts = fcntl(s, F_GETFL);
118 if (opts<0) APPL_TRACE_ERROR("set blocking (%s)", strerror(errno));
121 else opts |= O_NONBLOCK;
122 if (fcntl(s, F_SETFL, opts) < 0)
123 APPL_TRACE_ERROR("set blocking (%s)", strerror(errno));
126 static inline int create_server_socket(const char* name)
128 int s = socket(AF_LOCAL, SOCK_STREAM, 0);
131 APPL_TRACE_DEBUG("covert name to android abstract name:%s", name);
132 if(socket_local_server_bind(s, name, ANDROID_SOCKET_NAMESPACE_ABSTRACT) >= 0)
134 if(listen(s, 5) == 0)
136 APPL_TRACE_DEBUG("listen to local socket:%s, fd:%d", name, s);
139 else APPL_TRACE_ERROR("listen to local socket:%s, fd:%d failed, errno:%d", name, s, errno);
141 else APPL_TRACE_ERROR("create local socket:%s fd:%d, failed, errno:%d", name, s, errno);
145 static inline int connect_server_socket(const char* name)
147 int s = socket(AF_LOCAL, SOCK_STREAM, 0);
150 set_socket_blocking(s, TRUE);
151 if(socket_local_client_connect(s, name, ANDROID_SOCKET_NAMESPACE_ABSTRACT, SOCK_STREAM) >= 0)
153 APPL_TRACE_DEBUG("connected to local socket:%s, fd:%d", name, s);
156 else APPL_TRACE_ERROR("connect to local socket:%s, fd:%d failed, errno:%d", name, s, errno);
160 static inline int accept_server_socket(int s)
162 struct sockaddr_un client_address;
164 int fd = accept(s, (struct sockaddr*)&client_address, &clen);
165 APPL_TRACE_DEBUG("accepted fd:%d for server fd:%d", fd, s);
169 static inline int create_thread(void *(*start_routine)(void *), void * arg,
170 pthread_t * thread_id)
172 pthread_attr_t thread_attr;
173 pthread_attr_init(&thread_attr);
174 pthread_attr_setdetachstate(&thread_attr, PTHREAD_CREATE_JOINABLE);
178 struct sched_param param;
180 if ((ret = pthread_create(thread_id, &thread_attr, start_routine, arg))!=0 )
182 APPL_TRACE_ERROR("pthread_create : %s", strerror(errno));
185 /* We need to lower the priority of this thread to ensure the stack gets
186 * priority over transfer to a socket */
187 pthread_getschedparam(*thread_id, &policy, ¶m);
188 min_pri = sched_get_priority_min(policy);
189 if (param.sched_priority > min_pri) {
190 param.sched_priority -= 1;
192 pthread_setschedparam(*thread_id, policy, ¶m);
195 static void init_poll(int cmd_fd);
196 static int alloc_thread_slot()
199 //revserd order to save guard uninitialized access to 0 index
200 for(i = MAX_THREAD - 1; i >=0; i--)
202 APPL_TRACE_DEBUG("ts[%d].used:%d", i, ts[i].used);
209 APPL_TRACE_ERROR("execeeded max thread count");
212 static void free_thread_slot(int h)
214 if(0 <= h && h < MAX_THREAD)
219 else APPL_TRACE_ERROR("invalid thread handle:%d", h);
221 int btsock_thread_init()
223 static int initialized;
224 APPL_TRACE_DEBUG("in initialized:%d", initialized);
229 for(h = 0; h < MAX_THREAD; h++)
231 ts[h].cmd_fdr = ts[h].cmd_fdw = -1;
233 ts[h].thread_id = -1;
234 ts[h].poll_count = 0;
235 ts[h].callback = NULL;
236 ts[h].cmd_callback = NULL;
241 int btsock_thread_create(btsock_signaled_cb callback, btsock_cmd_cb cmd_callback)
243 asrt(callback || cmd_callback);
244 pthread_mutex_lock(&thread_slot_lock);
245 int h = alloc_thread_slot();
246 pthread_mutex_unlock(&thread_slot_lock);
247 APPL_TRACE_DEBUG("alloc_thread_slot ret:%d", h);
252 int status = create_thread(sock_poll_thread, (void*)(uintptr_t)h, &thread);
255 APPL_TRACE_ERROR("create_thread failed: %s", strerror(status));
260 ts[h].thread_id = thread;
261 APPL_TRACE_DEBUG("h:%d, thread id:%d", h, ts[h].thread_id);
262 ts[h].callback = callback;
263 ts[h].cmd_callback = cmd_callback;
268 /* create dummy socket pair used to wake up select loop */
269 static inline void init_cmd_fd(int h)
271 asrt(ts[h].cmd_fdr == -1 && ts[h].cmd_fdw == -1);
272 if(socketpair(AF_UNIX, SOCK_STREAM, 0, &ts[h].cmd_fdr) < 0)
274 APPL_TRACE_ERROR("socketpair failed: %s", strerror(errno));
277 APPL_TRACE_DEBUG("h:%d, cmd_fdr:%d, cmd_fdw:%d", h, ts[h].cmd_fdr, ts[h].cmd_fdw);
278 //add the cmd fd for read & write
279 add_poll(h, ts[h].cmd_fdr, 0, SOCK_THREAD_FD_RD, 0);
281 static inline void close_cmd_fd(int h)
283 if(ts[h].cmd_fdr != -1)
285 close(ts[h].cmd_fdr);
288 if(ts[h].cmd_fdw != -1)
290 close(ts[h].cmd_fdw);
302 int btsock_thread_add_fd(int h, int fd, int type, int flags, uint32_t user_id)
304 if(h < 0 || h >= MAX_THREAD)
306 APPL_TRACE_ERROR("invalid bt thread handle:%d", h);
309 if(ts[h].cmd_fdw == -1)
311 APPL_TRACE_ERROR("cmd socket is not created. socket thread may not initialized");
314 if(flags & SOCK_THREAD_ADD_FD_SYNC)
316 //must executed in socket poll thread
317 if(ts[h].thread_id == pthread_self())
319 //cleanup one-time flags
320 flags &= ~SOCK_THREAD_ADD_FD_SYNC;
321 add_poll(h, fd, type, flags, user_id);
324 APPL_TRACE_DEBUG("THREAD_ADD_FD_SYNC is not called in poll thread, fallback to async");
326 sock_cmd_t cmd = {CMD_ADD_FD, fd, type, flags, user_id};
327 APPL_TRACE_DEBUG("adding fd:%d, flags:0x%x", fd, flags);
328 return send(ts[h].cmd_fdw, &cmd, sizeof(cmd), 0) == sizeof(cmd);
331 bool btsock_thread_remove_fd_and_close(int thread_handle, int fd)
333 if (thread_handle < 0 || thread_handle >= MAX_THREAD)
335 APPL_TRACE_ERROR("%s invalid thread handle: %d", __func__, thread_handle);
340 APPL_TRACE_ERROR("%s invalid file descriptor.", __func__);
344 sock_cmd_t cmd = {CMD_REMOVE_FD, fd, 0, 0, 0};
345 return send(ts[thread_handle].cmd_fdw, &cmd, sizeof(cmd), 0) == sizeof(cmd);
348 int btsock_thread_post_cmd(int h, int type, const unsigned char* data, int size, uint32_t user_id)
350 if(h < 0 || h >= MAX_THREAD)
352 APPL_TRACE_ERROR("invalid bt thread handle:%d", h);
355 if(ts[h].cmd_fdw == -1)
357 APPL_TRACE_ERROR("cmd socket is not created. socket thread may not initialized");
360 sock_cmd_t cmd = {CMD_USER_PRIVATE, 0, type, size, user_id};
361 APPL_TRACE_DEBUG("post cmd type:%d, size:%d, h:%d, ", type, size, h);
362 sock_cmd_t* cmd_send = &cmd;
363 int size_send = sizeof(cmd);
366 size_send = sizeof(cmd) + size;
367 cmd_send = (sock_cmd_t*)alloca(size_send);
371 memcpy(cmd_send + 1, data, size);
375 APPL_TRACE_ERROR("alloca failed at h:%d, cmd type:%d, size:%d", h, type, size_send);
379 return send(ts[h].cmd_fdw, cmd_send, size_send, 0) == size_send;
381 int btsock_thread_wakeup(int h)
383 if(h < 0 || h >= MAX_THREAD)
385 APPL_TRACE_ERROR("invalid bt thread handle:%d", h);
388 if(ts[h].cmd_fdw == -1)
390 APPL_TRACE_ERROR("thread handle:%d, cmd socket is not created", h);
393 sock_cmd_t cmd = {CMD_WAKEUP, 0, 0, 0, 0};
394 return send(ts[h].cmd_fdw, &cmd, sizeof(cmd), 0) == sizeof(cmd);
396 int btsock_thread_exit(int h)
398 if(h < 0 || h >= MAX_THREAD)
400 APPL_TRACE_ERROR("invalid bt thread handle:%d", h);
403 if(ts[h].cmd_fdw == -1)
405 APPL_TRACE_ERROR("cmd socket is not created");
408 sock_cmd_t cmd = {CMD_EXIT, 0, 0, 0, 0};
409 if(send(ts[h].cmd_fdw, &cmd, sizeof(cmd), 0) == sizeof(cmd))
411 pthread_join(ts[h].thread_id, 0);
412 pthread_mutex_lock(&thread_slot_lock);
414 pthread_mutex_unlock(&thread_slot_lock);
419 static void init_poll(int h)
422 ts[h].poll_count = 0;
423 ts[h].thread_id = -1;
424 ts[h].callback = NULL;
425 ts[h].cmd_callback = NULL;
426 for(i = 0; i < MAX_POLL; i++)
428 ts[h].ps[i].pfd.fd = -1;
433 static inline unsigned int flags2pevents(int flags)
435 unsigned int pevents = 0;
436 if(flags & SOCK_THREAD_FD_WR)
438 if(flags & SOCK_THREAD_FD_RD)
440 pevents |= POLL_EXCEPTION_EVENTS;
444 static inline void set_poll(poll_slot_t* ps, int fd, int type, int flags, uint32_t user_id)
447 ps->user_id = user_id;
448 if(ps->type != 0 && ps->type != type)
449 APPL_TRACE_ERROR("poll socket type should not changed! type was:%d, type now:%d", ps->type, type);
452 ps->pfd.events = flags2pevents(flags);
455 static inline void add_poll(int h, int fd, int type, int flags, uint32_t user_id)
460 poll_slot_t* ps = ts[h].ps;
462 for(i = 0; i < MAX_POLL; i++)
464 if(ps[i].pfd.fd == fd)
466 asrt(ts[h].poll_count < MAX_POLL);
468 set_poll(&ps[i], fd, type, flags | ps[i].flags, user_id);
471 else if(empty < 0 && ps[i].pfd.fd == -1)
476 asrt(ts[h].poll_count < MAX_POLL);
477 set_poll(&ps[empty], fd, type, flags, user_id);
481 APPL_TRACE_ERROR("exceeded max poll slot:%d!", MAX_POLL);
483 static inline void remove_poll(int h, poll_slot_t* ps, int flags)
485 if(flags == ps->flags)
487 //all monitored events signaled. To remove it, just clear the slot
489 memset(ps, 0, sizeof(*ps));
494 //one read or one write monitor event signaled, removed the accordding bit
496 //update the poll events mask
497 ps->pfd.events = flags2pevents(ps->flags);
500 static int process_cmd_sock(int h)
502 sock_cmd_t cmd = {-1, 0, 0, 0, 0};
503 int fd = ts[h].cmd_fdr;
504 if(recv(fd, &cmd, sizeof(cmd), MSG_WAITALL) != sizeof(cmd))
506 APPL_TRACE_ERROR("recv cmd errno:%d", errno);
509 APPL_TRACE_DEBUG("cmd.id:%d", cmd.id);
513 add_poll(h, cmd.fd, cmd.type, cmd.flags, cmd.user_id);
516 for (int i = 1; i < MAX_POLL; ++i)
518 poll_slot_t *poll_slot = &ts[h].ps[i];
519 if (poll_slot->pfd.fd == cmd.fd)
521 remove_poll(h, poll_slot, poll_slot->flags);
529 case CMD_USER_PRIVATE:
530 asrt(ts[h].cmd_callback);
531 if(ts[h].cmd_callback)
532 ts[h].cmd_callback(fd, cmd.type, cmd.flags, cmd.user_id);
537 APPL_TRACE_DEBUG("unknown cmd: %d", cmd.id);
542 static void process_data_sock(int h, struct pollfd *pfds, int count)
544 asrt(count <= ts[h].poll_count);
546 for( i= 1; i < ts[h].poll_count; i++)
550 int ps_i = ts[h].psi[i];
551 asrt(pfds[i].fd == ts[h].ps[ps_i].pfd.fd);
552 uint32_t user_id = ts[h].ps[ps_i].user_id;
553 int type = ts[h].ps[ps_i].type;
555 print_events(pfds[i].revents);
556 if(IS_READ(pfds[i].revents))
558 flags |= SOCK_THREAD_FD_RD;
560 if(IS_WRITE(pfds[i].revents))
562 flags |= SOCK_THREAD_FD_WR;
564 if(IS_EXCEPTION(pfds[i].revents))
566 flags |= SOCK_THREAD_FD_EXCEPTION;
567 //remove the whole slot not flags
568 remove_poll(h, &ts[h].ps[ps_i], ts[h].ps[ps_i].flags);
571 remove_poll(h, &ts[h].ps[ps_i], flags); //remove the monitor flags that already processed
573 ts[h].callback(pfds[i].fd, type, flags, user_id);
578 static void prepare_poll_fds(int h, struct pollfd* pfds)
583 asrt(ts[h].poll_count <= MAX_POLL);
584 memset(pfds, 0, sizeof(pfds[0])*ts[h].poll_count);
585 while(count < ts[h].poll_count)
589 APPL_TRACE_ERROR("exceed max poll range, ps_i:%d, MAX_POLL:%d, count:%d, ts[h].poll_count:%d",
590 ps_i, MAX_POLL, count, ts[h].poll_count);
593 if(ts[h].ps[ps_i].pfd.fd >= 0)
595 pfds[pfd_i] = ts[h].ps[ps_i].pfd;
596 ts[h].psi[pfd_i] = ps_i;
603 static void *sock_poll_thread(void *arg)
605 struct pollfd pfds[MAX_POLL];
606 memset(pfds, 0, sizeof(pfds));
607 int h = (intptr_t)arg;
610 prepare_poll_fds(h, pfds);
611 int ret = poll(pfds, ts[h].poll_count, -1);
614 APPL_TRACE_ERROR("poll ret -1, exit the thread, errno:%d, err:%s", errno, strerror(errno));
619 int need_process_data_fd = TRUE;
620 if(pfds[0].revents) //cmd fd always is the first one
622 asrt(pfds[0].fd == ts[h].cmd_fdr);
623 if(!process_cmd_sock(h))
625 APPL_TRACE_DEBUG("h:%d, process_cmd_sock return false, exit...", h);
629 need_process_data_fd = FALSE;
630 else ret--; //exclude the cmd fd
632 if(need_process_data_fd)
633 process_data_sock(h, pfds, ret);
635 else {APPL_TRACE_DEBUG("no data, select ret: %d", ret)};
637 ts[h].thread_id = -1;
638 APPL_TRACE_DEBUG("socket poll thread exiting, h:%d", h);