/* DirectPlay Lite
 * Copyright (C) 2018 Daniel Collins <solemnwarning@solemnwarning.net>
 *
 * This program is free software; you can redistribute it and/or modify
 * it under the terms of the GNU General Public License as published by
 * the Free Software Foundation; either version 2 of the License, or
 * (at your option) any later version.
 *
 * This program is distributed in the hope that it will be useful,
 * but WITHOUT ANY WARRANTY; without even the implied warranty of
 * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
 * GNU General Public License for more details.
 *
 * You should have received a copy of the GNU General Public License along
 * with this program; if not, write to the Free Software Foundation, Inc.,
 * 51 Franklin Street, Fifth Floor, Boston, MA 02110-1301 USA.
*/

#ifndef DPLITE_HOSTENUMERATOR_HPP
#define DPLITE_HOSTENUMERATOR_HPP

#include <winsock2.h>
#include <dplay8.h>
#include <functional>
#include <thread>
#include <vector>
#include <windows.h>

#include "network.hpp"

#define DEFAULT_ENUM_COUNT    5
#define DEFAULT_ENUM_INTERVAL 1500
#define DEFAULT_ENUM_TIMEOUT  1500

class HostEnumerator
{
	private:
		/* No copy c'tor. */
		HostEnumerator(const HostEnumerator&) = delete;
		
		/* Pointer to the global refcount (if in use), for instantiating DirectPlay8Address objects. */
		std::atomic<unsigned int> * const global_refcount;
		
		/* DirectPlay message handler and context value */
		PFNDPNMESSAGEHANDLER message_handler;
		PVOID message_handler_ctx;
		
		std::function<void(HRESULT)> complete_cb;
		
		GUID service_provider;
		struct sockaddr_in send_addr;
		
		GUID application_guid;                 /* GUID of application to search for, or GUID_NULL */
		std::vector<unsigned char> user_data;  /* Data to include in request. */
		
		DWORD tx_remain;   /* Number of remaining requests to transmit, may be INFINITE. */
		DWORD tx_interval; /* Number of milliseconds to wait between transmits. */
		DWORD rx_timeout;  /* Number of milliseconds to wait for replies to a request. */
		
		void *user_context; /* DPNMSG_ENUM_HOSTS_REPONSE.pvUserContext */
		
		DWORD next_tx_at;
		DWORD stop_at;
		
		int sock;
		HANDLE wake_thread;
		std::thread *thread;
		bool req_cancel;
		
		unsigned char recv_buf[MAX_PACKET_SIZE];
		
		void main();
		void handle_packet(const void *data, size_t size, struct sockaddr_in *from_addr);
		
	public:
		HostEnumerator(
			std::atomic<unsigned int> * const global_refcount,
			
			PFNDPNMESSAGEHANDLER message_handler,
			PVOID message_handler_ctx,
			
			PDPN_APPLICATION_DESC const pApplicationDesc,
			IDirectPlay8Address *const pdpaddrHost,
			IDirectPlay8Address *const pdpaddrDeviceInfo,
			PVOID const pvUserEnumData,
			const DWORD dwUserEnumDataSize,
			const DWORD dwEnumCount,
			const DWORD dwRetryInterval,
			const DWORD dwTimeOut,
			PVOID const pvUserContext,
			
			std::function<void(HRESULT)> complete_cb
		);
		
		~HostEnumerator();
		
		void cancel();
		void wait();
};

#endif /* !DPLITE_HOSTENUMERATOR_HPP */