UPnP

天云上人

/*uPnP.h*/

1. 
   
2. #ifndef   MYUPNP_H_
   
3. 
   ) s& a! k5 |+ p- i; a$ y
4. #pragma   once
   9 g+ @' f" M. N+ e* P6 r; P
5. 
   : `: _. e6 g% @' B/ E; P
6. typedef   unsigned   long   ulong;
   2 q3 K% `* u! O; o+ J5 A# }
7. 
   ( s0 J; n4 \9 r0 _
8. class   MyUPnP
   
9. {
   2 @) k: P6 z) q
10. public:
    
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
    5 M. e( D# e9 F! h- S
15. UNAT_EXTERNAL_PORT_IN_USE, //   Error,   you   are   trying   to   add   a   port   mapping   with   an   external   port   in   use
    " g& E) Z* |( Q' ~  K
16. UNAT_NOT_IN_LAN //   Error,   you   aren 't   in   a   LAN   ->   no   router   or   firewall
    
17. }   UPNPNAT_RETURN;
    
18. 
    
19. typedef   enum{
    
20. UNAT_TCP, //   TCP   Protocol
    7 H2 `) s) ]7 [1 y  N$ f; T
21. UNAT_UDP //   UDP   Protocol
    
22. }   UPNPNAT_PROTOCOL;
    * D1 `4 N4 x0 J3 v+ Q/ B; \* K2 K6 }
23. 
    - G) z3 X# {2 P
24. typedef   struct{
    0 o- |/ j/ C3 |0 \/ s3 \2 k: }
25. WORD   internalPort; //   Port   mapping   internal   port
    
26. WORD   externalPort; //   Port   mapping   external   port
    - j! K% f- j9 G4 p) @
27. UPNPNAT_PROTOCOL   protocol; //   Protocol->   TCP   (UPNPNAT_PROTOCOL:UNAT_TCP)   ||   UDP   (UPNPNAT_PROTOCOL:UNAT_UDP)
    
28. CString   description; //   Port   mapping   description
    
29. }   UPNPNAT_MAPPING;
    9 u# c0 d# }4 w2 R$ B
30. 
    ' `7 t$ D* u& F2 C+ W3 f
31. MyUPnP();
    
32. ~MyUPnP();
    4 V) u% `$ I3 l3 g7 n
33. 
    9 |7 `- h6 c% b, A# b$ y% G( I
34. UPNPNAT_RETURN   AddNATPortMapping(UPNPNAT_MAPPING   *mapping,   bool   tryRandom   =   false);
    2 U2 F* A4 }! S4 _) s/ D$ L
35. UPNPNAT_RETURN   RemoveNATPortMapping(UPNPNAT_MAPPING   mapping,   bool   removeFromList   =   true);
    " d+ w) a8 ~0 s4 q( q) a! E* X' W
36. void   clearNATPortMapping();
    4 t8 }- y4 }* _2 x. J
37. 
    
38. CString GetLastError();
    ; ?- J2 N, e) |6 E$ e% E
39. CString GetLocalIPStr();
    / x  b0 u& r1 W& u5 i& |
40. WORD GetLocalIP();
    
41. bool IsLANIP(WORD   nIP);
    # [# [3 u& m' I: ^% _" M
42. 
    + q7 d. p. E2 l/ @
43. protected:
    5 L  S( _' L$ n# R1 a
44. void InitLocalIP();
    ( |& i# d. H- D! B. T% q
45. void SetLastError(CString   error);
    
46. 
    " H+ }3 y5 g$ j) x3 y
47. bool   addPortmap(int   eport,   int   iport,   const   CString&   iclient,
    " L8 G9 ]# m8 i/ c8 ^4 }
48.       const   CString&   descri,   const   CString&   type);
    + K0 T8 g8 D6 u7 T% {! M4 q
49. bool   deletePortmap(int   eport,   const   CString&   type);
    * N; O8 ]2 a. O: X
50. 
    
51. bool isComplete()   const   {   return   !m_controlurl.IsEmpty();   }
    ; r& n5 ~2 ]/ r. |' @
52. 
    
53. bool Search(int   version=1);
    ! t4 N* b5 z! B0 \
54. bool GetDescription();
    
55. CString GetProperty(const   CString&   name,   CString&   response);
    6 m" ]. F1 q: n+ D
56. bool InvokeCommand(const   CString&   name,   const   CString&   args);
    $ |9 i- j8 C$ q% i, I; z' i
57. 
    - N0 [$ t* O6 a# T5 |
58. bool Valid()const{return   (!m_name.IsEmpty()&&!m_description.IsEmpty());}
    
59. bool InternalSearch(int   version);
    
60. CString m_devicename;
    * S% f2 W! p3 c1 d6 W  H4 \. A' J1 O. N
61. CString m_name;
    # U; i% S1 L6 |* r' {0 @; m
62. CString m_description;
    
63. CString m_baseurl;
    * j) D, I- y# ]/ A
64. CString m_controlurl;
    
65. CString m_friendlyname;
    
66. CString m_modelname;
    
67. int m_version;
    8 _" S, h; j5 F' g, j# Q
68. 
    
69. private:
    
70. CList <UPNPNAT_MAPPING,   UPNPNAT_MAPPING>   m_Mappings;
    : D' \& [9 j* w/ Y, U8 ?
71. 
    ! V$ w. b6 r0 ]+ W
