UPnP

天云上人

/*uPnP.h*/

1. 
   2 n' Y; y. c/ h9 a* ^
2. #ifndef   MYUPNP_H_
   
3. 
   
4. #pragma   once
   
5. 
   ! P6 N: z7 F; c
6. typedef   unsigned   long   ulong;
   
7. 
   
8. class   MyUPnP
   
9. {
   
10. public:
    3 M1 F5 R  ~' N3 _  {9 Q
11. typedef   enum{
    
12. UNAT_OK, //   Successfull
    
13. UNAT_ERROR, //   Error,   use   GetLastError()   to   get   an   error   description
    
14. UNAT_NOT_OWNED_PORTMAPPING, //   Error,   you   are   trying   to   remove   a   port   mapping   not   owned   by   this   class
    3 C$ |* _/ r( r$ N
15. UNAT_EXTERNAL_PORT_IN_USE, //   Error,   you   are   trying   to   add   a   port   mapping   with   an   external   port   in   use
    - \4 r$ A; @! [7 ?
16. UNAT_NOT_IN_LAN //   Error,   you   aren 't   in   a   LAN   ->   no   router   or   firewall
    : X% t$ |5 [2 M6 ]7 p: g6 Q) B7 Z
17. }   UPNPNAT_RETURN;
    , X. `. w9 Z( l2 ?
18. 
    
19. typedef   enum{
    # Z' D. e' z$ x3 r$ T$ |: J
20. UNAT_TCP, //   TCP   Protocol
    
21. UNAT_UDP //   UDP   Protocol
    + k1 k* I( E* d0 L6 V% ]" @: [
22. }   UPNPNAT_PROTOCOL;
    - t: C# b. @+ _3 Z" l
23. 
    8 x& N# H2 T4 ~3 c' ?7 z
24. typedef   struct{
    
25. WORD   internalPort; //   Port   mapping   internal   port
    2 D0 r; L3 e% N1 _3 K& s# G
26. WORD   externalPort; //   Port   mapping   external   port
    
27. UPNPNAT_PROTOCOL   protocol; //   Protocol->   TCP   (UPNPNAT_PROTOCOL:UNAT_TCP)   ||   UDP   (UPNPNAT_PROTOCOL:UNAT_UDP)
    
28. CString   description; //   Port   mapping   description
    / d/ \6 k. M( Z) A
29. }   UPNPNAT_MAPPING;
    ! B% _! l/ B8 s( S# b% b: F; G  c
30. 
    ; ^* k0 Z! F# S' F# H( C% |4 O
31. MyUPnP();
    % G8 z8 w% y0 `6 E$ T
32. ~MyUPnP();
    
33. 
    * ]3 Z) W( @! M4 o" v8 A" d) d
34. UPNPNAT_RETURN   AddNATPortMapping(UPNPNAT_MAPPING   *mapping,   bool   tryRandom   =   false);
    5 s& l. B$ K" f( I8 `2 g
35. UPNPNAT_RETURN   RemoveNATPortMapping(UPNPNAT_MAPPING   mapping,   bool   removeFromList   =   true);
    9 H! ]  J8 X' K3 m, m5 o
36. void   clearNATPortMapping();
    6 m' {) ^( H/ i3 T
37. 
    ! M8 h3 _% |5 K+ h' n
38. CString GetLastError();
    
39. CString GetLocalIPStr();
    : Q, g5 E& h; u( C4 y
40. WORD GetLocalIP();
    , d  c5 b' a% D
41. bool IsLANIP(WORD   nIP);
    : X( X# d7 L/ b1 U6 D2 P
42. 
    0 d, X" X# k4 \- V
43. protected:
    : z8 h. A; b& E# A. d$ v1 P2 Y2 \
44. void InitLocalIP();
    
45. void SetLastError(CString   error);
    7 ?1 A0 x3 d4 M* t- B
46. 
    9 k( ?5 s" ?; v
47. bool   addPortmap(int   eport,   int   iport,   const   CString&   iclient,
    ' ^4 G5 l" t6 w9 w
48.       const   CString&   descri,   const   CString&   type);
    # F* J( p9 P5 t) l  c9 ?! A
49. bool   deletePortmap(int   eport,   const   CString&   type);
    . r9 l+ k- a- j8 y
50. 
    
51. bool isComplete()   const   {   return   !m_controlurl.IsEmpty();   }
    * |" ~# |9 o  v8 V
52. 
    
53. bool Search(int   version=1);
    
54. bool GetDescription();
    # q! M+ P0 r: [; X/ w# H
55. CString GetProperty(const   CString&   name,   CString&   response);
    
56. bool InvokeCommand(const   CString&   name,   const   CString&   args);
    1 _! M  @7 W! ^1 l# x( t' l
57. 
    
58. bool Valid()const{return   (!m_name.IsEmpty()&&!m_description.IsEmpty());}
    
59. bool InternalSearch(int   version);
    
60. CString m_devicename;
    : E0 o$ _# o, x- K
61. CString m_name;
    $ L3 E5 K/ p) }: L: k/ I2 L! t
62. CString m_description;
    % w. c; A. j( r/ W
63. CString m_baseurl;
    
64. CString m_controlurl;
    0 v- z6 P) y6 s
65. CString m_friendlyname;
    0 c/ M  }- W  k+ U1 Q
