tools/gator/daemon/AnnotateListener.cpp

/**
 * Copyright (C) ARM Limited 2014. All rights reserved.
 *
 * This program is free software; you can redistribute it and/or modify
 * it under the terms of the GNU General Public License version 2 as
 * published by the Free Software Foundation.
 */

#include "AnnotateListener.h"

#include <unistd.h>

#include "OlySocket.h"

struct AnnotateClient {
	AnnotateClient *next;
	int fd;
};

AnnotateListener::AnnotateListener() : mClients(NULL), mSock(NULL) {
}

AnnotateListener::~AnnotateListener() {
	close();
	delete mSock;
}

void AnnotateListener::setup() {
	mSock = new OlyServerSocket(8082);
}

int AnnotateListener::getFd() {
	return mSock->getFd();
}

void AnnotateListener::handle() {
	AnnotateClient *const client = new AnnotateClient();
	client->fd = mSock->acceptConnection();
	client->next = mClients;
	mClients = client;
}

void AnnotateListener::close() {
	mSock->closeServerSocket();
	while (mClients != NULL) {
		::close(mClients->fd);
		AnnotateClient *next = mClients->next;
		delete mClients;
		mClients = next;
	}
}

void AnnotateListener::signal() {
	const char ch = 0;
	AnnotateClient **ptr = &mClients;
	AnnotateClient *client = mClients;
	while (client != NULL) {
		if (write(client->fd, &ch, sizeof(ch)) != 1) {
			::close(client->fd);
			AnnotateClient *next = client->next;
			delete client;
			*ptr = next;
			client = next;
			continue;
		}
		ptr = &client->next;
		client = client->next;
	}
}