/*
 * Copyright 2018, Jérôme Duval, jerome.duval@gmail.com.
 * Distributed under the terms of the MIT License.
 */
#ifndef VIRTIO_BALLOON_PRIVATE_H
#define VIRTIO_BALLOON_PRIVATE_H


#include <condition_variable.h>
#include <lock.h>
#include <virtio.h>
#include <vm/vm_page.h>


//#define TRACE_VIRTIO_BALLOON
#ifdef TRACE_VIRTIO_BALLOON
#	define TRACE(x...) dprintf("virtio_balloon: " x)
#else
#	define TRACE(x...) ;
#endif
#define ERROR(x...)			dprintf("\33[33mvirtio_rng:\33[0m " x)
#define CALLED() 			TRACE("CALLED %s\n", __PRETTY_FUNCTION__)


extern device_manager_info* gDeviceManager;


#define PAGES_COUNT	256


struct PageInfo : DoublyLinkedListLinkImpl<PageInfo> {
	vm_page *page;
};


typedef DoublyLinkedList<PageInfo> PageInfoList;



class VirtioBalloonDevice {
public:
								VirtioBalloonDevice(device_node* node);
								~VirtioBalloonDevice();

			status_t			InitCheck();

private:
	static	void				_ConfigCallback(void* driverCookie);
	static	void				_QueueCallback(void* driverCookie,
									void *cookie);

	static	int32				_ThreadEntry(void *data);
			int32				_Thread();

			uint32				_DesiredSize();
			status_t			_UpdateSize();

			device_node*		fNode;

			virtio_device_interface* fVirtio;
			virtio_device*		fVirtioDevice;

			status_t			fStatus;
			uint64				fFeatures;
			::virtio_queue		fVirtioQueues[2];

			physical_entry 		fEntry;
			uint32				fBuffer[PAGES_COUNT];

			thread_id			fThread;
			uint32				fDesiredSize;
			uint32				fActualSize;

			spinlock			fInterruptLock;
			ConditionVariable	fQueueCondition;
			ConditionVariable	fConfigCondition;
			bool				fRunning;

			PageInfoList		fPages;
};


#endif // VIRTIO_BALLOON_PRIVATE_H