66. CString m_modelname;
    
67. int m_version;
    " I. O! n; a4 m' F
68. 
    / N  ?9 c/ ?( M- q" N
69. private:
    
70. CList <UPNPNAT_MAPPING,   UPNPNAT_MAPPING>   m_Mappings;
    
71. 
    
72. CString m_slocalIP;
    ) Z! q7 I) F  _) U/ B" A
73. CString m_slastError;
    ' ^- c' _: s6 d) k
74. WORD m_uLocalIP;
    1 H) [% A9 f" z, M; m0 v5 `" I
75. 
    / }, a$ v& M, J# K
76. bool isSearched;
    # ]. q; `' W# X' N" ~! B
77. };
    
78. #endif

复制代码

天云上人

/*UPnP.cpp*/

1. 
   
2. #include   "stdafx.h "
   
3. 
   
4. #include   "upnp.h "
   
5. 
   
6. #define   UPNPPORTMAP0       _T( "WANIPConnection ")
   , G6 D9 c0 F6 s. I7 V# Q. S( f' Z
7. #define   UPNPPORTMAP1       _T( "WANPPPConnection ")
   
8. #define   UPNPGETEXTERNALIP   _T( "GetExternalIPAddress "),_T( "NewExternalIPAddress ")
   ' Z& H- V7 t5 C9 X% G+ }
9. #define   UPNPADDPORTMAP   _T( "AddPortMapping ")
   0 p8 q! `0 C" \
10. #define   UPNPDELPORTMAP   _T( "DeletePortMapping ")
    
11. 
    
12. static   const   ulong UPNPADDR   =   0xFAFFFFEF;
    ( [1 o9 r# P0 P4 y
13. static   const   int UPNPPORT   =   1900;
    5 j- s, m9 O0 o$ O
14. static   const   CString URNPREFIX   =   _T( "urn:schemas-upnp-org: ");
      ]3 `$ n& F% N1 T9 W
15. 
    6 C; C! F8 ]$ K9 a- K, Y
16. const   CString   getString(int   i)
    ' r( o9 a6 S- H* W
