1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
|
/*
* Copyright (C) 2021- Team Kodi
* This file is part of Kodi - https://kodi.tv
*
* SPDX-License-Identifier: GPL-2.0-or-later
* See LICENSES/README.md for more information.
*/
#pragma once
#include "network/IWSDiscovery.h"
#include "threads/CriticalSection.h"
#include "threads/SingleLock.h"
#include "platform/posix/filesystem/SMBWSDiscoveryListener.h"
#include <memory>
#include <string>
#include <vector>
class CFileItemList;
namespace WSDiscovery
{
class CWSDiscoveryListenerUDP;
}
namespace WSDiscovery
{
struct wsd_req_info
{
std::string action; // ToDo: Action probably isnt required to be stored
std::string msgid;
std::string types; // ToDo: Types may not be needed.
std::string address;
std::string xaddrs;
std::string xaddrs_host;
std::string computer;
bool operator==(const wsd_req_info& item) const
{
return ((item.xaddrs == xaddrs) && (item.address == address));
}
};
class CWSDiscoveryPosix : public WSDiscovery::IWSDiscovery
{
public:
CWSDiscoveryPosix();
// IWSDiscovery interface methods
~CWSDiscoveryPosix() override;
bool StartServices() override;
bool StopServices() override;
bool IsRunning() override;
/*
* Get List of smb servers found by WSD
* out (CFileItemList&) List of fileitems populated with smb addresses
* return (bool) true if >0 WSD addresses found
*/
bool GetServerList(CFileItemList& items);
long long GetInstanceID() const { return wsd_instance_id; }
/*
* Set List of WSD info request
* out (vector<wsd_req_info>) vector of WSD responses received
* return void
*/
void SetItems(std::vector<WSDiscovery::wsd_req_info> entries);
/*
* Lookup host name in collected ws-discovery data
* in (const std::string&) Host name
* out (std::string&) IP address if found
* return (bool) true if found
*/
bool GetCached(const std::string& strHostName, std::string& strIpAddress);
static bool IsInitialized() { return m_isInitialized; }
private:
CCriticalSection m_critWSD;
/*
* As per docs - Pg32 - http://specs.xmlsoap.org/ws/2005/04/discovery/ws-discovery.pdf
* MUST be incremented by >= 1 each time the service has gone down, lost state,
* and came back up again. SHOULD NOT be incremented otherwise. Means to set
* this value include, but are not limited to:
* • A counter that is incremented on each 'cold' boot
* • The boot time of the service, expressed as seconds elapsed since midnight
* January 1, 1970
*
* Our implementation will only set this on creation of the class
* We arent providing services to clients, so this should be ok.
*/
long long wsd_instance_id;
// WSD UDP daemon
std::unique_ptr<WSDiscovery::CWSDiscoveryListenerUDP> m_WSDListenerUDP;
std::vector<WSDiscovery::wsd_req_info> m_vecWSDInfo;
static std::atomic<bool> m_isInitialized;
};
} // namespace WSDiscovery
|
