consumer.cc revision 301087
1300906Sasomers/*- 2300906Sasomers * Copyright (c) 2011, 2012, 2013, 2014 Spectra Logic Corporation 3300906Sasomers * All rights reserved. 4300906Sasomers * 5300906Sasomers * Redistribution and use in source and binary forms, with or without 6300906Sasomers * modification, are permitted provided that the following conditions 7300906Sasomers * are met: 8300906Sasomers * 1. Redistributions of source code must retain the above copyright 9300906Sasomers * notice, this list of conditions, and the following disclaimer, 10300906Sasomers * without modification. 11300906Sasomers * 2. Redistributions in binary form must reproduce at minimum a disclaimer 12300906Sasomers * substantially similar to the "NO WARRANTY" disclaimer below 13300906Sasomers * ("Disclaimer") and any redistribution must be conditioned upon 14300906Sasomers * including a substantially similar Disclaimer requirement for further 15300906Sasomers * binary redistribution. 16300906Sasomers * 17300906Sasomers * NO WARRANTY 18300906Sasomers * THIS SOFTWARE IS PROVIDED BY THE COPYRIGHT HOLDERS AND CONTRIBUTORS 19300906Sasomers * "AS IS" AND ANY EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT 20300906Sasomers * LIMITED TO, THE IMPLIED WARRANTIES OF MERCHANTIBILITY AND FITNESS FOR 21300906Sasomers * A PARTICULAR PURPOSE ARE DISCLAIMED. IN NO EVENT SHALL THE COPYRIGHT 22300906Sasomers * HOLDERS OR CONTRIBUTORS BE LIABLE FOR SPECIAL, EXEMPLARY, OR CONSEQUENTIAL 23300906Sasomers * DAMAGES (INCLUDING, BUT NOT LIMITED TO, PROCUREMENT OF SUBSTITUTE GOODS 24300906Sasomers * OR SERVICES; LOSS OF USE, DATA, OR PROFITS; OR BUSINESS INTERRUPTION) 25300906Sasomers * HOWEVER CAUSED AND ON ANY THEORY OF LIABILITY, WHETHER IN CONTRACT, 26300906Sasomers * STRICT LIABILITY, OR TORT (INCLUDING NEGLIGENCE OR OTHERWISE) ARISING 27300906Sasomers * IN ANY WAY OUT OF THE USE OF THIS SOFTWARE, EVEN IF ADVISED OF THE 28300906Sasomers * POSSIBILITY OF SUCH DAMAGES. 29300906Sasomers * 30300906Sasomers * Authors: Justin T. Gibbs (Spectra Logic Corporation) 31300906Sasomers */ 32300906Sasomers 33300906Sasomers/** 34300906Sasomers * \file consumer.cc 35300906Sasomers */ 36300906Sasomers 37300906Sasomers#include <sys/cdefs.h> 38300906Sasomers#include <sys/poll.h> 39300906Sasomers#include <sys/socket.h> 40300906Sasomers#include <sys/un.h> 41300906Sasomers 42300906Sasomers#include <err.h> 43300906Sasomers#include <errno.h> 44300906Sasomers#include <fcntl.h> 45300906Sasomers#include <syslog.h> 46300906Sasomers#include <unistd.h> 47300906Sasomers 48300906Sasomers#include <cstdarg> 49300906Sasomers#include <cstring> 50300906Sasomers#include <list> 51300906Sasomers#include <map> 52300906Sasomers#include <string> 53300906Sasomers 54300906Sasomers#include "guid.h" 55300906Sasomers#include "event.h" 56300906Sasomers#include "event_factory.h" 57300906Sasomers#include "exception.h" 58300906Sasomers 59300906Sasomers#include "consumer.h" 60300906Sasomers 61300906Sasomers__FBSDID("$FreeBSD: head/lib/libdevdctl/consumer.cc 301087 2016-05-31 23:26:45Z asomers $"); 62300906Sasomers 63300906Sasomers/*================================== Macros ==================================*/ 64300906Sasomers#define NUM_ELEMENTS(x) (sizeof(x) / sizeof(*x)) 65300906Sasomers 66300906Sasomers/*============================ Namespace Control =============================*/ 67300906Sasomersusing std::string; 68300906Sasomersnamespace DevdCtl 69300906Sasomers{ 70300906Sasomers 71300906Sasomers/*============================= Class Definitions ============================*/ 72300906Sasomers/*----------------------------- DevdCtl::Consumer ----------------------------*/ 73300906Sasomers//- Consumer Static Private Data ----------------------------------------------- 74300906Sasomersconst char Consumer::s_devdSockPath[] = "/var/run/devd.seqpacket.pipe"; 75300906Sasomers 76300906Sasomers//- Consumer Public Methods ---------------------------------------------------- 77300906SasomersConsumer::Consumer(Event::BuildMethod *defBuilder, 78300906Sasomers EventFactory::Record *regEntries, 79300906Sasomers size_t numEntries) 80300906Sasomers : m_devdSockFD(-1), 81300906Sasomers m_eventFactory(defBuilder), 82300906Sasomers m_replayingEvents(false) 83300906Sasomers{ 84300906Sasomers m_eventFactory.UpdateRegistry(regEntries, numEntries); 85300906Sasomers} 86300906Sasomers 87300906SasomersConsumer::~Consumer() 88300906Sasomers{ 89300906Sasomers DisconnectFromDevd(); 90300906Sasomers} 91300906Sasomers 92300906Sasomersbool 93300906SasomersConsumer::ConnectToDevd() 94300906Sasomers{ 95300906Sasomers struct sockaddr_un devdAddr; 96300906Sasomers int sLen; 97300906Sasomers int result; 98300906Sasomers 99300906Sasomers if (m_devdSockFD != -1) { 100300906Sasomers /* Already connected. */ 101300906Sasomers syslog(LOG_DEBUG, "%s: Already connected.", __func__); 102300906Sasomers return (true); 103300906Sasomers } 104300906Sasomers syslog(LOG_INFO, "%s: Connecting to devd.", __func__); 105300906Sasomers 106300906Sasomers memset(&devdAddr, 0, sizeof(devdAddr)); 107300906Sasomers devdAddr.sun_family= AF_UNIX; 108300906Sasomers strlcpy(devdAddr.sun_path, s_devdSockPath, sizeof(devdAddr.sun_path)); 109300906Sasomers sLen = SUN_LEN(&devdAddr); 110300906Sasomers 111300906Sasomers m_devdSockFD = socket(AF_UNIX, SOCK_SEQPACKET, 0); 112300906Sasomers if (m_devdSockFD == -1) 113300906Sasomers err(1, "Unable to create socket"); 114300906Sasomers if (fcntl(m_devdSockFD, F_SETFL, O_NONBLOCK) < 0) 115300906Sasomers err(1, "fcntl"); 116300906Sasomers result = connect(m_devdSockFD, 117300906Sasomers reinterpret_cast<sockaddr *>(&devdAddr), 118300906Sasomers sLen); 119300906Sasomers if (result == -1) { 120300906Sasomers syslog(LOG_INFO, "Unable to connect to devd"); 121300906Sasomers DisconnectFromDevd(); 122300906Sasomers return (false); 123300906Sasomers } 124300906Sasomers 125300906Sasomers syslog(LOG_INFO, "Connection to devd successful"); 126300906Sasomers return (true); 127300906Sasomers} 128300906Sasomers 129300906Sasomersvoid 130300906SasomersConsumer::DisconnectFromDevd() 131300906Sasomers{ 132301087Sasomers if (m_devdSockFD != -1) { 133300906Sasomers syslog(LOG_INFO, "Disconnecting from devd."); 134301087Sasomers close(m_devdSockFD); 135301087Sasomers } 136300906Sasomers m_devdSockFD = -1; 137300906Sasomers} 138300906Sasomers 139300906Sasomersstd::string 140300906SasomersConsumer::ReadEvent() 141300906Sasomers{ 142300906Sasomers char buf[MAX_EVENT_SIZE + 1]; 143300906Sasomers ssize_t len; 144300906Sasomers 145300906Sasomers len = ::recv(m_devdSockFD, buf, MAX_EVENT_SIZE, MSG_WAITALL); 146300906Sasomers if (len == -1) 147300906Sasomers return (std::string("")); 148300906Sasomers else { 149300906Sasomers /* NULL-terminate the result */ 150300906Sasomers buf[len] = '\0'; 151300906Sasomers return (std::string(buf)); 152300906Sasomers } 153300906Sasomers} 154300906Sasomers 155300906Sasomersvoid 156300906SasomersConsumer::ReplayUnconsumedEvents(bool discardUnconsumed) 157300906Sasomers{ 158300906Sasomers EventList::iterator event(m_unconsumedEvents.begin()); 159300906Sasomers bool replayed_any = (event != m_unconsumedEvents.end()); 160300906Sasomers 161300906Sasomers m_replayingEvents = true; 162300906Sasomers if (replayed_any) 163300906Sasomers syslog(LOG_INFO, "Started replaying unconsumed events"); 164300906Sasomers while (event != m_unconsumedEvents.end()) { 165300906Sasomers bool consumed((*event)->Process()); 166300906Sasomers if (consumed || discardUnconsumed) { 167300906Sasomers delete *event; 168300906Sasomers event = m_unconsumedEvents.erase(event); 169300906Sasomers } else { 170300906Sasomers event++; 171300906Sasomers } 172300906Sasomers } 173300906Sasomers if (replayed_any) 174300906Sasomers syslog(LOG_INFO, "Finished replaying unconsumed events"); 175300906Sasomers m_replayingEvents = false; 176300906Sasomers} 177300906Sasomers 178300906Sasomersbool 179300906SasomersConsumer::SaveEvent(const Event &event) 180300906Sasomers{ 181300906Sasomers if (m_replayingEvents) 182300906Sasomers return (false); 183300906Sasomers m_unconsumedEvents.push_back(event.DeepCopy()); 184300906Sasomers return (true); 185300906Sasomers} 186300906Sasomers 187300906SasomersEvent * 188300906SasomersConsumer::NextEvent() 189300906Sasomers{ 190300906Sasomers if (!Connected()) 191300906Sasomers return(NULL); 192300906Sasomers 193300906Sasomers Event *event(NULL); 194300906Sasomers try { 195300906Sasomers string evString; 196300906Sasomers 197300906Sasomers evString = ReadEvent(); 198300906Sasomers if (! evString.empty()) { 199300906Sasomers Event::TimestampEventString(evString); 200300906Sasomers event = Event::CreateEvent(m_eventFactory, evString); 201300906Sasomers } 202300906Sasomers } catch (const Exception &exp) { 203300906Sasomers exp.Log(); 204300906Sasomers DisconnectFromDevd(); 205300906Sasomers } 206300906Sasomers return (event); 207300906Sasomers} 208300906Sasomers 209300906Sasomers/* Capture and process buffered events. */ 210300906Sasomersvoid 211300906SasomersConsumer::ProcessEvents() 212300906Sasomers{ 213300906Sasomers Event *event; 214300906Sasomers while ((event = NextEvent()) != NULL) { 215300906Sasomers if (event->Process()) 216300906Sasomers SaveEvent(*event); 217300906Sasomers delete event; 218300906Sasomers } 219300906Sasomers} 220300906Sasomers 221300906Sasomersvoid 222300906SasomersConsumer::FlushEvents() 223300906Sasomers{ 224300906Sasomers std::string s; 225300906Sasomers 226300906Sasomers do 227300906Sasomers s = ReadEvent(); 228300906Sasomers while (! s.empty()) ; 229300906Sasomers} 230300906Sasomers 231300906Sasomersbool 232300906SasomersConsumer::EventsPending() 233300906Sasomers{ 234300906Sasomers struct pollfd fds[1]; 235300906Sasomers int result; 236300906Sasomers 237300906Sasomers do { 238300906Sasomers fds->fd = m_devdSockFD; 239300906Sasomers fds->events = POLLIN; 240300906Sasomers fds->revents = 0; 241300906Sasomers result = poll(fds, NUM_ELEMENTS(fds), /*timeout*/0); 242300906Sasomers } while (result == -1 && errno == EINTR); 243300906Sasomers 244300906Sasomers if (result == -1) 245300906Sasomers err(1, "Polling for devd events failed"); 246300906Sasomers 247300906Sasomers if ((fds->revents & POLLERR) != 0) 248300906Sasomers throw Exception("Consumer::EventsPending(): " 249300906Sasomers "POLLERR detected on devd socket."); 250300906Sasomers 251300906Sasomers if ((fds->revents & POLLHUP) != 0) 252300906Sasomers throw Exception("Consumer::EventsPending(): " 253300906Sasomers "POLLHUP detected on devd socket."); 254300906Sasomers 255300906Sasomers return ((fds->revents & POLLIN) != 0); 256300906Sasomers} 257300906Sasomers 258300906Sasomers} // namespace DevdCtl 259