| /* |
| * Copyright (C) 2007 - 2018 Vladislav Bolkhovitin |
| * Copyright (C) 2007 - 2018 Western Digital Corporation |
| * |
| * This program 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 |
| * of the License. |
| * |
| * 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. |
| */ |
| |
| #include <fcntl.h> |
| #include <sys/socket.h> |
| #include <netinet/tcp.h> |
| #include <string.h> |
| #include <errno.h> |
| #include <stdio.h> |
| #include <sys/types.h> |
| #include <sys/stat.h> |
| #include <unistd.h> |
| |
| #include "iscsid.h" |
| |
| static int driver_major(const char *dev) |
| { |
| FILE *f; |
| char devname[256]; |
| char buf[256]; |
| int devn; |
| |
| f = fopen("/proc/devices", "r"); |
| if (!f) { |
| devn = -errno; |
| perror("Cannot open control path to the driver"); |
| goto out; |
| } |
| |
| devn = -ENOENT; |
| while (fgets(buf, sizeof(buf), f)) { |
| if (sscanf(buf, "%d %s", &devn, devname) == 2 && |
| devn > 0 && strcmp(devname, dev) == 0) |
| break; |
| devn = -ENOENT; |
| } |
| fclose(f); |
| |
| if (devn < 0) |
| printf("cannot find %s in /proc/devices - " |
| "make sure the module is loaded\n", dev); |
| |
| out: |
| return devn; |
| } |
| |
| int create_and_open_dev(const char *dev, int readonly) |
| { |
| char devname[256]; |
| int devn; |
| int ctlfd = -1; |
| int err; |
| int flags; |
| |
| devn = driver_major(dev); |
| if (devn < 0) { |
| err = devn; |
| goto out; |
| } |
| |
| sprintf(devname, "/dev/%s", dev); |
| |
| unlink(devname); |
| if (mknod(devname, (S_IFCHR | 0600), (devn << 8))) { |
| err = -errno; |
| printf("cannot create %s %s\n", devname, strerror(errno)); |
| goto out; |
| } |
| |
| if (readonly) |
| flags = O_RDONLY; |
| else |
| flags = O_RDWR; |
| |
| err = ctlfd = open(devname, flags); |
| if (ctlfd < 0) { |
| err = -errno; |
| printf("cannot open %s %s\n", devname, strerror(errno)); |
| goto out; |
| } |
| |
| out: |
| return err; |
| } |
| |
| void set_non_blocking(int fd) |
| { |
| int res = fcntl(fd, F_GETFL); |
| |
| if (res != -1) { |
| res = fcntl(fd, F_SETFL, res | O_NONBLOCK); |
| if (res) |
| log_warning("unable to set fd flags (%s)!", strerror(errno)); |
| } else |
| log_warning("unable to get fd flags (%s)!", strerror(errno)); |
| } |
| |
| void sock_set_keepalive(int sock, int timeout) |
| { |
| if (timeout) { /* timeout [s] */ |
| int opt = 2; |
| |
| if (setsockopt(sock, SOL_TCP, TCP_KEEPCNT, &opt, sizeof(opt))) |
| log_warning("unable to set TCP_KEEPCNT on server socket (%s)!", strerror(errno)); |
| |
| if (setsockopt(sock, SOL_TCP, TCP_KEEPIDLE, &timeout, sizeof(timeout))) |
| log_warning("unable to set TCP_KEEPIDLE on server socket (%s)!", strerror(errno)); |
| |
| opt = 3; |
| if (setsockopt(sock, SOL_TCP, TCP_KEEPINTVL, &opt, sizeof(opt))) |
| log_warning("unable to set KEEPINTVL on server socket (%s)!", strerror(errno)); |
| |
| opt = 1; |
| if (setsockopt(sock, SOL_SOCKET, SO_KEEPALIVE, &opt, sizeof(opt))) |
| log_warning("unable to set SO_KEEPALIVE on server socket (%s)!", strerror(errno)); |
| } |
| } |
| |