72. CString m_slocalIP;
    ! M" n) ?0 \6 Y# N2 C
73. CString m_slastError;
    % J) {% Y2 {* z2 @
74. WORD m_uLocalIP;
    
75. 
    ) O/ {/ I. M  x8 M7 A7 p
76. bool isSearched;
    
77. };
    
78. #endif

复制代码

天云上人

/*UPnP.cpp*/

1. 
   
2. #include   "stdafx.h "
   5 ~; n* I2 K% d: B. Z% ~
3. 
     G6 p# X, {8 u  V  a+ V# _
4. #include   "upnp.h "
   4 ^6 f) ?# h0 Z" M
5. 
   5 e) c$ L8 O4 J3 F, v9 x, g
6. #define   UPNPPORTMAP0       _T( "WANIPConnection ")
     Y) Y6 ?5 O7 j
7. #define   UPNPPORTMAP1       _T( "WANPPPConnection ")
   
8. #define   UPNPGETEXTERNALIP   _T( "GetExternalIPAddress "),_T( "NewExternalIPAddress ")
   
9. #define   UPNPADDPORTMAP   _T( "AddPortMapping ")
   
10. #define   UPNPDELPORTMAP   _T( "DeletePortMapping ")
    ) F% G% t+ m3 f$ U( G6 G
11. 
    + @5 q; i6 h+ L+ ?' I3 h) `
12. static   const   ulong UPNPADDR   =   0xFAFFFFEF;
    % Z* m5 Z$ ]. O- g& O" R
13. static   const   int UPNPPORT   =   1900;
    8 n, a5 y9 |7 ~" \2 k8 f
