tools/gator/daemon/UEvent.cpp

   1 /**
   2  * Copyright (C) ARM Limited 2013-2014. All rights reserved.
   3  *
   4  * This program is free software; you can redistribute it and/or modify
   5  * it under the terms of the GNU General Public License version 2 as
   6  * published by the Free Software Foundation.
   7  */
   8
   9 #include "UEvent.h"
  10
  11 #include <sys/socket.h>
  12 #include <linux/netlink.h>
  13 #include <string.h>
  14
  15 #include <unistd.h>
  16
  17 #include "Logging.h"
  18
  19 static const char EMPTY[] = "";
  20 static const char ACTION[] = "ACTION=";
  21 static const char DEVPATH[] = "DEVPATH=";
  22 static const char SUBSYSTEM[] = "SUBSYSTEM=";
  23
  24 UEvent::UEvent() : mFd(-1) {
  25 }
  26
  27 UEvent::~UEvent() {
  28         if (mFd >= 0) {
  29                 close(mFd);
  30         }
  31 }
  32
  33 bool UEvent::init() {
  34         mFd = socket(PF_NETLINK, SOCK_RAW, NETLINK_KOBJECT_UEVENT);
  35         if (mFd < 0) {
  36                 logg->logMessage("%s(%s:%i): socket failed", __FUNCTION__, __FILE__, __LINE__);
  37                 return false;
  38         }
  39
  40         struct sockaddr_nl sockaddr;
  41         memset(&sockaddr, 0, sizeof(sockaddr));
  42         sockaddr.nl_family = AF_NETLINK;
  43         sockaddr.nl_groups = 1; // bitmask: (1 << 0) == kernel events, (1 << 1) == udev events
  44         sockaddr.nl_pid = 0;
  45         if (bind(mFd, (struct sockaddr *)&sockaddr, sizeof(sockaddr)) != 0) {
  46                 logg->logMessage("%s(%s:%i): bind failed", __FUNCTION__, __FILE__, __LINE__);
  47                 return false;
  48         }
  49
  50         return true;
  51 }
  52
  53 bool UEvent::read(UEventResult *const result) {
  54         ssize_t bytes = recv(mFd, result->mBuf, sizeof(result->mBuf), 0);
  55         if (bytes <= 0) {
  56                 logg->logMessage("%s(%s:%i): recv failed", __FUNCTION__, __FILE__, __LINE__);
  57                 return false;
  58         }
  59
  60         result->mAction = EMPTY;
  61         result->mDevPath = EMPTY;
  62         result->mSubsystem = EMPTY;
  63
  64         for (int pos = 0; pos < bytes; pos += strlen(result->mBuf + pos) + 1) {
  65                 char *const str = result->mBuf + pos;
  66                 if (strncmp(str, ACTION, sizeof(ACTION) - 1) == 0) {
  67                         result->mAction = str + sizeof(ACTION) - 1;
  68                 } else if (strncmp(str, DEVPATH, sizeof(DEVPATH) - 1) == 0) {
  69                         result->mDevPath = str + sizeof(DEVPATH) - 1;
  70                 } else if (strncmp(str, SUBSYSTEM, sizeof(SUBSYSTEM) - 1) == 0) {
  71                         result->mSubsystem = str + sizeof(SUBSYSTEM) - 1;
  72                 }
  73         }
  74
  75         return true;
  76 }