17. {
    
18. CString   s;
    
19. 
    ; ~6 K8 {* k( Z' o5 c2 b
20. s.Format(_T( "%d "),   i);
    , e9 u/ |4 H4 {
21. 
    : \* K! n, G9 ]
22. return   s;
    $ X3 x  b6 T/ R6 ^1 a6 y2 X, G
23. }
    2 v8 z( q2 ]5 i: _9 w# `6 q' i
24. 
    / p1 p% i/ g6 [
25. const   CString   GetArgString(const   CString&   name,   const   CString&   value)
    : h1 A% V& ?6 [7 i: t4 p
26. {
    & U# V. q/ {6 M" y
27. return   _T( " < ")   +   name   +   _T( "> ")   +   value   +   _T( " </ ")   +   name   +   _T( "> ");
    
28. }
    
29. 
    
30. const   CString   GetArgString(const   CString&   name,   int   value)
    9 a$ i1 w1 p+ \; J+ M
31. {
    8 E( i& M5 V; X, l! `# i- |
32. return   _T( " < ")   +   name   +   _T( "> ")   +   getString(value)   +   _T( " </ ")   +   name   +   _T( "> ");
    7 e+ n* Z! T+ C7 ^
33. }
    0 j$ f" d0 O2 N# Z! t5 O
34. 
    $ n2 _+ H7 v8 H, R- ^
35. bool   SOAP_action(CString   addr,   uint16   port,   const   CString   request,   CString   &response)
    
36. {
    
37. char   buffer[10240];
    $ P0 V) @; k9 m# `
38. 
    
39. const   CStringA   sa(request);
    
40. int   length   =   sa.GetLength();
    
41. strcpy(buffer,   (const   char*)sa);
    ! n9 q* O! T- E! g6 Y7 |
42. 
    
43. uint32   ip   =   inet_addr(CStringA(addr));
    
44. struct   sockaddr_in   sockaddr;
    
45. memset(&sockaddr,   0,   sizeof(sockaddr));
    & z  Y3 \8 `' n! S  m+ p* t. t3 J9 X
46. sockaddr.sin_family   =   AF_INET;
    
47. sockaddr.sin_port   =   htons(port);
      s  m1 s- l% M' _
48. sockaddr.sin_addr.S_un.S_addr   =   ip;
    7 E" n! E6 W2 x
49. int   s   =   socket(AF_INET,   SOCK_STREAM,   0);
    
50. u_long   lv   =   1;
    
51. ioctlsocket(s,   FIONBIO,   &lv);
    / R# w! h% N) j" I; @
52. connect(s,   (struct   sockaddr   *)&sockaddr,   sizeof(sockaddr));
    
53. Sleep(20);
    
54. int   n   =   send(s,   buffer,   length,   0);
    / l' p0 q% k% N- q. ?
55. Sleep(100);
    5 l9 z4 W: W1 Q
56. int   rlen   =   recv(s,   buffer,   sizeof(buffer),   0);
    % b. f) E: x: i- m0 _7 r3 F
57. closesocket(s);
    
58. if   (rlen   ==   SOCKET_ERROR)   return   false;
    
59. if   (!rlen)   return   false;
    , F8 l% ]" h: Z" y7 s" o
60. 
    
61. response   =   CString(CStringA(buffer,   rlen));
    
62. 
    & W+ i' u- J; x7 V
63. return   true;
    
64. }
    % K6 }5 S1 A" m
65. 
    
66. int   SSDP_sendRequest(int   s,   uint32   ip,   uint16   port,   const   CString&   request)
    
67. {
    
68. char   buffer[10240];
    : y# T. |8 H, y4 ^, n8 p
69. 
    . ^) Y7 [- F# v% b  z+ B
70. const   CStringA   sa(request);
    
71. int   length   =   sa.GetLength();
    6 Z" D4 r0 C2 q) D" b8 g- P
72. strcpy(buffer,   (const   char*)sa);
    4 g2 j. k, x& d% b( W: l) N
73. 
    $ O( }2 H/ X7 ]* j5 I1 \6 z
74. struct   sockaddr_in   sockaddr;
    
75. memset(&sockaddr,   0,   sizeof(sockaddr));
    $ j2 y7 d, O6 G$ g' ?
76. sockaddr.sin_family   =   AF_INET;
    
77. sockaddr.sin_port   =   htons(port);
    
78. sockaddr.sin_addr.S_un.S_addr   =   ip;
    * Y1 X' u! V* m: V
79. 
    
80. return   sendto(s,   buffer,   length,   0,   (struct   sockaddr   *)&sockaddr,   sizeof(sockaddr));
    ' ]' f" w, Q; Q# ]& Z6 }' h
81. }
    3 l* A. M" z, u+ Z8 A
82. 
    9 S6 |% A3 K% _
83. bool   parseHTTPResponse(const   CString&   response,   CString&   result)
    
84. {
    & P. q" J! E# F  b/ l
85. int   pos   =   0;
    
86. 
    
87. CString   status   =   response.Tokenize(_T( "\r\n "),   pos);
    2 q2 R$ @! P" T, m3 @% n* @- }
88. 
      }9 U% q1 F8 ?, J: Z) R
89. result   =   response;
    
90. result.Delete(0,   pos);
    6 e6 {5 y1 G% F( i* U  `- ]
91. 
    . ~1 T0 w  p* @# O0 w$ @
92. pos   =   0;
    
93. status.Tokenize(_T( "   "),   pos);
    , X" O( P1 [, |. x
94. status   =   status.Tokenize(_T( "   "),   pos);
    & i1 }& F# `3 B$ M. w: G' Q* N. k
95. if   (status.IsEmpty()   ||   status[0]!= '2 ')   return   false;
    5 X& X, X% O5 q) n6 l
96. return   true;
    
97. }
    
98. 
    
99. const   CString   getProperty(const   CString&   all,   const   CString&   name)
    
100. {
     " |& R5 p+ \9 D
101. CString   startTag   =   ' < '   +   name   +   '> ';
       R7 Z2 M9 K2 m7 M; a' \+ e% g4 W
102. CString   endTag   =   _T( " </ ")   +   name   +   '> ';
     : p, [& M- g- H0 S
103. CString   property;
     ! F% F) ]3 F7 Q( ^
104. 
     ) g! X1 K1 M; f9 E# `! n5 Q& m
105. int   posStart   =   all.Find(startTag);
     - y4 n- [' W/ d6 e. P
106. if   (posStart <0)   return   CString();
     
107. 
     
108. int   posEnd   =   all.Find(endTag,   posStart);
       |) ?8 z$ j' O9 U6 V, t
109. if   (posStart> =posEnd)   return   CString();
     
110. 
     
111. return   all.Mid(posStart   +   startTag.GetLength(),   posEnd   -   posStart   -   startTag.GetLength());
     
112. }
     
113. 
     
114. MyUPnP::MyUPnP()
     , ?# P& X, W8 p5 C
115. :   m_version(1)
     
116. {
     
117. m_uLocalIP   =   0;
     
118. isSearched   =   false;
     
119. }
     9 B6 A5 x' L, {+ k# Y7 V1 v$ O
120. 
     3 \: _/ G  `  `3 J9 ?7 Z4 i* R) K
121. MyUPnP::~MyUPnP()
     * }4 L; a6 p2 V, P
