#include "device.h"
#include "devnode.h"
#include "input.h"
#include "led.h"
#include "notify.h"

Include dependency graph for main.c:

Functions
static void	quitWithLock (char mut)
	quitWithLock More...

int	restart ()

void	timespec_add (struct timespec *timespec, long nanoseconds)

static void	quit ()
	quit Stop working the daemon. function is called if the daemon received a sigterm In this case, locking the device-mutex is ok. More...

void	sighandler2 (int type)

void	sighandler (int type)

void	localecase (char dst, size_t length, const char src)

int	main (int argc, char **argv)

Variables
static int	main_ac

static char **	main_av

volatile int	reset_stop
	brief . More...

int	features_mask
	brief . More...

int	hwload_mode
	hwload_mode = 1 means read hardware once. should be enough More...

Function Documentation

void localecase	(	char *	dst,
		size_t	length,
		const char *	src
	)

Definition at line 71 of file main.c.

                                                           {
     char* ldst = dst + length;
     char s;
     while((s = *src++)){
         if(s == '_')
             s = '-';
         else
             s = tolower(s);
         *dst++ = s;
         if(dst == ldst){
             dst--;
             break;
         }
     }
     *dst = 0;
 }

int main	(	int	argc,
		char **	argv
	)

Definition at line 88 of file main.c.

References ckb_fatal_nofile, ckb_info(), ckb_info_nofile, ckb_warn_nofile, devpath, FEAT_BIND, FEAT_MOUSEACCEL, FEAT_NOTIFY, features_mask, gid, hwload_mode, keyboard, main_ac, main_av, mkdevpath(), quit(), restart(), sighandler(), and usbmain().