14. static   const   CString URNPREFIX   =   _T( "urn:schemas-upnp-org: ");
    0 z+ z. ^4 V3 Q/ {; u* Y* H5 R9 B7 F
15. 
    
16. const   CString   getString(int   i)
    7 W; m. z  r7 E4 b6 ^
17. {
    
18. CString   s;
    + U  l( h( p% O. L2 N' _7 S8 d
19. 
    ( T! l- ?# T( V; W) [
20. s.Format(_T( "%d "),   i);
    4 l8 S. |: N: s7 l3 m" A
21. 
    ; D9 G' ?" e( L3 t4 y! c
22. return   s;
    
23. }
    
24. 
    
25. const   CString   GetArgString(const   CString&   name,   const   CString&   value)
    
26. {
    
27. return   _T( " < ")   +   name   +   _T( "> ")   +   value   +   _T( " </ ")   +   name   +   _T( "> ");
    
28. }
    ) z- J: H* `& ]$ Y) V/ T1 u; A
29. 
    4 o8 ], a% L6 i9 J1 {: R2 ?
30. const   CString   GetArgString(const   CString&   name,   int   value)
    
31. {
    
32. return   _T( " < ")   +   name   +   _T( "> ")   +   getString(value)   +   _T( " </ ")   +   name   +   _T( "> ");
    ( _1 c: }/ U+ D  G4 {% B
33. }
    
34. 
    ' w# E  e6 ]' Z0 R  E( j; e
35. bool   SOAP_action(CString   addr,   uint16   port,   const   CString   request,   CString   &response)
    
36. {
    . {# ?3 R: u3 O& N
37. char   buffer[10240];
    0 z. ^( [9 M1 q% T$ |. p! E- ^
38. 
    
39. const   CStringA   sa(request);
    
40. int   length   =   sa.GetLength();
    
41. strcpy(buffer,   (const   char*)sa);
    
42. 
    & \' {1 ]) r, ]- B; B( L3 M5 }
43. uint32   ip   =   inet_addr(CStringA(addr));
    + N" k6 Q( S9 F" ]% ]8 z
44. struct   sockaddr_in   sockaddr;
    & v/ a$ L7 s8 |' G
45. memset(&sockaddr,   0,   sizeof(sockaddr));
    
46. sockaddr.sin_family   =   AF_INET;
    4 l0 p) _/ R. q+ K' }$ F
47. sockaddr.sin_port   =   htons(port);
    6 q0 z* g8 p8 Q7 L- Y
48. sockaddr.sin_addr.S_un.S_addr   =   ip;
    
49. int   s   =   socket(AF_INET,   SOCK_STREAM,   0);
      I7 d$ X. A" s2 ~0 V
50. u_long   lv   =   1;
    # J0 R7 Y6 k( V1 q
51. ioctlsocket(s,   FIONBIO,   &lv);
    ! v' k# ~) l  [* V# U
52. connect(s,   (struct   sockaddr   *)&sockaddr,   sizeof(sockaddr));
    6 C$ _/ m" }( M1 f3 Q. Z
53. Sleep(20);
    4 R' b- l# V, {9 `/ K) ^
54. int   n   =   send(s,   buffer,   length,   0);
    
55. Sleep(100);
    
56. int   rlen   =   recv(s,   buffer,   sizeof(buffer),   0);
    / q" p  z2 Y" @. v% u
57. closesocket(s);
    6 f- u; W7 w# x! D" t8 F
58. if   (rlen   ==   SOCKET_ERROR)   return   false;
    
59. if   (!rlen)   return   false;
    
60. 
    2 d/ x2 ?- R/ l1 {
61. response   =   CString(CStringA(buffer,   rlen));
    ' W8 ?; j: h: F# T4 M
62. 
    
63. return   true;
    # `" ~2 `. v* W" G3 {$ f, a
64. }
    
65. 
    2 }# t7 u/ a: |6 d8 r  E
66. int   SSDP_sendRequest(int   s,   uint32   ip,   uint16   port,   const   CString&   request)
    , W+ ~# \& }8 |! K0 D7 O
67. {
    
68. char   buffer[10240];
    # g# U; ]# |* a+ E$ K
69. 
    
70. const   CStringA   sa(request);
    9 v# ~) w, f& x8 A6 B1 X
71. int   length   =   sa.GetLength();
    . f2 Y' N2 l1 N1 O
72. strcpy(buffer,   (const   char*)sa);
    
73. 
    , y) D% @; z: J3 T- Z. S( t/ N
74. struct   sockaddr_in   sockaddr;
    
75. memset(&sockaddr,   0,   sizeof(sockaddr));
    
76. sockaddr.sin_family   =   AF_INET;
    
77. sockaddr.sin_port   =   htons(port);
    1 [0 a( P! u/ f
78. sockaddr.sin_addr.S_un.S_addr   =   ip;
      o/ t0 C0 |* Y
79. 
    
80. return   sendto(s,   buffer,   length,   0,   (struct   sockaddr   *)&sockaddr,   sizeof(sockaddr));
    
81. }
    
82. 
    
83. bool   parseHTTPResponse(const   CString&   response,   CString&   result)
    
84. {
    
85. int   pos   =   0;
    . [  x0 }4 ?4 C  E  y0 ~; d3 V
86. 
    
87. CString   status   =   response.Tokenize(_T( "\r\n "),   pos);
    8 ^- r* [6 q: [, U$ \
88. 
    
89. result   =   response;
    , D2 p. K9 ^4 A9 n2 z6 M: d1 L" c
90. result.Delete(0,   pos);
      K/ `) h. G; M9 l0 _
91. 
    
92. pos   =   0;
    3 R3 {3 v- a  P/ Y0 i
93. status.Tokenize(_T( "   "),   pos);
    $ ^4 _3 s, V. m: E
94. status   =   status.Tokenize(_T( "   "),   pos);
    0 \5 y) I4 E- u. R
95. if   (status.IsEmpty()   ||   status[0]!= '2 ')   return   false;
    ; X8 ]  X4 z" T$ }
96. return   true;
    . {! ?, C3 o8 @+ N; H. L
97. }
    
98. 
    
99. const   CString   getProperty(const   CString&   all,   const   CString&   name)
    ) g% S4 B( U  O& H2 }5 k6 W
100. {
     " y% \- F- ^- S/ n# H( s  j
101. CString   startTag   =   ' < '   +   name   +   '> ';
     
102. CString   endTag   =   _T( " </ ")   +   name   +   '> ';
     
103. CString   property;
     
104. 
     . O: j: b& z: m3 a
105. int   posStart   =   all.Find(startTag);
     ' ~7 \' q$ F1 V; H% O$ r3 S
106. if   (posStart <0)   return   CString();
     * T3 h6 [3 L) {8 H4 j4 h
107. 
     
108. int   posEnd   =   all.Find(endTag,   posStart);
     . P- M5 x" l7 a; M: l3 H
109. if   (posStart> =posEnd)   return   CString();
     
110. 
     7 V8 r. ]" |( f6 s
111. return   all.Mid(posStart   +   startTag.GetLength(),   posEnd   -   posStart   -   startTag.GetLength());
     : q! Q7 R6 ~9 j; A, i
112. }
     
113. 
     + r1 L5 U3 X2 K/ X: V
114. MyUPnP::MyUPnP()
     
115. :   m_version(1)
     ( O, N# {/ _5 R$ a5 r' e
116. {
     
117. m_uLocalIP   =   0;
     9 `2 P/ z) D; `! \
118. isSearched   =   false;
     
119. }
     
120. 
     
121. MyUPnP::~MyUPnP()
     4 `, M! q' d, A
122. {
     
123. UPNPNAT_MAPPING   search;
     " B- y8 t( y+ t( N. t; y( L) ~% k
124. POSITION   pos   =   m_Mappings.GetHeadPosition();
     
125. while(pos){
       w) Y3 a! q5 p( w" }# M; Q: e3 M1 s
126. search   =   m_Mappings.GetNext(pos);
     1 m. {  A, E4 p2 u# C7 f# J* n
127. RemoveNATPortMapping(search,   false);
     
128. }
     ; a: g+ ]6 U1 Z0 y. t- c( c9 z2 z8 N
129. 
     : G+ Q* u7 d9 ^7 B$ f8 o7 b% Y: V
130. m_Mappings.RemoveAll();
     4 ?7 F1 \8 ?8 ~1 x5 X; L
131. }
       p9 h7 B" L6 @+ u8 V" R
132. 
     
133. 
     . Q  x3 o$ M2 c! h% e
134. bool   MyUPnP::InternalSearch(int   version)
     2 D/ _& Y; Q/ V& |' L' @" n/ l  j+ S
135. {
     
136. if(version <=0)version   =   1;
     
137. m_version   =   version;
     - B2 ~8 O2 E$ X! k% ~
138. 
     
139. #define   NUMBEROFDEVICES 2
     
140. CString   devices[][2]   =   {
     + I% ~9 J* U- x' ~! I% m- P
141. {UPNPPORTMAP1,   _T( "service ")},
     
142. {UPNPPORTMAP0,   _T( "service ")},
     3 v" m' J2 K: X% p
143. {_T( "InternetGatewayDevice "),   _T( "device ")},
     : n) J" A1 x/ O: I/ Y: [
144. };
     
145. 
     - m7 Z& g1 Q! a5 i; J& C! B
146. int   s   =   socket(AF_INET,   SOCK_DGRAM,   0);
     ; p0 d! t$ n8 i$ }/ h8 t" N
147. u_long   lv   =   1;
     
148. ioctlsocket(s,   FIONBIO,   &lv);
     
149. 
     ! H5 S: j: H( Q! d4 O5 ^* s
150. int   rlen   =   0;
     
151. for   (int   i=0;   rlen <=0   &&   i <500;   i++)   {
     
152. if   (!(i%100))   {
     , [) |/ H+ }; i+ c, R  G0 u! y
153. for   (int   i=0;   i <NUMBEROFDEVICES;   i++)   {
     . u! R5 ^7 Q  X6 g& J% l
154. m_name.Format(_T( "%s%s:%s:%d "),   URNPREFIX,   devices[i][1],   devices[i][0],   version);
     
155. CString   request;
     
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);
     / O$ x8 P, ~# ^: r$ D; `$ M& X
158. SSDP_sendRequest(s,   UPNPADDR,   UPNPPORT,   request);
     2 C- ?0 A8 m. v( p  U! I' r4 @" x
159. }
     
160. }
     
161. 
     
162. Sleep(10);
     
163. 
     
164. char   buffer[10240];
     6 z  z0 T' y% C# d* y- [
165. rlen   =   recv(s,   buffer,   sizeof(buffer),   0);
     " O8 j# m9 S, R' l, a- {
166. if   (rlen   <=   0)   continue;
     
167. closesocket(s);
     - n% z# {0 d! |
168. 
     0 v/ C) S1 o& r  o* c. ~
169. CString   response   =   CString(CStringA(buffer,   rlen));
     7 e, _7 w7 I; c; a
170. CString   result;
     / z. r, h) n) e
171. if   (!parseHTTPResponse(response,   result))   return   false;
     5 E6 p7 w- B* o9 M. v6 Q) m
172. 
     
173. for   (int   d=0;   d <NUMBEROFDEVICES;   d++)   {
     
174. m_name.Format(_T( "%s%s:%s:%d "),   URNPREFIX,   devices[d][1],   devices[d][0],   version);
     
175. if   (result.Find(m_name)   > =   0)   {
     
176. for   (int   pos   =   0;;)   {
     
177. CString   line   =   result.Tokenize(_T( "\r\n "),   pos);
     
178. if   (line.IsEmpty())   return   false;
     
179. CString   name   =   line.Mid(0,   9);
     + t" c/ P. B7 A$ x2 Y* C" _
180. name.MakeUpper();
     
181. if   (name   ==   _T( "LOCATION: "))   {
     
182. line.Delete(0,   9);
     
183. m_description   =   line;
     4 f" ]: e8 G" C/ j
184. m_description.Trim();
     ) G; r. J" m; i; Z# d
185. return   GetDescription();
     ; V6 C  Q" K& F& m3 o
186. }
     
187. }
     
188. }
     * n9 @8 j0 U" H  L* o/ Y
189. }
     
190. }
     6 y! i  u7 w$ h  g) g- J