122. {
     
123. UPNPNAT_MAPPING   search;
     ' i2 ^$ O7 @0 Y* c5 \" K2 b2 N
124. POSITION   pos   =   m_Mappings.GetHeadPosition();
     0 U5 l/ ~+ q* k$ X+ F* P
125. while(pos){
     
126. search   =   m_Mappings.GetNext(pos);
     $ C  e- ?+ S! m! H# h
127. RemoveNATPortMapping(search,   false);
     $ y; a/ |6 \4 h6 V* M  B
128. }
     ; q( h6 ?/ B) `3 J
129. 
     . w2 f, }. a. f/ K" J3 [. ~
130. m_Mappings.RemoveAll();
     
131. }
     
132. 
     
133. 
     # M3 H" C4 o5 O" |
134. bool   MyUPnP::InternalSearch(int   version)
     
135. {
     6 X; s7 l( I  G( G+ U" v' Z- q
136. if(version <=0)version   =   1;
     
137. m_version   =   version;
     
138. 
     5 h4 G. G9 X2 ?2 m" _. m
139. #define   NUMBEROFDEVICES 2
     , B7 o1 J- f% d/ X( X2 e
140. CString   devices[][2]   =   {
     * l8 g" y% J+ F9 E) O! O: m% _
141. {UPNPPORTMAP1,   _T( "service ")},
     * ~1 }. F9 t2 m4 b
142. {UPNPPORTMAP0,   _T( "service ")},
     
143. {_T( "InternetGatewayDevice "),   _T( "device ")},
     ) a0 V1 t: z/ `: y4 A
144. };
     * E; @. [$ V0 x% y# _
145. 
     * P/ v7 R2 t- k! N' L8 W2 \
146. int   s   =   socket(AF_INET,   SOCK_DGRAM,   0);
     1 |; M  `( d  `) [  ?+ U
147. u_long   lv   =   1;
     , C( E0 Q: x* Q, W1 \5 a' w! w8 I
148. ioctlsocket(s,   FIONBIO,   &lv);
     
149. 
     ) N- v6 R. R& B
150. int   rlen   =   0;
     ( o, W3 C8 ~* O' B. `) M2 t
151. for   (int   i=0;   rlen <=0   &&   i <500;   i++)   {
     5 N" N$ O$ O- T+ U1 W7 V2 D
152. if   (!(i%100))   {
     ) r  C# z9 S- q
153. for   (int   i=0;   i <NUMBEROFDEVICES;   i++)   {
     
154. m_name.Format(_T( "%s%s:%s:%d "),   URNPREFIX,   devices[i][1],   devices[i][0],   version);
     # a( T, M( A6 `( L  j: B9 L
155. CString   request;
     ; d( M* y8 g; B/ M
156. request.Format(_T( "M-SEARCH   *   HTTP/1.1\r\nHOST:   239.255.255.250:1900\r\nMAN:   \ "ssdp:discover\ "\r\nMX:   %d\r\nST:   %s\r\n\r\n "),
     
157. 6,   m_name);
     
158. SSDP_sendRequest(s,   UPNPADDR,   UPNPPORT,   request);
     
159. }
     % A0 E& \& A% |: H' C" j
160. }
     
161. 
     ' g6 G  h! }9 k0 h1 p
162. Sleep(10);
     + A" C; L- {; q7 F' f7 V
163. 
     
164. char   buffer[10240];
     $ R# u3 k* J4 m; A3 X/ @
165. rlen   =   recv(s,   buffer,   sizeof(buffer),   0);
     
166. if   (rlen   <=   0)   continue;
     
167. closesocket(s);
     
168. 
       {. v& U$ x; _: C
169. CString   response   =   CString(CStringA(buffer,   rlen));
     8 l" m$ y9 w4 v6 _+ m
170. CString   result;
     + i6 X% O; G! |; O# D/ v. L: N
171. if   (!parseHTTPResponse(response,   result))   return   false;
     4 b* h- b+ Q+ x5 E* h9 m
172. 
     
173. for   (int   d=0;   d <NUMBEROFDEVICES;   d++)   {
     6 Q6 f; w: G7 b) e8 k
174. m_name.Format(_T( "%s%s:%s:%d "),   URNPREFIX,   devices[d][1],   devices[d][0],   version);
     
175. if   (result.Find(m_name)   > =   0)   {
     " M9 a* m& x/ s" T6 c$ f3 }
176. for   (int   pos   =   0;;)   {
       I  I! \$ s4 H+ O0 n1 d
177. CString   line   =   result.Tokenize(_T( "\r\n "),   pos);
     
178. if   (line.IsEmpty())   return   false;
     
179. CString   name   =   line.Mid(0,   9);
     7 C5 u. q) y; z0 U- X
180. name.MakeUpper();
     
181. if   (name   ==   _T( "LOCATION: "))   {
     ! n' }" d% B  ?
182. line.Delete(0,   9);
     * r% Q8 ]2 ]1 J1 _( o+ V  C
183. m_description   =   line;
     % ?& R! v" C) C$ w+ ^6 ?6 y. x
184. m_description.Trim();
     # ]) \8 y/ ~2 ~0 q% F" s  M+ ?" l2 h
185. return   GetDescription();
     
186. }
     
187. }
     
188. }
     
189. }
     
190. }
     
191. closesocket(s);
     ; E- f9 Y8 G  K" m) P" E
192. 
     
193. return   false;
     
194. }
       w6 E2 ?) m9 f. }2 C: B

复制代码

天云上人

以下有关upnp的接口来自emule，


///////////////////////////////////////////
& N! q& d/ ]# C, k) [//upnp的抽象接口类,只需调用这些接口即可实现端口映射功能.


#pragma once
* b% _# z; t. o+ U$ h#include <exception>

; Q5 J) V/ ~" l. _6 R' ^8 ?7 r1 E
( D! W6 c0 V5 H  enum TRISTATE{
        TRIS_FALSE,
, Z: B# H9 m7 I8 r0 Q        TRIS_UNKNOWN,
        TRIS_TRUE
};
& G9 b: r: m1 ]  P  o7 \, N7 Y% F8 @

