error handling
This commit is contained in:
parent
a0b2392ac6
commit
12f3f59620
1 changed files with 9 additions and 2 deletions
|
@ -114,7 +114,11 @@ uint8_t comm_tcp_socket::get_byte()
|
||||||
}
|
}
|
||||||
|
|
||||||
uint8_t c = 0;
|
uint8_t c = 0;
|
||||||
read(use_fd, &c, 1); // TODO error checking
|
if (read(use_fd, &c, 1) <= 0) {
|
||||||
|
std::unique_lock<std::mutex> lck(cfd_lock);
|
||||||
|
close(cfd);
|
||||||
|
cfd = INVALID_SOCKET;
|
||||||
|
}
|
||||||
|
|
||||||
return c;
|
return c;
|
||||||
}
|
}
|
||||||
|
@ -127,8 +131,11 @@ void comm_tcp_socket::send_data(const uint8_t *const in, const size_t n)
|
||||||
while(len > 0) {
|
while(len > 0) {
|
||||||
std::unique_lock<std::mutex> lck(cfd_lock);
|
std::unique_lock<std::mutex> lck(cfd_lock);
|
||||||
int rc = write(cfd, p, len);
|
int rc = write(cfd, p, len);
|
||||||
if (rc <= 0) // TODO error checking
|
if (rc <= 0) { // TODO error checking
|
||||||
|
close(cfd);
|
||||||
|
cfd = INVALID_SOCKET;
|
||||||
break;
|
break;
|
||||||
|
}
|
||||||
|
|
||||||
p += rc;
|
p += rc;
|
||||||
len -= rc;
|
len -= rc;
|
||||||
|
|
Loading…
Add table
Reference in a new issue