191. closesocket(s);
     
192. 
     
193. return   false;
     $ }! I+ I1 Q9 V" V5 }1 }
194. }
     

复制代码

天云上人

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


///////////////////////////////////////////
0 b# F. T' J- `; `  O4 q//upnp的抽象接口类,只需调用这些接口即可实现端口映射功能.

' W+ {% G0 K& r$ E
7 E/ \5 b5 @9 s- L7 t#pragma once
#include <exception>


  enum TRISTATE{
        TRIS_FALSE,
2 P; J2 m! N) C        TRIS_UNKNOWN,
        TRIS_TRUE
: |6 {# |1 m$ v: U6 b: z2 \};

$ C5 `0 h) e) G& ^- z- p3 Q; @
* n: ~% Z" V+ {- benum UPNP_IMPLEMENTATION{
/ u- Y& g+ u, u# Y9 ]        UPNP_IMPL_WINDOWSERVICE = 0,
7 i. s1 \# p3 i. F" v        UPNP_IMPL_MINIUPNPLIB,
+ J, W" z$ `8 ?: w6 z( H2 E. d        UPNP_IMPL_NONE /*last*/
* W0 c1 K/ }0 M) l};
1 Z) _& Z& N$ h) F; n



. o- k0 f; n. _* D) Y5 d5 `. @. mclass CUPnPImpl
& A) W  G% U( g$ Q$ |{
! n8 z* J9 C% l1 P9 ]public:
, Y  ~& Q* R) i4 Q        CUPnPImpl();
        virtual ~CUPnPImpl();
        struct UPnPError : std::exception {};
& D( ?6 G' X! p; R7 e5 {) S8 Q" r        enum {
% e" k" U! _3 o3 R                UPNP_OK,
0 r1 g: j. d$ k; A) a                UPNP_FAILED,
# {5 U0 A2 P( }2 Z5 G( o  h                UPNP_TIMEOUT
        };
. R1 d" a1 t6 r5 Q

. e0 V; @- R( L% R        virtual void        StartDiscovery(uint16 nTCPPort, uint16 nUDPPort, uint16 nTCPWebPort) = 0;
        virtual bool        CheckAndRefresh() = 0;
$ n, O5 ~2 ^* Q& K$ E/ J2 v        virtual void        StopAsyncFind() = 0;
        virtual void        DeletePorts() = 0;
        virtual bool        IsReady() = 0;
        virtual int                GetImplementationID() = 0;
, m$ e' @% S- t' A5 s# y$ @       
7 m* r2 I& L5 `- w& o        void                        LateEnableWebServerPort(uint16 nPort); // Add Webserverport on already installed portmapping
/ G0 w  F( N8 Y; ~" V$ }5 H  m

* z% ]6 J4 W/ U) f        void                        SetMessageOnResult(HWND hWindow, UINT nMessageID);
        TRISTATE                ArePortsForwarded() const                                                                { return m_bUPnPPortsForwarded; }
