tools/gator/daemon/UEvent.cpp

   1 /**
   2  * Copyright (C) ARM Limited 2013-2015. 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 <string.h>
  12 #include <sys/socket.h>
  13 #include <unistd.h>
  14
  15 #include <linux/netlink.h>
  16
  17 #include "Logging.h"
  18 #include "OlySocket.h"
  19
  20 static const char EMPTY[] = "";
  21 static const char ACTION[] = "ACTION=";
  22 static const char DEVPATH[] = "DEVPATH=";
  23 static const char SUBSYSTEM[] = "SUBSYSTEM=";
  24
  25 UEvent::UEvent() : mFd(-1) {
  26 }
  27
  28 UEvent::~UEvent() {
  29         if (mFd >= 0) {
  30                 close(mFd);
  31         }
  32 }
  33
  34 bool UEvent::init() {
  35         mFd = socket_cloexec(PF_NETLINK, SOCK_RAW, NETLINK_KOBJECT_UEVENT);
  36         if (mFd < 0) {
  37                 logg->logMessage("socket failed");
  38                 return false;
  39         }
  40
  41         struct sockaddr_nl sockaddr;
  42         memset(&sockaddr, 0, sizeof(sockaddr));
  43         sockaddr.nl_family = AF_NETLINK;
  44         sockaddr.nl_groups = 1; // bitmask: (1 << 0) == kernel events, (1 << 1) == udev events
  45         sockaddr.nl_pid = 0;
  46         if (bind(mFd, (struct sockaddr *)&sockaddr, sizeof(sockaddr)) != 0) {
  47                 logg->logMessage("bind failed");
  48                 return false;
  49         }
  50
  51         return true;
  52 }
  53
  54 bool UEvent::read(UEventResult *const result) {
  55         ssize_t bytes = recv(mFd, result->mBuf, sizeof(result->mBuf), 0);
  56         if (bytes <= 0) {
  57                 logg->logMessage("recv failed");
  58                 return false;
  59         }
  60
  61         result->mAction = EMPTY;
  62         result->mDevPath = EMPTY;
  63         result->mSubsystem = EMPTY;
  64
  65         for (int pos = 0; pos < bytes; pos += strlen(result->mBuf + pos) + 1) {
  66                 char *const str = result->mBuf + pos;
  67                 logg->logMessage("uevent + %i: %s", pos, str);
  68                 if (strncmp(str, ACTION, sizeof(ACTION) - 1) == 0) {
  69                         result->mAction = str + sizeof(ACTION) - 1;
  70                 } else if (strncmp(str, DEVPATH, sizeof(DEVPATH) - 1) == 0) {
  71                         result->mDevPath = str + sizeof(DEVPATH) - 1;
  72                 } else if (strncmp(str, SUBSYSTEM, sizeof(SUBSYSTEM) - 1) == 0) {
  73                         result->mSubsystem = str + sizeof(SUBSYSTEM) - 1;
  74                 }
  75         }
  76
  77         return true;
  78 }