common.h

/*
 * Copyright 2016 Google Inc.
 *
 * Licensed under the Apache License, Version 2.0 (the "License");
 * you may not use this file except in compliance with the License.
 * You may obtain a copy of the License at
 *
 *      http://www.apache.org/licenses/LICENSE-2.0
 *
 * Unless required by applicable law or agreed to in writing, software
 * distributed under the License is distributed on an "AS IS" BASIS,
 * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
 * See the License for the specific language governing permissions and
 * limitations under the License.
 */

#ifndef NEPER_COMMON_H
#define NEPER_COMMON_H

#include <errno.h>
#include <netdb.h>
#include <signal.h>
#include <stdio.h>
#include <stdlib.h>
#include <string.h>
#include <sys/epoll.h>
#include <sys/socket.h>
#include <time.h>
#include <unistd.h>
#include "lib.h"
#include "logging.h"

#define PROCFILE_SOMAXCONN "/proc/sys/net/core/somaxconn"

static inline void epoll_ctl_or_die(int epfd, int op, int fd,
                                    struct epoll_event *ev,
                                    struct callbacks *cb)
{
        if (epoll_ctl(epfd, op, fd, ev))
                NP_PLOG_FATAL(cb, "epoll_ctl");
}

static inline void epoll_del_or_err(int epfd, int fd, struct callbacks *cb)
{
        if (epoll_ctl(epfd, EPOLL_CTL_DEL, fd, NULL))
                NP_PLOG_ERROR(cb, "epoll_ctl");
}

static inline double seconds_between(struct timespec *a, struct timespec *b)
{
        return (b->tv_sec - a->tv_sec) + (b->tv_nsec - a->tv_nsec) * 1e-9;
}

static inline int flows_in_thread(int num_flows, int num_threads, int tid)
{
        const int min_flows_per_thread = num_flows / num_threads;
        const int remaining_flows = num_flows % num_threads;
        const int flows_in_this_thread = tid < remaining_flows ?
                                         min_flows_per_thread + 1 :
                                         min_flows_per_thread;
        return flows_in_this_thread;
}

struct addrinfo *do_getaddrinfo(const char *host, const char *port, int flags,
                                const struct options *opts,
                                struct callbacks *cb);
struct addrinfo *do_getaddrinfo_udp(const char *host, const char *port, int flags,
                                    const struct options *opts,
                                    struct callbacks *cb);
long long parse_rate(const char *str, struct callbacks *cb);
void set_reuseport(int fd, struct callbacks *cb);
void set_nonblocking(int fd, struct callbacks *cb);
void set_reuseaddr(int fd, int on, struct callbacks *cb);
void set_debug(int fd, int onoff, struct callbacks *cb);
void set_min_rto(int fd, int min_rto_ms, struct callbacks *cb);
void set_local_host(int fd, struct options *opt, struct callbacks *cb);
int procfile_int(const char *path, struct callbacks *cb);

void fill_random(char *buf, int size);
int do_close(int fd);
int do_connect(int s, const struct sockaddr *addr, socklen_t addr_len);
struct addrinfo *copy_addrinfo(struct addrinfo *in);
void reset_port(struct addrinfo *ai, int port, struct callbacks *cb);
void reset_port_udp(struct addrinfo **ai, struct host *host,
                    struct options *options, int flags, struct callbacks *cb);
int try_connect(const char *host, const char *port, struct addrinfo **ai,
                struct options *opts, struct callbacks *cb);
void parse_all_samples(char *arg, void *out, struct callbacks *cb);
void parse_max_pacing_rate(char *arg, void *out, struct callbacks *cb);

int create_suicide_timeout(int sec_to_suicide);

void parse_hosts(char *str, void *out, struct callbacks *cb);
void print_hosts(const char *name, const void *var, struct callbacks *cb);
int hosts_len(const struct host *array);

void parse_slaves(char *str, void *out, struct callbacks *cb);
void print_slaves(const char *name, const void *var, struct callbacks *cb);

#endif