Referenced by restart().

                                {
     // Set output pipes to buffer on newlines, if they weren't set that way already
     setlinebuf(stdout);
     setlinebuf(stderr);
     main_ac = argc;
     main_av = argv;
 
     printf("    ckb: Corsair RGB driver %s\n", CKB_VERSION_STR);
     // If --help occurs anywhere in the command-line, don't launch the program but instead print usage
     for(int i = 1; i < argc; i++){
         if(!strcmp(argv[i], "--help")){
             printf(
 #ifdef OS_MAC
                         "Usage: ckb-daemon [--gid=<gid>] [--hwload=<always|try|never>] [--nonotify] [--nobind] [--nomouseaccel] [--nonroot]\n"
 #else
                         "Usage: ckb-daemon [--gid=<gid>] [--hwload=<always|try|never>] [--nonotify] [--nobind] [--nonroot]\n"
 #endif
                         "\n"
                         "See https://github.com/ccMSC/ckb/blob/master/DAEMON.md for full instructions.\n"
                         "\n"
                         "Command-line parameters:\n"
                         "    --gid=<gid>\n"
                         "        Restrict access to %s* nodes to users in group <gid>.\n"
                         "        (Ordinarily they are accessible to anyone)\n"
                         "    --hwload=<always|try|never>\n"
                         "        --hwload=always will force loading of stored hardware profiles on compatible devices. May result in long start up times.\n"
                         "        --hwload=try will try to load the profiles, but give up if not immediately successful (default).\n"
                         "        --hwload=never will ignore hardware profiles completely.\n"
                         "    --nonotify\n"
                         "        Disables key monitoring/notifications.\n"
                         "        Note that this makes reactive lighting impossible.\n"
                         "    --nobind\n"
                         "        Disables all key rebinding, macros, and notifications. Implies --nonotify.\n"
 #ifdef OS_MAC
                         "    --nomouseaccel\n"
                         "        Disables mouse acceleration, even if the system preferences enable it.\n"
 #endif
                         "    --nonroot\n"
                         "        Allows running ckb-daemon as a non root user.\n"
                         "        This will almost certainly not work. Use only if you know what you're doing.\n"
                         "\n", devpath);
             exit(0);
         }
     }
 
     // Check PID, quit if already running
     char pidpath[strlen(devpath) + 6];
     snprintf(pidpath, sizeof(pidpath), "%s0/pid", devpath);
     FILE* pidfile = fopen(pidpath, "r");
     if(pidfile){
         pid_t pid;
         fscanf(pidfile, "%d", &pid);
         fclose(pidfile);
         if(pid > 0){
             // kill -s 0 checks if the PID is active but doesn't send a signal
             if(!kill(pid, 0)){
                 ckb_fatal_nofile("ckb-daemon is already running (PID %d). Try `killall ckb-daemon`.\n", pid);
                 ckb_fatal_nofile("(If you're certain the process is dead, delete %s and try again)\n", pidpath);
                 return 0;
             }
         }
     }
 
     // Read parameters
     int forceroot = 1;
     for(int i = 1; i < argc; i++){
         char* argument = argv[i];
         unsigned newgid;
         char hwload[7];
         if(sscanf(argument, "--gid=%u", &newgid) == 1){
             // Set dev node GID
             gid = newgid;
             ckb_info_nofile("Setting /dev node gid: %u\n", newgid);
         } else if(!strcmp(argument, "--nobind")){
             // Disable key notifications and rebinding
             features_mask &= ~FEAT_BIND & ~FEAT_NOTIFY;
             ckb_info_nofile("Key binding and key notifications are disabled\n");
         } else if(!strcmp(argument, "--nonotify")){
             // Disable key notifications
             features_mask &= ~FEAT_NOTIFY;
             ckb_info_nofile("Key notifications are disabled\n");
         } else if(sscanf(argument, "--hwload=%6s", hwload) == 1){
             if(!strcmp(hwload, "always") || !strcmp(hwload, "yes") || !strcmp(hwload, "y") || !strcmp(hwload, "a")){
                 hwload_mode = 2;
                 ckb_info_nofile("Setting hardware load: always\n");
             } else if(!strcmp(hwload, "tryonce") || !strcmp(hwload, "try") || !strcmp(hwload, "once") || !strcmp(hwload, "t") || !strcmp(hwload, "o")){
                 hwload_mode = 1;
                 ckb_info_nofile("Setting hardware load: tryonce\n");
             } else if(!strcmp(hwload, "never") || !strcmp(hwload, "none") || !strcmp(hwload, "no") || !strcmp(hwload, "n")){
                 hwload_mode = 0;
                 ckb_info_nofile("Setting hardware load: never\n");
             }
         } else if(!strcmp(argument, "--nonroot")){
             // Allow running as a non-root user
             forceroot = 0;
         }
 #ifdef OS_MAC
         else if(!strcmp(argument, "--nomouseaccel")){
             // On OSX, provide an option to disable mouse acceleration
             features_mask &= ~FEAT_MOUSEACCEL;
             ckb_info_nofile("Mouse acceleration disabled\n");
         }
 #endif
     }
 
     // Check UID
     if(getuid() != 0){
         if(forceroot){
             ckb_fatal_nofile("ckb-daemon must be run as root. Try `sudo %s`\n", argv[0]);
             exit(0);
         } else
             ckb_warn_nofile("Warning: not running as root, allowing anyway per command-line parameter...\n");
     }
 
     // Make root keyboard
     umask(0);
     memset(keyboard, 0, sizeof(keyboard));
     if(!mkdevpath(keyboard))
         ckb_info("Root controller ready at %s0\n", devpath);
 
     // Set signals
     sigset_t signals;
     sigfillset(&signals);
     sigdelset(&signals, SIGTERM);
     sigdelset(&signals, SIGINT);
     sigdelset(&signals, SIGQUIT);
     sigdelset(&signals, SIGUSR1);
     // Set up signal handlers for quitting the service.
     sigprocmask(SIG_SETMASK, &signals, 0);
     signal(SIGTERM, sighandler);
     signal(SIGINT, sighandler);
     signal(SIGQUIT, sighandler);
     signal(SIGUSR1, (void (*)())restart);
 
     // Start the USB system
     int result = usbmain();
     quit();
     return result;
 }