enum UPNP_IMPLEMENTATION{
6 `1 G& A2 |% A7 o8 p5 O        UPNP_IMPL_WINDOWSERVICE = 0,
1 p6 A( E1 `$ h, x# H  f& L        UPNP_IMPL_MINIUPNPLIB,
        UPNP_IMPL_NONE /*last*/
1 ]3 T4 N* A! Z% w* K0 Y};

& \" j, u/ J% k' _1 t4 w$ h
% D2 n) U7 \2 X
/ D7 J: W1 n0 _; D7 Y, \5 c9 C8 U
3 l5 E5 i1 V# O6 ~" `' U" wclass CUPnPImpl
{
$ _8 S. t7 [6 J, fpublic:
        CUPnPImpl();
8 `7 ]  N; M" g) m$ n        virtual ~CUPnPImpl();
        struct UPnPError : std::exception {};
        enum {
) a# z: @2 E- P1 k                UPNP_OK,
4 \# e4 `2 p# q, l5 |; \8 N: m                UPNP_FAILED,
                UPNP_TIMEOUT
        };
$ J7 ?" R9 \8 I4 [. k- w9 S
: U, V' H2 ~. A
        virtual void        StartDiscovery(uint16 nTCPPort, uint16 nUDPPort, uint16 nTCPWebPort) = 0;
        virtual bool        CheckAndRefresh() = 0;
        virtual void        StopAsyncFind() = 0;
        virtual void        DeletePorts() = 0;
' e4 `. h2 R$ G9 x1 O3 x: K  g+ B, W        virtual bool        IsReady() = 0;
        virtual int                GetImplementationID() = 0;
6 i: N& n5 w: R% O  r3 O4 |: G/ b       
4 t- e! B  a) u0 q        void                        LateEnableWebServerPort(uint16 nPort); // Add Webserverport on already installed portmapping

: X* `! V. n9 b
        void                        SetMessageOnResult(HWND hWindow, UINT nMessageID);
6 s; v6 t6 m% u! t        TRISTATE                ArePortsForwarded() const                                                                { return m_bUPnPPortsForwarded; }
        uint16                        GetUsedTCPPort()                                                                                { return m_nTCPPort; }
        uint16                        GetUsedUDPPort()                                                                                { return m_nUDPPort; }       
5 ~3 u/ d0 V& j( S  m8 |
% M! P: a7 Y! O; f3 o
5 @+ y9 e, |6 i6 M/ ^// Implementation
protected:
/ o3 o5 H% q% R/ @7 X' T# s$ S9 J        volatile TRISTATE        m_bUPnPPortsForwarded;
        void                                SendResultMessage();
        uint16                                m_nUDPPort;
: u2 M- I( H# ~5 n# _, p, T        uint16                                m_nTCPPort;
# |# O, O3 y4 ^) L* n        uint16                                m_nTCPWebPort;
        bool                                m_bCheckAndRefresh;

$ r" f' A9 J& _: S
private:
) j6 l( r" @1 k1 a: l6 k4 p        HWND        m_hResultMessageWindow;
! r3 N) A2 @/ a5 ?2 ~! G# \% m        UINT        m_nResultMessageID;
4 F2 O8 V, M  r: F$ i6 |  U2 c
9 `( J! c% |( z  K8 o/ D8 a
* j4 x6 ^+ K* _! F7 x};
8 t( z/ M: W+ ^  b5 D- S
7 a1 ~5 I6 R+ X0 m3 A
% O& V: A% S0 c% A& {# p* [// Dummy Implementation to be used when no other implementation is available
class CUPnPImplNone: public CUPnPImpl
{
/ ~9 z9 L$ _; W: Dpublic:
7 B! |# o) N9 x$ G  P. g        virtual void        StartDiscovery(uint16, uint16, uint16)                                        { ASSERT( false ); }
7 t  }7 x% \8 Y& }2 K& s        virtual bool        CheckAndRefresh()                                                                                { return false; }
        virtual void        StopAsyncFind()                                                                                        { }
" E4 ~( O, Y: I) }; u  I( Z9 C        virtual void        DeletePorts()                                                                                        { }
7 \) D' [( \" ~3 ?! t  P; T% r* a        virtual bool        IsReady()                                                                                                { return false; }
        virtual int                GetImplementationID()                                                                        { return UPNP_IMPL_NONE; }
};
  Q. i- p# O/ F8 F
0 @% J8 Z* J# _. e' h2 m" ^
+ [+ a, W9 N* N  M5 ?2 _/////////////////////////////////////
1 j' o4 V. T5 f3 a% _4 B//下面是使用windows操作系统自带的UPNP功能的子类

' |7 P9 t8 H8 S$ y
! X/ _+ Y- B7 e% L, Z#pragma once
4 }0 B8 Q0 ]; L! ]#pragma warning( disable: 4355 )
  |# I; z+ p/ A/ ?8 O
