void create(ClientMap& clientMap, char ID[], char plainPwd[], TABLE& id_table)
{
//clientMap.showMap();
if(clientMap.search(ID) != nullptr) {
printf("ID %s exists, ", ID);
// show 10 unused ID
int count = 0;
NextWordGenerator gen(ID);
while(count != 10)
{
char *next = gen.getNext();
if(clientMap.search(next) == nullptr) // if the ID doesn't exist, show it
{
printf("%s%c", next, (count == 9) ? '\n' : ',');
count++;
}
}
return;
}
int pos = id_table.insert(ID);
char encryptedPwd[110]; md5(encryptedPwd, plainPwd);
clientMap.insert(ID, encryptedPwd, pos);
printf("success\n");
return;
}
void LogForwardThread::forwardMessage(LogMessage *msg)
{
#ifdef DUMP_PACKET
QList<QByteArray>::const_iterator it = msg->begin();
int i = 0;
for (; it != msg->end(); ++it, i++)
{
QByteArray buf = *it;
cout << i << ":\t" << buf.size() << endl << "\t"
<< buf.toHex().constData() << endl << "\t"
<< buf.constData() << endl;
}
#endif
msgsSinceHeartbeat.ref();
// First section is the client id
QByteArray clientBa = msg->first();
QString clientId = QString(clientBa.toHex());
QByteArray json = msg->at(1);
if (json.size() == 0)
{
// This is either a ping response or a first gasp
logClientMapMutex.lock();
LoggerListItem *logItem = logClientMap.value(clientId, NULL);
logClientMapMutex.unlock();
if (!logItem)
{
// Send an initial ping so the client knows we are in the house
emit pingClient(clientId);
}
else
{
// cout << "pong " << clientId.toLocal8Bit().constData() << endl;
loggingGetTimeStamp(&logItem->epoch, NULL);
}
return;
}
QMutexLocker lock(&logClientMapMutex);
LoggerListItem *logItem = logClientMap.value(clientId, NULL);
// cout << "msg " << clientId.toLocal8Bit().constData() << endl;
if (logItem)
{
loggingGetTimeStamp(&logItem->epoch, NULL);
}
else
{
LoggingItem *item = LoggingItem::create(json);
logClientCount.ref();
LOG(VB_FILE, LOG_DEBUG, QString("New Logging Client: ID: %1 (#%2)")
.arg(clientId).arg(logClientCount.fetchAndAddOrdered(0)));
#ifndef NOLOGSERVER
// TODO FIXME This is not thread-safe!
if (logClientCount.fetchAndAddOrdered(0) > 1 && m_shutdownTimer &&
m_shutdownTimer->isActive())
{
LOG(VB_GENERAL, LOG_INFO, "Aborting shutdown timer");
m_shutdownTimer->stop();
}
#endif
QMutexLocker lock2(&loggerMapMutex);
QMutexLocker lock3(&logRevClientMapMutex);
// Need to find or create the loggers
LoggerList *loggers = new LoggerList;
LoggerBase *logger;
// FileLogger from logFile
QString logfile = item->logFile();
logfile.detach();
if (!logfile.isEmpty())
{
logger = FileLogger::create(logfile, lock2.mutex());
ClientList *clients = logRevClientMap.value(logger);
if (clients)
clients->insert(0, clientId);
if (logger && loggers)
loggers->insert(0, logger);
}
#ifndef _WIN32
// SyslogLogger from facility
int facility = item->facility();
#if CONFIG_SYSTEMD_JOURNAL
if ((facility > 0) || (facility == SYSTEMD_JOURNAL_FACILITY))
{
logger = SyslogLogger::create(lock2.mutex(), facility > 0);
#else
if (facility > 0)
{
logger = SyslogLogger::create(lock2.mutex());
#endif
ClientList *clients = logRevClientMap.value(logger);
if (clients)
clients->insert(0, clientId);
if (logger && loggers)
loggers->insert(0, logger);
}
#endif
// DatabaseLogger from table
QString table = item->table();
if (!table.isEmpty())
{
logger = DatabaseLogger::create(table, lock2.mutex());
ClientList *clients = logRevClientMap.value(logger);
if (clients)
clients->insert(0, clientId);
if (logger && loggers)
loggers->insert(0, logger);
}
logItem = new LoggerListItem;
loggingGetTimeStamp(&logItem->epoch, NULL);
logItem->list = loggers;
logClientMap.insert(clientId, logItem);
item->DecrRef();
}
#ifndef NOLOGSERVER
m_zmqPubSock->sendMessage(*msg);
#else
if (logItem && logItem->list && !logItem->list->isEmpty())
{
LoggerList::iterator it = logItem->list->begin();
LoggingItem *item = LoggingItem::create(json);
if (!item)
return;
for (; it != logItem->list->end(); ++it)
{
(*it)->logmsg(item);
}
item->DecrRef();
}
#endif
}
/// \brief Stop the thread by setting the abort flag
void LogForwardThread::stop(void)
{
m_aborted = true;
}