Here is the call graph for this function:

Here is the caller graph for this function:

static void quit ( )

static

Definition at line 30 of file main.c.

References quitWithLock().

Referenced by KbFirmware::_fileForBoard(), KbFirmware::_latestForBoard(), FwUpgradeDialog::fwUpdateFinished(), main(), and sighandler().

                   {
     quitWithLock(1);
 }

Here is the call graph for this function:

Here is the caller graph for this function:

void quitWithLock ( char mut )

static

Parameters

mut	try to close files maybe without locking the mutex if mut == true then lock

Definition at line 40 of file main.c.

References ckb_info(), closeusb(), DEV_MAX, devmutex, IS_CONNECTED, keyboard, reset_stop, revertusb(), rmdevpath(), and usbkill().

Referenced by quit(), and restart().

                             {
     // Abort any USB resets in progress
     reset_stop = 1;
     for(int i = 1; i < DEV_MAX; i++){
         // Before closing, set all keyboards back to HID input mode so that the stock driver can still talk to them
         if (mut) pthread_mutex_lock(devmutex + i);
         if(IS_CONNECTED(keyboard + i)){
             revertusb(keyboard + i);
             closeusb(keyboard + i);
         }
         pthread_mutex_unlock(devmutex + i);
     }
     ckb_info("Closing root controller\n");
     rmdevpath(keyboard);
     usbkill();
 }

Here is the call graph for this function:

Here is the caller graph for this function:

int restart ( )

Definition at line 228 of file main.c.

References ckb_err, main(), main_ac, main_av, and quitWithLock().

Referenced by cmd_restart(), and main().

               {
     ckb_err("restart called, running quit without mutex-lock.\n");
     quitWithLock(0);
     return main(main_ac, main_av);
 }

Here is the call graph for this function:

Here is the caller graph for this function:

void sighandler ( int type )

Definition at line 62 of file main.c.

References quit(), and sighandler2().

Referenced by main().

                          {
     signal(SIGTERM, sighandler2);
     signal(SIGINT, sighandler2);
     signal(SIGQUIT, sighandler2);
     printf("\n[I] Caught signal %d\n", type);
     quit();
     exit(0);
 }

Here is the call graph for this function:

Here is the caller graph for this function:

void sighandler2 ( int type )

Definition at line 57 of file main.c.

Referenced by sighandler().

                           {
     // Don't use ckb_warn, we want an extra \n at the beginning
     printf("\n[W] Ignoring signal %d (already shutting down)\n", type);
 }

Here is the caller graph for this function:

void timespec_add	(	struct timespec *	timespec,
		long	nanoseconds
	)

Definition at line 19 of file main.c.

                                                               {
     nanoseconds += timespec->tv_nsec;
     timespec->tv_sec += nanoseconds / 1000000000;
     timespec->tv_nsec = nanoseconds % 1000000000;
 }

Variable Documentation

int features_mask

features_mask Mask of features to exclude from all devices

That bit mask ist set to enable all (-1). When interpreting the input parameters, some of these bits can be cleared.
At the moment binding, notifying and mouse-acceleration can be disabled via command line.
Have a look at main() in main.c for details.

Definition at line 35 of file usb.c.

Referenced by _setupusb(), and main().

int hwload_mode

Definition at line 7 of file device.c.

Referenced by main().

int main_ac

static

Definition at line 7 of file main.c.

Referenced by main(), and restart().

char** main_av

static

Definition at line 8 of file main.c.

Referenced by main(), and restart().

volatile int reset_stop

reset_stop is boolean: Reset stopper for when the program shuts down.

Is set only by quit() to true (1) to inform several usb_* functions to end their loops and tries.

Definition at line 25 of file usb.c.

Referenced by _usbrecv(), _usbsend(), quitWithLock(), and usb_tryreset().

Functions

Variables

Function Documentation

Variable Documentation