. X  h: w, L4 J4 x; Q: U) q
. w# m. ^9 I0 ^, {3 C: o/ t4 D! I#include "UPnPImpl.h"
#include <upnp.h>
1 P$ A* P) q9 t0 s) y, U#include <iphlpapi.h>
5 x+ h. v2 c1 E#include <comdef.h>
4 `& P) `& F9 i8 n#include <winsvc.h>
8 E+ m- R' h4 h' T
3 n& v" E9 [: ^  R
5 x9 g( B, h4 Q1 L9 j; H+ b#include <vector>
, Y2 z: j) Q9 Q# b4 @#include <exception>
#include <functional>
0 X5 f1 J! I7 U# P) F
! ?% D* S/ G; m7 ~1 `
. B6 N- ^8 b, w* B. p4 P. N: @2 _

typedef _com_ptr_t<_com_IIID<IUPnPDeviceFinder,&IID_IUPnPDeviceFinder> >        FinderPointer;
typedef _com_ptr_t<_com_IIID<IUPnPDevice,&IID_IUPnPDevice>        >                                DevicePointer;
typedef _com_ptr_t<_com_IIID<IUPnPService,&IID_IUPnPService> >                                ServicePointer;
typedef        _com_ptr_t<_com_IIID<IUPnPDeviceFinderCallback,&IID_IUPnPDeviceFinderCallback> > DeviceFinderCallback;
typedef        _com_ptr_t<_com_IIID<IUPnPServiceCallback,&IID_IUPnPServiceCallback> >                        ServiceCallback;
typedef _com_ptr_t<_com_IIID<IEnumUnknown,&IID_IEnumUnknown> >                                EnumUnknownPtr;
typedef _com_ptr_t<_com_IIID<IUnknown,&IID_IUnknown> >                                                UnknownPtr;


typedef DWORD (WINAPI* TGetBestInterface) (
  IPAddr dwDestAddr,
  PDWORD pdwBestIfIndex
8 B& |* q: ~, ]% l+ p0 C1 L7 z);
3 v2 y# V; q- J+ {4 a- y

+ i1 m. N' ]/ t9 Z/ otypedef DWORD (WINAPI* TGetIpAddrTable) (
$ D. q' F/ r4 S" u3 N% l  PMIB_IPADDRTABLE pIpAddrTable,
2 q# d$ T# M/ M+ a  PULONG pdwSize,
0 ^  A5 R& u: b  BOOL bOrder
- J8 ?& e$ @& R$ {$ r. l' ?);
4 A* X& W2 h1 Y; r* ?/ A

" |6 N. R  i- a% f) D( ]typedef DWORD (WINAPI* TGetIfEntry) (
5 a: u( C/ K3 a" Z) e& X  PMIB_IFROW pIfRow
);
1 R! a: W- D4 E8 u+ x( ~  l# m
% x! J0 k4 Z" s9 Z# i8 s% R7 _
CString translateUPnPResult(HRESULT hr);
& C. \% K5 _& R& s- l& n# zHRESULT UPnPMessage(HRESULT hr);
0 m% m& w- t: m

class CUPnPImplWinServ: public CUPnPImpl
5 b. A" v6 B% z0 y6 w% y{
        friend class CDeviceFinderCallback;
4 s9 D0 @7 Z' e, b. |) N        friend class CServiceCallback;
// Construction
public:
        virtual ~CUPnPImplWinServ();
" N9 t# T( }1 V8 T8 k$ Y! |# _        CUPnPImplWinServ();


        virtual void        StartDiscovery(uint16 nTCPPort, uint16 nUDPPort, uint16 nTCPWebPort)                { StartDiscovery(nTCPPort, nUDPPort, nTCPWebPort, false); }
' b* B/ x! Q/ `; `        virtual void        StopAsyncFind();
( u% G! b* v( `# K2 v        virtual void        DeletePorts();
$ h2 C( l4 _4 E        virtual bool        IsReady();
        virtual int                GetImplementationID()                                                                        { return UPNP_IMPL_WINDOWSERVICE; }
- S+ K* e7 x1 T* K

        // No Support for Refreshing on this  (fallback) implementation yet - in many cases where it would be needed (router reset etc)
- g6 D6 Z/ X  w+ b        // the windows side of the implementation tends to get bugged untill reboot anyway. Still might get added later
        virtual bool        CheckAndRefresh()                                                                                { return false; };
6 c' n+ I+ |: V) N

protected:
6 g& T( h5 ]+ c1 `' L* B        void        StartDiscovery(uint16 nTCPPort, uint16 nUDPPort, uint16 nTCPWebPort, bool bSecondTry);
        void        AddDevice(DevicePointer pDevice, bool bAddChilds, int nLevel = 0);
        void        RemoveDevice(CComBSTR bsUDN);
* ?6 c/ q( U7 e0 g8 {- h) H, e        bool        OnSearchComplete();
. f& J9 `: p0 H- d1 R: D' E        void        Init();
% J# b) L$ v5 q# O# H, p

        inline bool IsAsyncFindRunning()
% X2 W! j0 p9 v9 g: O2 k  x! K        {
$ U7 Y  N: V( w) u$ Y7 z( J                if ( m_pDeviceFinder != NULL && m_bAsyncFindRunning && GetTickCount() - m_tLastEvent > 10000 )
                {
                        m_pDeviceFinder->CancelAsyncFind( m_nAsyncFindHandle );
                        m_bAsyncFindRunning = false;
% Z# L- I  c; B2 v6 S7 T                }
5 g9 P9 C; W0 n( q7 V  U                MSG msg;
                while ( PeekMessage( &msg, NULL, 0, 0, PM_REMOVE ) )
                {
                        TranslateMessage( &msg );
) i# Y5 |; m# s% ]  t8 n4 g% }                        DispatchMessage( &msg );
+ c+ ]3 S. b1 \1 S: {& r* c                }
8 Y# N* W3 s; G                return m_bAsyncFindRunning;
& Y4 j( ]0 K5 [        }


        TRISTATE                        m_bUPnPDeviceConnected;

& s# {: [2 o7 b
// Implementation
, d8 X5 i1 N7 q* b        // API functions
9 B" y. U! l0 j" O8 x        SC_HANDLE (WINAPI *m_pfnOpenSCManager)(LPCTSTR, LPCTSTR, DWORD);
. x9 B7 |) `5 S  i- b% @6 [        SC_HANDLE (WINAPI *m_pfnOpenService)(SC_HANDLE, LPCTSTR, DWORD);
6 y2 s( n0 `: k8 @/ U        BOOL (WINAPI *m_pfnQueryServiceStatusEx)(SC_HANDLE, SC_STATUS_TYPE, LPBYTE, DWORD, LPDWORD);
        BOOL (WINAPI *m_pfnCloseServiceHandle)(SC_HANDLE);
; z) q9 m- w( y8 f2 G1 m        BOOL (WINAPI *m_pfnStartService)(SC_HANDLE, DWORD, LPCTSTR*);
% X7 [: \, O- T+ s0 L4 b3 U        BOOL (WINAPI *m_pfnControlService)(SC_HANDLE, DWORD, LPSERVICE_STATUS);
* f5 W' p1 \; w9 Y8 x
5 t4 Q( l* f/ ]+ }5 j4 h
        TGetBestInterface                m_pfGetBestInterface;
$ C- A# z# P6 k+ k3 @        TGetIpAddrTable                        m_pfGetIpAddrTable;
5 X  N: g& n8 L+ S: V  V        TGetIfEntry                                m_pfGetIfEntry;

1 Z9 o. X' P5 G" K& z: e
+ V# b8 q* i5 Z* c1 t5 j        static FinderPointer CreateFinderInstance();
        struct FindDevice : std::unary_function< DevicePointer, bool >
& p1 C) }+ G' K' K        {
% j' d* j2 [* C8 i  }, N" U" _- z                FindDevice(const CComBSTR& udn) : m_udn( udn ) {}
                result_type operator()(argument_type device) const
+ Z' o# ]+ J9 y" Q! k; J3 m1 W                {
) o) ~, g  }2 l8 Z3 u, ?( {                        CComBSTR deviceName;
0 _, b- V' T5 W+ \7 [( S                        HRESULT hr = device->get_UniqueDeviceName( &deviceName );
: |) u* K8 q( P3 s

) c8 W3 d+ w6 A6 G9 D  _" }7 ]. C% f                        if ( FAILED( hr ) )
8 ]1 `) ]8 s( y                                return UPnPMessage( hr ), false;

' f; o% Z4 s, a& F# g1 w
                        return wcscmp( deviceName.m_str, m_udn ) == 0;
                }
                CComBSTR m_udn;
* s! [. o. A+ y        };
8 w: g' P/ L: f. }! y( ^4 {. C5 [       
        void        ProcessAsyncFind(CComBSTR bsSearchType);
, T, i1 x# O* U  J+ G        HRESULT        GetDeviceServices(DevicePointer pDevice);
        void        StartPortMapping();
        HRESULT        MapPort(const ServicePointer& service);
        void        DeleteExistingPortMappings(ServicePointer pService);
$ @" x. G, \* s) Y        void        CreatePortMappings(ServicePointer pService);
        HRESULT SaveServices(EnumUnknownPtr pEU, const LONG nTotalItems);
        HRESULT InvokeAction(ServicePointer pService, CComBSTR action,
  x5 O6 Z, w. e; x1 O3 M- ]                LPCTSTR pszInArgString, CString& strResult);
* I2 X6 \# D$ G: ^. \        void        StopUPnPService();


& q: h' F$ w3 C; p$ f0 F- I# \        // Utility functions
        HRESULT CreateSafeArray(const VARTYPE vt, const ULONG nArgs, SAFEARRAY** ppsa);
        INT_PTR CreateVarFromString(const CString& strArgs, VARIANT*** pppVars);
        INT_PTR        GetStringFromOutArgs(const VARIANT* pvaOutArgs, CString& strArgs);
. d8 n* y  C' [, C4 S+ P) P6 w# |        void        DestroyVars(const INT_PTR nCount, VARIANT*** pppVars);
        HRESULT GetSafeArrayBounds(SAFEARRAY* psa, LONG* pLBound, LONG* pUBound);
% e4 a8 U! \: R$ R; t5 i* U# i8 E        HRESULT GetVariantElement(SAFEARRAY* psa, LONG pos, VARIANT* pvar);
        CString        GetLocalRoutableIP(ServicePointer pService);


// Private members
( E+ M! J7 ^- iprivate:
        DWORD        m_tLastEvent;        // When the last event was received?
8 ]7 `) C: {' N" |3 ~$ |        std::vector< DevicePointer >  m_pDevices;
        std::vector< ServicePointer > m_pServices;
        FinderPointer                        m_pDeviceFinder;
        DeviceFinderCallback        m_pDeviceFinderCallback;
/ q8 C8 ^& I* T& f        ServiceCallback                        m_pServiceCallback;
( x( Y- w* r4 s

8 v2 }& h  t) P# Q# y) D" l        LONG        m_nAsyncFindHandle;
        bool        m_bCOM;
        bool        m_bPortIsFree;
        CString m_sLocalIP;
: Q! K0 F* @" }8 ]6 M! t! P        CString m_sExternalIP;
+ \! O* L  E/ E        bool        m_bADSL;                // Is the device ADSL?
) L7 Q0 D7 T  {' ?: V9 ?        bool        m_ADSLFailed;        // Did port mapping failed for the ADSL device?
        bool        m_bInited;
1 N, u0 A& I" B$ h8 ]3 ~6 _        bool        m_bAsyncFindRunning;
% g) @: k4 ?. F2 n- O; X+ m' y3 j; B        HMODULE m_hADVAPI32_DLL;
        HMODULE        m_hIPHLPAPI_DLL;