! T/ q+ K" [  t- h        uint16                        GetUsedTCPPort()                                                                                { return m_nTCPPort; }
        uint16                        GetUsedUDPPort()                                                                                { return m_nUDPPort; }       

0 j# c' k# t/ j' u' m, W
; h, s+ W8 J* C: R" z* a* T// Implementation
protected:
- Z. J# A, s/ `        volatile TRISTATE        m_bUPnPPortsForwarded;
        void                                SendResultMessage();
5 n& ~5 e, A/ o% k9 i        uint16                                m_nUDPPort;
        uint16                                m_nTCPPort;
$ j) C: O$ R' f# a4 B        uint16                                m_nTCPWebPort;
        bool                                m_bCheckAndRefresh;


/ J8 }& {  X) ]9 C4 g2 y; g2 Pprivate:
+ C7 t5 p* H8 W8 G2 y5 Z, y! Q& t        HWND        m_hResultMessageWindow;
: _& [5 Q1 _% b. n        UINT        m_nResultMessageID;
$ D! `7 v; Z2 B

};
- [( S  T8 q. [0 m# S

// Dummy Implementation to be used when no other implementation is available
# d  D+ P& a: L- ]8 z# K8 C: Zclass CUPnPImplNone: public CUPnPImpl
{
! y4 ]+ y- ~' V0 p6 Apublic:
        virtual void        StartDiscovery(uint16, uint16, uint16)                                        { ASSERT( false ); }
        virtual bool        CheckAndRefresh()                                                                                { return false; }
; @+ n/ I/ U) X8 L) g        virtual void        StopAsyncFind()                                                                                        { }
        virtual void        DeletePorts()                                                                                        { }
        virtual bool        IsReady()                                                                                                { return false; }
. }9 G5 v' D- \6 S2 K4 e        virtual int                GetImplementationID()                                                                        { return UPNP_IMPL_NONE; }
};
, Q0 p5 c- g& @9 }8 {
7 S' O" X4 Y' i
! M- d9 [( }/ D& T" G+ S/ s* z/////////////////////////////////////
  n: C1 h2 }. \* V- W//下面是使用windows操作系统自带的UPNP功能的子类


( i5 M$ t7 n4 `$ s/ L, X4 o#pragma once
9 N; i' q9 o4 ]* h* W3 e1 _% E#pragma warning( disable: 4355 )
2 t5 ]! _) [9 d% n& X

#include "UPnPImpl.h"
#include <upnp.h>
#include <iphlpapi.h>
8 i/ D1 n& X( W4 o' L+ C$ e; ]#include <comdef.h>
#include <winsvc.h>
0 i: R1 E! H" i$ g

#include <vector>
1 d& h, Q6 H0 D" F" G& a#include <exception>
#include <functional>
. E0 z. j& q" ~: t: J3 ]
. G: Z- p: G* k' C% M9 I/ Z' ^' u


typedef _com_ptr_t<_com_IIID<IUPnPDeviceFinder,&IID_IUPnPDeviceFinder> >        FinderPointer;
" z3 _& V* |$ T) h. a+ H" D5 Atypedef _com_ptr_t<_com_IIID<IUPnPDevice,&IID_IUPnPDevice>        >                                DevicePointer;
! o* @, a9 t$ ]" Ytypedef _com_ptr_t<_com_IIID<IUPnPService,&IID_IUPnPService> >                                ServicePointer;
typedef        _com_ptr_t<_com_IIID<IUPnPDeviceFinderCallback,&IID_IUPnPDeviceFinderCallback> > DeviceFinderCallback;
3 o! T( f8 Z. O+ f: \5 btypedef        _com_ptr_t<_com_IIID<IUPnPServiceCallback,&IID_IUPnPServiceCallback> >                        ServiceCallback;
typedef _com_ptr_t<_com_IIID<IEnumUnknown,&IID_IEnumUnknown> >                                EnumUnknownPtr;
. x2 g# ^  V) C3 B" V% a' ytypedef _com_ptr_t<_com_IIID<IUnknown,&IID_IUnknown> >                                                UnknownPtr;
8 a( {0 m* _$ [7 Y; a) w& Q
* J# g/ Q. B; f4 S" D2 D2 I
- `; T" X: J  a. ~: ttypedef DWORD (WINAPI* TGetBestInterface) (
* L* C( G! C$ L1 U2 r& C- z  IPAddr dwDestAddr,
0 R. x7 D; Y* q' h+ F/ ^. c  PDWORD pdwBestIfIndex
! P# j) Q4 N& k2 L! s);
- n( J6 L; }& U5 G6 e. B

4 l. ~, _' U. {5 i* c- p5 Z" Ptypedef DWORD (WINAPI* TGetIpAddrTable) (
  PMIB_IPADDRTABLE pIpAddrTable,
9 A. R% X: p: I$ b  PULONG pdwSize,
% h6 V9 e/ `* q; @3 b% W" N  BOOL bOrder
);


  W% g% W! q: T1 Ftypedef DWORD (WINAPI* TGetIfEntry) (
9 q- K1 z8 `2 V  PMIB_IFROW pIfRow
: E5 a5 m6 [- m2 N$ Y! f4 n);
2 ~; X8 [) v, G1 H' T- z; m5 O! |; H, r
# ]5 P( Y5 A: b- O
; U5 d" n0 {0 E5 q* b  X/ mCString translateUPnPResult(HRESULT hr);
HRESULT UPnPMessage(HRESULT hr);
3 ?; j% e- e$ f+ @$ J

) n! `1 \- \5 Z7 n& Nclass CUPnPImplWinServ: public CUPnPImpl
{
; Q4 I3 w( P0 c  `        friend class CDeviceFinderCallback;
0 Q+ q( D3 E1 ?+ A. l6 {$ m  O        friend class CServiceCallback;
0 c! v8 _1 w' Q' n, }& D' k0 r// Construction
" g3 A0 z! O# s7 cpublic:
( y/ }" f- b& o* C& X        virtual ~CUPnPImplWinServ();
        CUPnPImplWinServ();


        virtual void        StartDiscovery(uint16 nTCPPort, uint16 nUDPPort, uint16 nTCPWebPort)                { StartDiscovery(nTCPPort, nUDPPort, nTCPWebPort, false); }
        virtual void        StopAsyncFind();
        virtual void        DeletePorts();
+ E2 R% s) Q2 H. F+ }/ w        virtual bool        IsReady();
        virtual int                GetImplementationID()                                                                        { return UPNP_IMPL_WINDOWSERVICE; }

; }  h! {, v1 R1 L* K& ]
        // No Support for Refreshing on this  (fallback) implementation yet - in many cases where it would be needed (router reset etc)
        // the windows side of the implementation tends to get bugged untill reboot anyway. Still might get added later
& O, S) J9 {  \- F: {5 S! x        virtual bool        CheckAndRefresh()                                                                                { return false; };


5 ~- p/ k, @) n' T0 H# aprotected:
        void        StartDiscovery(uint16 nTCPPort, uint16 nUDPPort, uint16 nTCPWebPort, bool bSecondTry);
        void        AddDevice(DevicePointer pDevice, bool bAddChilds, int nLevel = 0);
        void        RemoveDevice(CComBSTR bsUDN);
9 J8 f% o( g- x: s0 Q4 l3 w  o        bool        OnSearchComplete();
5 Q( o) p/ M- h$ Y. ~# O$ f        void        Init();
$ R7 Z6 e/ _9 g3 w
( O& ^' _! e$ m* ^9 a% O6 s
; g$ W' w" y  G" \        inline bool IsAsyncFindRunning()
        {
/ B: \4 ~  {6 e                if ( m_pDeviceFinder != NULL && m_bAsyncFindRunning && GetTickCount() - m_tLastEvent > 10000 )
" @9 [4 _) Y1 a5 K2 c; J                {
3 D) e1 e9 n3 O' U: z5 K                        m_pDeviceFinder->CancelAsyncFind( m_nAsyncFindHandle );
& H+ R6 a6 l( S: `/ \$ c) w' L. T; q                        m_bAsyncFindRunning = false;
                }
! |& w1 T" `9 B( X                MSG msg;
                while ( PeekMessage( &msg, NULL, 0, 0, PM_REMOVE ) )
                {
                        TranslateMessage( &msg );
; ~  w$ R; [. ~9 j* C9 H  L                        DispatchMessage( &msg );
                }
8 h1 I9 L( d9 P6 r8 y1 K                return m_bAsyncFindRunning;
        }

' F+ V) S) f; u$ ~6 U* b
4 ?5 l+ v. v# d$ Y2 E6 V        TRISTATE                        m_bUPnPDeviceConnected;
$ n; ^3 B& p( r7 A
% c* R) x: k6 A! u) {
$ [3 e  X) F" k8 L) q' B$ j// Implementation
+ ~( J2 n3 B! _: X        // API functions
* Q9 C4 E. A' w' z: w$ ?1 v2 L        SC_HANDLE (WINAPI *m_pfnOpenSCManager)(LPCTSTR, LPCTSTR, DWORD);
* b& u2 @: u! f+ E- R        SC_HANDLE (WINAPI *m_pfnOpenService)(SC_HANDLE, LPCTSTR, DWORD);
3 t/ U! P* X$ k( z) F! `& @        BOOL (WINAPI *m_pfnQueryServiceStatusEx)(SC_HANDLE, SC_STATUS_TYPE, LPBYTE, DWORD, LPDWORD);
        BOOL (WINAPI *m_pfnCloseServiceHandle)(SC_HANDLE);
6 J) n* h+ X4 Y3 k1 p  l  T        BOOL (WINAPI *m_pfnStartService)(SC_HANDLE, DWORD, LPCTSTR*);
. O& [2 Q; o' q$ O/ i        BOOL (WINAPI *m_pfnControlService)(SC_HANDLE, DWORD, LPSERVICE_STATUS);


& ]: w9 m) V! l% i# i* _        TGetBestInterface                m_pfGetBestInterface;
0 ^6 F3 M' z3 i3 w* V) A- H        TGetIpAddrTable                        m_pfGetIpAddrTable;
0 D9 d  d6 ]7 u0 O  b8 c# D        TGetIfEntry                                m_pfGetIfEntry;
& R* h0 J/ H% B2 m
( o: ]0 ~# o( ^
        static FinderPointer CreateFinderInstance();
        struct FindDevice : std::unary_function< DevicePointer, bool >
        {
, e/ n+ @9 G0 x4 X' }                FindDevice(const CComBSTR& udn) : m_udn( udn ) {}
                result_type operator()(argument_type device) const
                {
                        CComBSTR deviceName;
                        HRESULT hr = device->get_UniqueDeviceName( &deviceName );


                        if ( FAILED( hr ) )
                                return UPnPMessage( hr ), false;

- K* X! G- P4 f, a  _$ Y
                        return wcscmp( deviceName.m_str, m_udn ) == 0;
5 f  U* |- t1 h2 x- J/ q" C                }
# E/ h. h. ?& N' a& s                CComBSTR m_udn;
        };
; i+ s$ X# J! P: M6 {8 A5 @% V       
        void        ProcessAsyncFind(CComBSTR bsSearchType);
        HRESULT        GetDeviceServices(DevicePointer pDevice);
        void        StartPortMapping();
0 H8 W9 l7 }2 O% |+ X  v5 k        HRESULT        MapPort(const ServicePointer& service);
4 `2 J) f8 y5 A, \        void        DeleteExistingPortMappings(ServicePointer pService);
. w: j  E9 Y. r7 C2 s: w+ N        void        CreatePortMappings(ServicePointer pService);
        HRESULT SaveServices(EnumUnknownPtr pEU, const LONG nTotalItems);
        HRESULT InvokeAction(ServicePointer pService, CComBSTR action,
                LPCTSTR pszInArgString, CString& strResult);
        void        StopUPnPService();


6 u2 ]) v, A  Y8 O1 [. G6 n6 v8 u        // Utility functions
        HRESULT CreateSafeArray(const VARTYPE vt, const ULONG nArgs, SAFEARRAY** ppsa);
$ i6 D' D8 p; Q0 `' D        INT_PTR CreateVarFromString(const CString& strArgs, VARIANT*** pppVars);
3 P6 w7 S8 F& e4 Q5 y        INT_PTR        GetStringFromOutArgs(const VARIANT* pvaOutArgs, CString& strArgs);
3 B2 M* \% `( o+ e  j8 P        void        DestroyVars(const INT_PTR nCount, VARIANT*** pppVars);
' V- B+ k7 v2 Q  Y: n- _4 M1 x        HRESULT GetSafeArrayBounds(SAFEARRAY* psa, LONG* pLBound, LONG* pUBound);
        HRESULT GetVariantElement(SAFEARRAY* psa, LONG pos, VARIANT* pvar);
, S7 L/ W* d) ^2 X, W3 V. c5 e        CString        GetLocalRoutableIP(ServicePointer pService);

7 f( U; o% `2 \1 F( z4 _! S  l
. A) C1 a0 I. _0 N// Private members
. B& O* W+ c5 x) \  yprivate:
        DWORD        m_tLastEvent;        // When the last event was received?
2 |, m7 z8 k7 R3 ^        std::vector< DevicePointer >  m_pDevices;
) g/ a4 o) n8 A$ |        std::vector< ServicePointer > m_pServices;
        FinderPointer                        m_pDeviceFinder;
        DeviceFinderCallback        m_pDeviceFinderCallback;
        ServiceCallback                        m_pServiceCallback;
7 W2 P, C6 d) T4 O/ r9 ?4 {* f( E9 P

) P0 y) s$ }" |8 Q5 H, t6 m6 C        LONG        m_nAsyncFindHandle;
" e: n' a! n8 C+ ?; T        bool        m_bCOM;
4 r# C! H# ^8 P1 r4 U        bool        m_bPortIsFree;
        CString m_sLocalIP;
