mirror of
https://github.com/solemnwarning/directplay-lite
synced 2024-12-30 16:45:37 +01:00
102 lines
3.0 KiB
C++
102 lines
3.0 KiB
C++
/* 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 */
|