8 |. m1 G* y5 W" v' O/ L        bool        m_bSecondTry;
( I2 |5 S0 i; e8 U0 X# y        bool        m_bServiceStartedByEmule;
        bool        m_bDisableWANIPSetup;
        bool        m_bDisableWANPPPSetup;
4 l+ g% R" f) |( w1 r8 }& Z3 H
! s" ?1 x% a7 k5 I
8 w9 s3 r+ g6 @};
4 o1 a# O1 O7 V/ e. n
% m5 h" @! G! N4 r
// DeviceFinder Callback
. G" y+ n! f& z4 U4 ~class CDeviceFinderCallback
! O* {9 O  f9 h9 o1 F: [- }& J1 r        : public IUPnPDeviceFinderCallback
{
; ^6 n( o% }# ~1 F! Apublic:
% m5 ^; A, r/ `3 N. q        CDeviceFinderCallback(CUPnPImplWinServ& instance)
                : m_instance( instance )
4 o1 x( B# Z2 G" r2 y! F        { m_lRefCount = 0; }
, f! {% {* h# v  o# Y& h

   STDMETHODIMP QueryInterface(REFIID iid, LPVOID* ppvObject);
# Q2 q& u8 e# r   STDMETHODIMP_(ULONG) AddRef();
# u0 H# F, l2 c0 ]5 D3 t: q   STDMETHODIMP_(ULONG) Release();

2 [# U" d& D; A  o' f6 [6 u
  f1 f3 U& i# d. G4 X// implementation
! `! t. r% [; T( Zprivate:
        HRESULT __stdcall DeviceAdded(LONG nFindData, IUPnPDevice* pDevice);
; h4 {) |6 c" V! _8 \, G        HRESULT __stdcall DeviceRemoved(LONG nFindData, BSTR bsUDN);
        HRESULT __stdcall SearchComplete(LONG nFindData);
4 v4 x) {- A6 j) a9 L! e9 V/ c$ T+ v
- z& S$ `: a! U- d! D% n9 V
6 e3 s! z0 j5 }5 Wprivate:
        CUPnPImplWinServ& m_instance;
        LONG m_lRefCount;
, h) Z- @9 ]5 c9 {$ H. j1 Z- h$ {};
1 B0 Q" N' v. O4 e' ~
+ w2 `. ~, W* v
// Service Callback
class CServiceCallback
        : public IUPnPServiceCallback
{
public:
        CServiceCallback(CUPnPImplWinServ& instance)
* q1 P8 X5 ?7 e) ?0 x( \- q                : m_instance( instance )
* U4 Y; W. M9 F4 s$ w$ s( z, N        { m_lRefCount = 0; }
   
   STDMETHODIMP QueryInterface(REFIID iid, LPVOID* ppvObject);
   STDMETHODIMP_(ULONG) AddRef();
   STDMETHODIMP_(ULONG) Release();
8 x- p3 ?: [$ k8 Y) w

// implementation
5 Y8 v2 s$ l& I* K; j* T7 U# \7 Kprivate:
        HRESULT __stdcall StateVariableChanged(IUPnPService* pService, LPCWSTR pszStateVarName, VARIANT varValue);
) `" ~" |/ t% @4 u        HRESULT __stdcall ServiceInstanceDied(IUPnPService* pService);
( I& B6 G$ i) t/ Q' B0 u8 r

private:
        CUPnPImplWinServ& m_instance;
" S- v8 q: L; x, v  m        LONG m_lRefCount;
};

$ ^' @! b: d4 y, N" m/ b: D
) G. Z( T3 F; H- h5 q/////////////////////////////////////////////////
- u8 s$ R9 K3 x) _6 g
8 m. @8 Y- n+ c: K
使用时只需要使用抽象类的接口。
CUPnPImpl::SetMessageOnResult设置需要接受UPNP端口映射是否成功的窗口句柄和消息ID.
8 q4 C/ v& p$ Y  B5 Y4 jCUPnPImpl::StartDiscovery将开启一个异步设备查找并进行端口映射,其参数为需要映射的内网端口.
: c( Z" \  C: g( l+ H3 d) WCUPnPImpl::StopAsyncFind停止设备查找.
$ w9 S# A' \! qCUPnPImpl::DeletePorts删除端口映射.