9 T1 i( }6 c) g( R4 }        CString m_sExternalIP;
, l7 \. A/ H4 y5 r  R        bool        m_bADSL;                // Is the device ADSL?
        bool        m_ADSLFailed;        // Did port mapping failed for the ADSL device?
        bool        m_bInited;
! H5 X  N5 A: W        bool        m_bAsyncFindRunning;
# t- ^( o: E5 ?        HMODULE m_hADVAPI32_DLL;
        HMODULE        m_hIPHLPAPI_DLL;
  k! E. @) z4 i        bool        m_bSecondTry;
        bool        m_bServiceStartedByEmule;
        bool        m_bDisableWANIPSetup;
' E! ]6 y; x4 x4 I9 D7 B        bool        m_bDisableWANPPPSetup;


( x* V" x  t' ?: k8 _};

; Q: _: D( r6 k2 F
; R& {$ f( H0 e' x# j- |// DeviceFinder Callback
4 d8 p' N# b/ \+ E& {class CDeviceFinderCallback
, m2 i% r. W$ X9 N        : public IUPnPDeviceFinderCallback
{
& a% A! Y" p* [% S% P' E# J0 C$ o* z2 Wpublic:
/ h8 w( Z/ e% m+ j        CDeviceFinderCallback(CUPnPImplWinServ& instance)
2 U+ v% |0 a* r                : m_instance( instance )
3 |8 {" |6 B' H' P        { m_lRefCount = 0; }


   STDMETHODIMP QueryInterface(REFIID iid, LPVOID* ppvObject);
   STDMETHODIMP_(ULONG) AddRef();
6 [1 D+ {* F) y. A9 V$ ?/ u   STDMETHODIMP_(ULONG) Release();


+ S; o% I9 `, p// implementation
! P( ?+ h5 e7 l0 x$ lprivate:
        HRESULT __stdcall DeviceAdded(LONG nFindData, IUPnPDevice* pDevice);
$ N8 c5 l* h& l- o6 e* k  L6 e        HRESULT __stdcall DeviceRemoved(LONG nFindData, BSTR bsUDN);
        HRESULT __stdcall SearchComplete(LONG nFindData);
; A' v. j+ _, B

private:
3 }+ k" M0 Y! f7 H. j' t7 D        CUPnPImplWinServ& m_instance;
! k5 M% T3 o& E: D/ M        LONG m_lRefCount;
};
& @; x# E0 S; Y. Q6 S: a  ]0 M

( N) B6 x* `) V8 K// Service Callback
class CServiceCallback
        : public IUPnPServiceCallback
{
/ v2 b9 E) i/ j& l  Epublic:
        CServiceCallback(CUPnPImplWinServ& instance)
                : m_instance( instance )
        { m_lRefCount = 0; }
4 {- ]( e3 w8 p: q6 R* Z% X4 L   
   STDMETHODIMP QueryInterface(REFIID iid, LPVOID* ppvObject);
   STDMETHODIMP_(ULONG) AddRef();
8 D; L" l. F# E/ x" @, h   STDMETHODIMP_(ULONG) Release();

" P! V4 R7 Y7 _# n, V
// implementation
: V3 O! c' G0 S" _+ E2 [private:
. {" |3 D0 w5 s! u1 Q% `6 r        HRESULT __stdcall StateVariableChanged(IUPnPService* pService, LPCWSTR pszStateVarName, VARIANT varValue);
        HRESULT __stdcall ServiceInstanceDied(IUPnPService* pService);


private:
        CUPnPImplWinServ& m_instance;
        LONG m_lRefCount;
};

/ N. [+ A7 U9 _1 P
' B/ T2 n3 ~* v" G% N1 q' Y5 ]/////////////////////////////////////////////////
& |% K) ?( P( n8 k9 ~

使用时只需要使用抽象类的接口。
CUPnPImpl::SetMessageOnResult设置需要接受UPNP端口映射是否成功的窗口句柄和消息ID.
, u) x! I2 n7 n3 U3 VCUPnPImpl::StartDiscovery将开启一个异步设备查找并进行端口映射,其参数为需要映射的内网端口.
CUPnPImpl::StopAsyncFind停止设备查找.
/ a: K0 @4 o! z: TCUPnPImpl::DeletePorts删除端口映射.