mirror of https://github.com/esphome/esphome.git
106 lines
3.3 KiB
C++
106 lines
3.3 KiB
C++
#include "socket.h"
|
|
#include "esphome/core/defines.h"
|
|
|
|
#ifdef USE_SOCKET_IMPL_BSD_SOCKETS
|
|
|
|
#include <string.h>
|
|
|
|
namespace esphome {
|
|
namespace socket {
|
|
|
|
std::string format_sockaddr(const struct sockaddr_storage &storage) {
|
|
if (storage.ss_family == AF_INET) {
|
|
const struct sockaddr_in *addr = reinterpret_cast<const struct sockaddr_in *>(&storage);
|
|
char buf[INET_ADDRSTRLEN];
|
|
const char *ret = inet_ntop(AF_INET, &addr->sin_addr, buf, sizeof(buf));
|
|
if (ret == NULL)
|
|
return {};
|
|
return std::string{buf};
|
|
} else if (storage.ss_family == AF_INET6) {
|
|
const struct sockaddr_in6 *addr = reinterpret_cast<const struct sockaddr_in6 *>(&storage);
|
|
char buf[INET6_ADDRSTRLEN];
|
|
const char *ret = inet_ntop(AF_INET6, &addr->sin6_addr, buf, sizeof(buf));
|
|
if (ret == NULL)
|
|
return {};
|
|
return std::string{buf};
|
|
}
|
|
return {};
|
|
}
|
|
|
|
class BSDSocketImpl : public Socket {
|
|
public:
|
|
BSDSocketImpl(int fd) : Socket(), fd_(fd) {}
|
|
~BSDSocketImpl() override {
|
|
if (!closed_) {
|
|
close();
|
|
}
|
|
}
|
|
std::unique_ptr<Socket> accept(struct sockaddr *addr, socklen_t *addrlen) override {
|
|
int fd = ::accept(fd_, addr, addrlen);
|
|
if (fd == -1)
|
|
return {};
|
|
return std::unique_ptr<BSDSocketImpl>{new BSDSocketImpl(fd)};
|
|
}
|
|
int bind(const struct sockaddr *addr, socklen_t addrlen) override { return ::bind(fd_, addr, addrlen); }
|
|
int close() override {
|
|
int ret = ::close(fd_);
|
|
closed_ = true;
|
|
return ret;
|
|
}
|
|
int shutdown(int how) override { return ::shutdown(fd_, how); }
|
|
|
|
int getpeername(struct sockaddr *addr, socklen_t *addrlen) override { return ::getpeername(fd_, addr, addrlen); }
|
|
std::string getpeername() override {
|
|
struct sockaddr_storage storage;
|
|
socklen_t len = sizeof(storage);
|
|
int err = this->getpeername((struct sockaddr *) &storage, &len);
|
|
if (err != 0)
|
|
return {};
|
|
return format_sockaddr(storage);
|
|
}
|
|
int getsockname(struct sockaddr *addr, socklen_t *addrlen) override { return ::getsockname(fd_, addr, addrlen); }
|
|
std::string getsockname() override {
|
|
struct sockaddr_storage storage;
|
|
socklen_t len = sizeof(storage);
|
|
int err = this->getsockname((struct sockaddr *) &storage, &len);
|
|
if (err != 0)
|
|
return {};
|
|
return format_sockaddr(storage);
|
|
}
|
|
int getsockopt(int level, int optname, void *optval, socklen_t *optlen) override {
|
|
return ::getsockopt(fd_, level, optname, optval, optlen);
|
|
}
|
|
int setsockopt(int level, int optname, const void *optval, socklen_t optlen) override {
|
|
return ::setsockopt(fd_, level, optname, optval, optlen);
|
|
}
|
|
int listen(int backlog) override { return ::listen(fd_, backlog); }
|
|
ssize_t read(void *buf, size_t len) override { return ::read(fd_, buf, len); }
|
|
ssize_t write(const void *buf, size_t len) override { return ::write(fd_, buf, len); }
|
|
int setblocking(bool blocking) override {
|
|
int fl = ::fcntl(fd_, F_GETFL, 0);
|
|
if (blocking) {
|
|
fl &= ~O_NONBLOCK;
|
|
} else {
|
|
fl |= O_NONBLOCK;
|
|
}
|
|
::fcntl(fd_, F_SETFL, fl);
|
|
return 0;
|
|
}
|
|
|
|
protected:
|
|
int fd_;
|
|
bool closed_ = false;
|
|
};
|
|
|
|
std::unique_ptr<Socket> socket(int domain, int type, int protocol) {
|
|
int ret = ::socket(domain, type, protocol);
|
|
if (ret == -1)
|
|
return nullptr;
|
|
return std::unique_ptr<Socket>{new BSDSocketImpl(ret)};
|
|
}
|
|
|
|
} // namespace socket
|
|
} // namespace esphome
|
|
|
|
#endif // USE_SOCKET_IMPL_BSD_SOCKETS
|