diff --git a/gw_utils.c b/gw_utils.c new file mode 100644 index 000000000..e34c238ea --- /dev/null +++ b/gw_utils.c @@ -0,0 +1,207 @@ +/* +This file is distributed as part of the SkySQL Gateway. It is free +software: you can redistribute it and/or modify it under the terms of the +GNU General Public License as published by the Free Software Foundation, +version 2. + +This program is distributed in the hope that it will be useful, but WITHOUT +ANY WARRANTY; without even the implied warranty of MERCHANTABILITY or FITNESS +FOR A PARTICULAR PURPOSE. See the GNU General Public License for more +details. + +You should have received a copy of the GNU General Public License along with +this program; if not, write to the Free Software Foundation, Inc., 51 +Franklin Street, Fifth Floor, Boston, MA 02110-1301 USA. + +Copyright SkySQL Ab + +*/ + +/* +03-06-2013 +gateway utils +Massimiliano Pinto +*/ + +#include "gw.h" +#include "dcb.h" + +/// +// set ip address in sockect struct +/// +void setipaddress(struct in_addr *a, char *p) { + struct hostent *h = gethostbyname(p); + if (h == NULL) { + if ((a->s_addr = inet_addr(p)) == -1) { + error("unknown or invalid address [%s]\n", p); + } + } else { + memcpy(a, h->h_addr, h->h_length); + } +} + +int gw_daemonize(void) { + pid_t pid; + + pid = fork(); + + if (pid < 0) { + fprintf(stderr, "fork() error %s\n", strerror(errno)); + exit(1); + } + + if (pid != 0) { + // exit from main + exit(0); + } + + if (setsid() < 0) { + fprintf(stderr, "setsid() error %s\n", strerror(errno)); + exit(1); + } +} + +////////////////////////////////////// +// Generic read from filedescriptor +////////////////////////////////////// +int do_read_fd(int fd) { + char buf[MAX_BUFFER_SIZE]=""; + int n; + + n = read(fd, buf, MAX_BUFFER_SIZE); + + if (n == 0) { + fprintf(stderr, "Generic Socket EOF\n"); + return -1; + } + + if (n < 0) { + fprintf(stderr, "Read error: %i [%s]\n", errno, strerror(errno)); + if ((errno != EAGAIN) || (errno != EWOULDBLOCK)) + return -1; + else + return 0; + } + +#ifdef DEBUG_GATEWAY_READ + fprintf(stderr, "socket %i, do_read %i bytes\n", fd, n); +#endif + + return n; +} + +int do_read_dcb10(DCB *dcb) { + char buf[MAX_BUFFER_SIZE]=""; + int n; + n = read(dcb->fd, buf, 10); + + if (n == 0) { + fprintf(stderr, "Read DCB Socket EOF\n"); + return -1; + } + + if (n < 0) { + fprintf(stderr, "ReadDcb error: %i [%s]\n", errno, strerror(errno)); + if ((errno != EAGAIN) || (errno != EWOULDBLOCK)) + return -1; + else + return 0; + } + +#ifdef GW_READ_DEBUG + fprintf(stderr, "socket %i, do_read % i bytes, [%s]\n", dcb->fd, n, buf+5); +#endif + + return n; +} + +int do_read_dcb(DCB *dcb) { + char buf[MAX_BUFFER_SIZE]=""; + int n; + n = read(dcb->fd, buf, MAX_BUFFER_SIZE); + + if (n == 0) { + fprintf(stderr, "Read DCB Socket EOF\n"); + return -1; + } + + if (n < 0) { + fprintf(stderr, "ReadDcb error: %i [%s]\n", errno, strerror(errno)); + if ((errno != EAGAIN) || (errno != EWOULDBLOCK)) + return -1; + else + return 0; + } + +#ifdef GW_READ_DEBUG + fprintf(stderr, "socket %i, do_read % i bytes, [%s]\n", dcb->fd, n, buf+5); +#endif + + return n; +} + + +int do_read_buffer10(DCB *dcb, uint8_t *buffer) { + int n; + n = read(dcb->fd, buffer, 10); + + if (n == 0 & (errno==EOF)) { + fprintf(stderr, "ReadBuffer Socket EOF\n"); + return -1; + } + + if (n < 0) { + fprintf(stderr, "ReadBuffer error: %i [%s]\n", errno, strerror(errno)); + if ((errno != EAGAIN) || (errno != EWOULDBLOCK)) + return -1; + else + return 0; + } + +#ifdef GW_READ_DEBUG + fprintf(stderr, "socket %i, do_read % i bytes\n", dcb->fd, n); +#endif + + return n; +} + +int do_read_buffer(DCB *dcb, uint8_t *buffer) { + int n; + n = read(dcb->fd, buffer, MAX_BUFFER_SIZE); + + if (n == 0 & (errno==EOF)) { + fprintf(stderr, "ReadBuffer Socket EOF\n"); + return -1; + } + + if (n < 0) { + fprintf(stderr, "ReadBuffer error: %i [%s]\n", errno, strerror(errno)); + if ((errno != EAGAIN) || (errno != EWOULDBLOCK)) + return -1; + else + return 0; + } + +#ifdef GW_READ_DEBUG + fprintf(stderr, "socket %i, do_read % i bytes\n", dcb->fd, n); +#endif + + return n; +} + +const char *gw_dcb_state2string (int state) { + switch(state) { + case DCB_STATE_ALLOC: + return "DCB Allocated"; + case DCB_STATE_IDLE: + return "DCB not yet in polling"; + case DCB_STATE_POLLING: + return "DCB in the EPOLL"; + case DCB_STATE_PROCESSING: + return "DCB processing event"; + case DCB_STATE_LISTENING: + return "DCB for listening socket"; + default: + return "DCB (unknown)"; + } +}