From 834078a53da95d2261d5f8d042b773d3ed2f04b3 Mon Sep 17 00:00:00 2001 From: Jason Fesler Date: Tue, 17 Feb 2015 11:52:32 -0800 Subject: Add daemon flag --- init.d/mtu1280 | 158 ++++++++++++++++++++++++++++++++++++++++++++++++++++++++ mtu1280.c | 20 +++++-- upstart/mtu1280 | 18 +++++++ 3 files changed, 193 insertions(+), 3 deletions(-) create mode 100644 init.d/mtu1280 create mode 100755 upstart/mtu1280 diff --git a/init.d/mtu1280 b/init.d/mtu1280 new file mode 100644 index 0000000..8d4d3e6 --- /dev/null +++ b/init.d/mtu1280 @@ -0,0 +1,158 @@ +#! /bin/sh +### BEGIN INIT INFO +# Provides: mtu1280 +# Required-Start: $network $syslog +# Required-Stop: $network $syslog +# Default-Start: 2 3 4 5 +# Default-Stop: 0 1 6 +# Short-Description: mtu1280 Synthetic ICMPv6 Packet Too Big response +# Description: This file should be used to construct scripts to be +# placed in /etc/init.d. +### END INIT INFO + +# Author: Jason Fesler +# +# Do NOT "set -e" + +# PATH should only include /usr/* if it runs after the mountnfs.sh script +PATH=/sbin:/usr/sbin:/bin:/usr/bin +DESC="Synthetic ICMPv6 Packet Too Big response" +NAME=mtu1280 +DAEMON=/usr/sbin/$NAME +DAEMON_ARGS="-d -q 1280" +PIDFILE=/var/run/$NAME.pid +SCRIPTNAME=/etc/init.d/$NAME + +# Exit if the package is not installed +[ -x "$DAEMON" ] || exit 0 + +# Read configuration variable file if it is present +[ -r /etc/default/$NAME ] && . /etc/default/$NAME + +# Load the VERBOSE setting and other rcS variables +. /lib/init/vars.sh + +# Define LSB log_* functions. +# Depend on lsb-base (>= 3.2-14) to ensure that this file is present +# and status_of_proc is working. +. /lib/lsb/init-functions + +# +# Function that starts the daemon/service +# +do_start() +{ + # Return + # 0 if daemon has been started + # 1 if daemon was already running + # 2 if daemon could not be started + start-stop-daemon --start --quiet --pidfile $PIDFILE --exec $DAEMON --test > /dev/null \ + || return 1 + start-stop-daemon --start --quiet --pidfile $PIDFILE --exec $DAEMON -- \ + $DAEMON_ARGS \ + || return 2 + # Add code here, if necessary, that waits for the process to be ready + # to handle requests from services started subsequently which depend + # on this one. As a last resort, sleep for some time. +} + +# +# Function that stops the daemon/service +# +do_stop() +{ + # Return + # 0 if daemon has been stopped + # 1 if daemon was already stopped + # 2 if daemon could not be stopped + # other if a failure occurred + start-stop-daemon --stop --quiet --retry=TERM/30/KILL/5 --pidfile $PIDFILE --name $NAME + RETVAL="$?" + [ "$RETVAL" = 2 ] && return 2 + # Wait for children to finish too if this is a daemon that forks + # and if the daemon is only ever run from this initscript. + # If the above conditions are not satisfied then add some other code + # that waits for the process to drop all resources that could be + # needed by services started subsequently. A last resort is to + # sleep for some time. + start-stop-daemon --stop --quiet --oknodo --retry=0/30/KILL/5 --exec $DAEMON + [ "$?" = 2 ] && return 2 + # Many daemons don't delete their pidfiles when they exit. + rm -f $PIDFILE + return "$RETVAL" +} + +# +# Function that sends a SIGHUP to the daemon/service +# +do_reload() { + # + # If the daemon can reload its configuration without + # restarting (for example, when it is sent a SIGHUP), + # then implement that here. + # + echo reload not implemented + return 1 + start-stop-daemon --stop --signal 1 --quiet --pidfile $PIDFILE --name $NAME + return 0 +} + +case "$1" in + start) + [ "$VERBOSE" != no ] && log_daemon_msg "Starting $DESC" "$NAME" + do_start + case "$?" in + 0|1) [ "$VERBOSE" != no ] && log_end_msg 0 ;; + 2) [ "$VERBOSE" != no ] && log_end_msg 1 ;; + esac + ;; + stop) + [ "$VERBOSE" != no ] && log_daemon_msg "Stopping $DESC" "$NAME" + do_stop + case "$?" in + 0|1) [ "$VERBOSE" != no ] && log_end_msg 0 ;; + 2) [ "$VERBOSE" != no ] && log_end_msg 1 ;; + esac + ;; + status) + status_of_proc "$DAEMON" "$NAME" && exit 0 || exit $? + ;; + #reload|force-reload) + # + # If do_reload() is not implemented then leave this commented out + # and leave 'force-reload' as an alias for 'restart'. + # + #log_daemon_msg "Reloading $DESC" "$NAME" + #do_reload + #log_end_msg $? + #;; + restart|force-reload) + # + # If the "reload" option is implemented then remove the + # 'force-reload' alias + # + log_daemon_msg "Restarting $DESC" "$NAME" + do_stop + case "$?" in + 0|1) + do_start + case "$?" in + 0) log_end_msg 0 ;; + 1) log_end_msg 1 ;; # Old process is still running + *) log_end_msg 1 ;; # Failed to start + esac + ;; + *) + # Failed to stop + log_end_msg 1 + ;; + esac + ;; + *) + #echo "Usage: $SCRIPTNAME {start|stop|restart|reload|force-reload}" >&2 + echo "Usage: $SCRIPTNAME {start|stop|status|restart|force-reload}" >&2 + exit 3 + ;; +esac + +: diff --git a/mtu1280.c b/mtu1280.c index 60a9248..7948af9 100644 --- a/mtu1280.c +++ b/mtu1280.c @@ -408,7 +408,6 @@ cb (struct nfq_q_handle *qh, struct nfgenmsg *nfmsg, } - int main (int argc, char **argv) { @@ -417,16 +416,22 @@ main (int argc, char **argv) struct nfnl_handle *nh; int fd; int rv; - unsigned int queue = 1280; // default value + unsigned int queue = 1280; // -q + unsigned int do_fork = 0; // -d char *interface; char buf[4096] __attribute__ ((aligned)); // Getopt int c; int opterr = 0; - while ((c = getopt (argc, argv, "q:")) != -1) + while ((c = getopt (argc, argv, "dq:")) != -1) switch (c) { + case 'd': + fprintf (stderr, "setting\n"); + do_fork = 1; + fprintf (stderr, "set!\n"); + break; case 'q': queue = strtol (optarg, NULL, 10); break; @@ -479,6 +484,7 @@ main (int argc, char **argv) exit (1); } + printf ("setting copy_packet mode\n"); if (nfq_set_mode (qh, NFQNL_COPY_PACKET, 0xffff) < 0) { @@ -486,6 +492,14 @@ main (int argc, char **argv) exit (1); } + + if (do_fork) + { + fprintf (stdout, "forking to background\n"); + daemon (0, 0); + } + + fd = nfq_fd (h); while ((rv = recv (fd, buf, sizeof (buf), 0)) && rv >= 0) diff --git a/upstart/mtu1280 b/upstart/mtu1280 new file mode 100755 index 0000000..a96cdf9 --- /dev/null +++ b/upstart/mtu1280 @@ -0,0 +1,18 @@ +# ssh - OpenBSD Secure Shell server +# +# The OpenSSH server provides secure shell access to the system. + +description "Synthetic ICMPv6 Packet Too Big response" + +start on runlevel [2345] +stop on runlevel [!2345] + +respawn +respawn limit 10 5 +umask 022 + +#expect stop + +console none + +exec /usr/sbin/mtu1280 -q 1280 -- cgit v1.2.3