1 /* Simple binding of nanopb streams to TCP sockets. 2 */ 3 4 #include <sys/socket.h> 5 #include <sys/types.h> 6 #include <pb_encode.h> 7 #include <pb_decode.h> 8 9 #include "common.h" 10 11 static bool write_callback(pb_ostream_t *stream, const uint8_t *buf, size_t count) 12 { 13 int fd = (intptr_t)stream->state; 14 return send(fd, buf, count, 0) == count; 15 } 16 17 static bool read_callback(pb_istream_t *stream, uint8_t *buf, size_t count) 18 { 19 int fd = (intptr_t)stream->state; 20 int result; 21 22 result = recv(fd, buf, count, MSG_WAITALL); 23 24 if (result == 0) 25 stream->bytes_left = 0; /* EOF */ 26 27 return result == count; 28 } 29 30 pb_ostream_t pb_ostream_from_socket(int fd) 31 { 32 pb_ostream_t stream = {&write_callback, (void*)(intptr_t)fd, SIZE_MAX, 0}; 33 return stream; 34 } 35 36 pb_istream_t pb_istream_from_socket(int fd) 37 { 38 pb_istream_t stream = {&read_callback, (void*)(intptr_t)fd, SIZE_MAX}; 39 return stream; 40 } 41
