UPnP

天云上人

/*uPnP.h*/

1. 
   6 ?+ q/ g6 W2 Z% s" n
2. #ifndef   MYUPNP_H_
   
3. 
   , b, f, M! z6 a4 p& O. E
4. #pragma   once
   " @! ]$ T; i6 o2 k/ _
5. 
   
6. typedef   unsigned   long   ulong;
   2 p, Q  R3 A1 c$ a* c4 r
7. 
   + T" w4 C/ E" Z- ?' b
8. class   MyUPnP
   ( t- |3 W6 J5 }2 |7 }& l
9. {
   6 j: T) c& [2 _9 M% B) e; R
10. public:
    # s7 y* @6 m# X) Z
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
    
15. UNAT_EXTERNAL_PORT_IN_USE, //   Error,   you   are   trying   to   add   a   port   mapping   with   an   external   port   in   use
    
16. UNAT_NOT_IN_LAN //   Error,   you   aren 't   in   a   LAN   ->   no   router   or   firewall
    
17. }   UPNPNAT_RETURN;
    % H+ [; g$ k9 h# s
18. 
    - c7 e6 C* x8 x7 V
19. typedef   enum{
    $ q9 s( |9 s/ w+ x4 _! j* G
20. UNAT_TCP, //   TCP   Protocol
    
21. UNAT_UDP //   UDP   Protocol
    
22. }   UPNPNAT_PROTOCOL;
    
23. 
    
24. typedef   struct{
    + y" R$ N( @0 l0 g/ W
25. WORD   internalPort; //   Port   mapping   internal   port
    ( X# o% _; z  h+ P
26. WORD   externalPort; //   Port   mapping   external   port
    2 F! l" m( H5 b& t
27. UPNPNAT_PROTOCOL   protocol; //   Protocol->   TCP   (UPNPNAT_PROTOCOL:UNAT_TCP)   ||   UDP   (UPNPNAT_PROTOCOL:UNAT_UDP)
    * L1 ?4 G* ]4 H* r3 P" V7 X  |
28. CString   description; //   Port   mapping   description
    . }9 q, `8 \$ x8 L3 e7 s
29. }   UPNPNAT_MAPPING;
    
30. 
    ' B% e! E6 B! e6 g
31. MyUPnP();
    1 v6 {( Q) ~3 o& S$ m$ r
32. ~MyUPnP();
    
33. 
    ) v! Z& P+ N. x/ K$ D% {& V. T
34. UPNPNAT_RETURN   AddNATPortMapping(UPNPNAT_MAPPING   *mapping,   bool   tryRandom   =   false);
    ; _3 e0 A) G+ R3 N$ C
35. UPNPNAT_RETURN   RemoveNATPortMapping(UPNPNAT_MAPPING   mapping,   bool   removeFromList   =   true);
    0 D; q+ W& ^% O9 K5 A% x- p. V
36. void   clearNATPortMapping();
    9 T% O. m+ u  R0 Q1 [$ N, x
37. 
    
38. CString GetLastError();
    
39. CString GetLocalIPStr();
    
40. WORD GetLocalIP();
    
41. bool IsLANIP(WORD   nIP);
    
42. 
    
43. protected:
    7 Y% F$ C8 N$ O6 A7 N
44. void InitLocalIP();
    6 J+ D" J: ^; `! g! i
45. void SetLastError(CString   error);
    ) H8 E: n8 Y3 Q
46. 
    % ]7 m& q: X  w+ j
47. bool   addPortmap(int   eport,   int   iport,   const   CString&   iclient,
    * Q% i. b4 f$ g+ r) q! [* y) D# X
48.       const   CString&   descri,   const   CString&   type);
    
49. bool   deletePortmap(int   eport,   const   CString&   type);
    + w1 x) S1 e5 v8 j" o
50. 
    + z: X5 U) p; E6 l; t- q  O1 ~1 d
51. bool isComplete()   const   {   return   !m_controlurl.IsEmpty();   }
    4 C; v2 t6 D$ y8 S: E2 h
52. 
    3 P9 n" J3 I% i! D
53. bool Search(int   version=1);
    
54. bool GetDescription();
    
55. CString GetProperty(const   CString&   name,   CString&   response);
    
56. bool InvokeCommand(const   CString&   name,   const   CString&   args);
    
57. 
    ' e1 q. c$ M& F9 Q* @5 h8 F% i  ~& U: V3 _
58. bool Valid()const{return   (!m_name.IsEmpty()&&!m_description.IsEmpty());}
    
59. bool InternalSearch(int   version);
    
60. CString m_devicename;
    8 N+ v9 Q5 ]: C% U) q, E
61. CString m_name;
    
62. CString m_description;
    : \: y! X: ^+ U* O; G1 c0 \
63. CString m_baseurl;
    ( S# f3 W, v% I# A' D2 f
64. CString m_controlurl;
    + V8 q+ S7 z2 i! A& @0 p
65. CString m_friendlyname;
    
66. CString m_modelname;
    
67. int m_version;
    
68. 
      S2 c, h- e+ [; |; R# h# p3 p
69. private:
    
70. CList <UPNPNAT_MAPPING,   UPNPNAT_MAPPING>   m_Mappings;
    , y+ A& h& ?/ o8 \7 s6 X
71. 
      D$ \% r! a- V9 b
72. CString m_slocalIP;
    " _! N5 N$ N0 u0 g8 M& I1 l( K5 S1 q
73. CString m_slastError;
    5 n& H! G" s# f4 d  t: q2 I
74. WORD m_uLocalIP;
    7 {4 A* b3 w- N+ t! S
75. 
    
76. bool isSearched;
    
77. };
    : \5 p9 n+ n( Z/ M% G" R
78. #endif

复制代码

天云上人

/*UPnP.cpp*/

1. 
   
2. #include   "stdafx.h "
   7 ?0 \+ J) ]7 @# P
3. 
   
4. #include   "upnp.h "
   
5. 
   
6. #define   UPNPPORTMAP0       _T( "WANIPConnection ")
   
7. #define   UPNPPORTMAP1       _T( "WANPPPConnection ")
   
8. #define   UPNPGETEXTERNALIP   _T( "GetExternalIPAddress "),_T( "NewExternalIPAddress ")
   
9. #define   UPNPADDPORTMAP   _T( "AddPortMapping ")
     [4 t! D- G- q# i9 D. r: K
10. #define   UPNPDELPORTMAP   _T( "DeletePortMapping ")
    
11. 
    
12. static   const   ulong UPNPADDR   =   0xFAFFFFEF;
    
13. static   const   int UPNPPORT   =   1900;
    0 I# ~  O3 l. m% K5 U' I
14. static   const   CString URNPREFIX   =   _T( "urn:schemas-upnp-org: ");
    + @* q5 B6 g" Y  t/ H) S
15. 
    
16. const   CString   getString(int   i)
    
17. {
    
18. CString   s;
    0 F; z; V8 x" e; v! t1 h1 V
19. 
    5 V0 Z" }0 I+ C$ _) d4 O2 f; s5 c3 s
20. s.Format(_T( "%d "),   i);
    % s3 d2 v. ]% X+ |% y9 A
21. 
    # q  \0 M9 x8 ]0 G
22. return   s;
    3 W9 y4 _7 V* \* @3 a( X
23. }
    
24. 
    
25. const   CString   GetArgString(const   CString&   name,   const   CString&   value)
    
26. {
    
27. return   _T( " < ")   +   name   +   _T( "> ")   +   value   +   _T( " </ ")   +   name   +   _T( "> ");
    
28. }
    ! E% k- ]) j: Q
29. 
    $ z+ r/ U+ }- u. n& D0 g
30. const   CString   GetArgString(const   CString&   name,   int   value)
    ; ]4 b0 A* E# }/ @4 g6 E
31. {
    
32. return   _T( " < ")   +   name   +   _T( "> ")   +   getString(value)   +   _T( " </ ")   +   name   +   _T( "> ");
    
33. }
    ; o6 [  u2 O1 ~: {/ o+ I
34. 
    
35. bool   SOAP_action(CString   addr,   uint16   port,   const   CString   request,   CString   &response)
    
36. {
    
37. char   buffer[10240];
    
38. 
    , v* ^/ s3 l( Q4 J0 J
39. const   CStringA   sa(request);
    % g( N4 o( U0 ]' J; w. Z
40. int   length   =   sa.GetLength();
    * P/ a2 w$ q/ S! d, K8 A
41. strcpy(buffer,   (const   char*)sa);
    8 \3 W6 u. I" R$ M5 s$ ?# V
42. 
    
43. uint32   ip   =   inet_addr(CStringA(addr));
    
44. struct   sockaddr_in   sockaddr;
      q, i3 m' q+ l) f" I; i, C$ y9 p- A
45. memset(&sockaddr,   0,   sizeof(sockaddr));
    
46. sockaddr.sin_family   =   AF_INET;
    
47. sockaddr.sin_port   =   htons(port);
    . ^6 S; d6 O; q0 h  j
48. sockaddr.sin_addr.S_un.S_addr   =   ip;
    4 o% b6 M- G4 H
49. int   s   =   socket(AF_INET,   SOCK_STREAM,   0);
    % q5 H+ Z+ p  h2 k3 L2 }
50. u_long   lv   =   1;
    
51. ioctlsocket(s,   FIONBIO,   &lv);
    " Q- ]9 u% t% N/ Y0 j! x  M
52. connect(s,   (struct   sockaddr   *)&sockaddr,   sizeof(sockaddr));
    
53. Sleep(20);
    
54. int   n   =   send(s,   buffer,   length,   0);
    
55. Sleep(100);
    % h' u5 V* v2 j/ N" S5 d: i: x7 h
56. int   rlen   =   recv(s,   buffer,   sizeof(buffer),   0);
    ; l; G& n$ {; V& k& I
57. closesocket(s);
    
58. if   (rlen   ==   SOCKET_ERROR)   return   false;
    
59. if   (!rlen)   return   false;
    . o6 j8 y; D5 [/ h5 @6 r2 E
60. 
    ' }# u, ~" R8 ^  g
61. response   =   CString(CStringA(buffer,   rlen));
    " C0 S/ |1 L$ ^- f
62. 
    
63. return   true;
    
64. }
    
65. 
    7 m8 d% `( Q9 P" K
66. int   SSDP_sendRequest(int   s,   uint32   ip,   uint16   port,   const   CString&   request)
    
67. {
    & B5 f0 e8 a1 w, L
68. char   buffer[10240];
    6 ~: c$ y5 W& v. f
69. 
    
70. const   CStringA   sa(request);
    
71. int   length   =   sa.GetLength();
    & N  N  H5 v1 N2 h$ p
72. strcpy(buffer,   (const   char*)sa);
    3 B! B" H3 z6 R# h1 U
73. 
    ' H8 _" C# Z0 L8 W
74. struct   sockaddr_in   sockaddr;
    , K, G/ e$ v  Y* i% z* ^  [: ^
75. memset(&sockaddr,   0,   sizeof(sockaddr));
    
76. sockaddr.sin_family   =   AF_INET;
    + c. o; U5 I1 f& a$ I6 L8 L4 u7 _
77. sockaddr.sin_port   =   htons(port);
    8 `1 \5 B# l% }# t. N
78. sockaddr.sin_addr.S_un.S_addr   =   ip;
    . Z8 l# |* w$ S( p" C$ V
79. 
    
80. return   sendto(s,   buffer,   length,   0,   (struct   sockaddr   *)&sockaddr,   sizeof(sockaddr));
    4 y0 ~( W9 ]6 B4 M% z: T
81. }
    ' r  ?2 y0 c, T& D/ q
82. 
    
83. bool   parseHTTPResponse(const   CString&   response,   CString&   result)
    # r' R+ _, j) ?. F
84. {
    7 `6 v4 {) m& @
85. int   pos   =   0;
    
86. 
    
87. CString   status   =   response.Tokenize(_T( "\r\n "),   pos);
    / d0 y% t7 G# O0 ~
88. 
    
89. result   =   response;
    3 M" [! s7 p0 |8 V
90. result.Delete(0,   pos);
    
91. 
    2 D" j- U& s7 _. w6 S+ g
92. pos   =   0;
    
93. status.Tokenize(_T( "   "),   pos);
    
94. status   =   status.Tokenize(_T( "   "),   pos);
    
95. if   (status.IsEmpty()   ||   status[0]!= '2 ')   return   false;
    
96. return   true;
    
97. }
    + I3 N* s! \( i
98. 
    % q9 ~2 P+ @2 y  o9 ~7 i5 i
99. const   CString   getProperty(const   CString&   all,   const   CString&   name)
    
100. {
     ( }! j% n- p5 O- w" I
101. CString   startTag   =   ' < '   +   name   +   '> ';
     9 |/ x' k# ~9 A2 j, F& R5 y: W
102. CString   endTag   =   _T( " </ ")   +   name   +   '> ';
     
103. CString   property;
     " X0 r: y4 i8 a3 x* a1 f
104. 
     
105. int   posStart   =   all.Find(startTag);
     ! z& S) F$ V; A. h3 Y! Z5 D9 ]' h
106. if   (posStart <0)   return   CString();
     
107. 
     / W1 ?" M2 B- A! w. V* r
108. int   posEnd   =   all.Find(endTag,   posStart);
     4 L+ F6 T/ q' u. ~
109. if   (posStart> =posEnd)   return   CString();
     / o. R8 `1 `" t: {6 w5 ^5 O
110. 
     
111. return   all.Mid(posStart   +   startTag.GetLength(),   posEnd   -   posStart   -   startTag.GetLength());
     
112. }
     
113. 
     
114. MyUPnP::MyUPnP()
     
115. :   m_version(1)
     
116. {
     
117. m_uLocalIP   =   0;
     % N; [5 _* T! b, r
118. isSearched   =   false;
     ( q: h0 Y  g0 M" m' j/ M! o* G
119. }
     + \  z1 s) y$ n: Z0 `3 `, ?- M6 h5 y' Q
120. 
     
121. MyUPnP::~MyUPnP()
     , J! x" }4 I' i5 W7 s0 Q8 h: F
122. {
     ) v+ z2 y. \4 V" g- z
123. UPNPNAT_MAPPING   search;
     ; ?; \+ L9 z$ R! \- L
124. POSITION   pos   =   m_Mappings.GetHeadPosition();
     
125. while(pos){
     2 h, E6 X) ~' F, f6 j
126. search   =   m_Mappings.GetNext(pos);
     
127. RemoveNATPortMapping(search,   false);
     " v2 G0 e0 t# p/ h, D
128. }
     
129. 
     
130. m_Mappings.RemoveAll();
     4 z3 G; J1 m6 J4 A
131. }
     0 {5 s" U9 J6 ]4 d  O: e, H5 ]
132. 
     
133. 
     
134. bool   MyUPnP::InternalSearch(int   version)
     ' H/ T2 ~+ i) {1 c1 ?8 l7 f/ [: R3 b
135. {
     2 P5 y; K7 P' z6 B4 Q& R4 W
136. if(version <=0)version   =   1;
     , T  ~, W- s* @- i, f! N1 f
137. m_version   =   version;
     % k0 d% B( J' {. u4 W- n9 a; M' y
138. 
     ! V  F- A0 |5 Q0 L- H3 R% D1 N
139. #define   NUMBEROFDEVICES 2
     
140. CString   devices[][2]   =   {
     
141. {UPNPPORTMAP1,   _T( "service ")},
     
142. {UPNPPORTMAP0,   _T( "service ")},
     / M$ K% @- x0 o$ w4 |$ W
143. {_T( "InternetGatewayDevice "),   _T( "device ")},
     
144. };
     
145. 
     ( A1 [4 {8 w) I( b3 D9 ]* M
146. int   s   =   socket(AF_INET,   SOCK_DGRAM,   0);
     1 g- Z% R; d' L8 E" Q
147. u_long   lv   =   1;
     1 O  j' K" f" |% d
148. ioctlsocket(s,   FIONBIO,   &lv);
     
149. 
     
150. int   rlen   =   0;
     
151. for   (int   i=0;   rlen <=0   &&   i <500;   i++)   {
     ( T2 a: X9 a3 x  B- S
152. if   (!(i%100))   {
     ; i" L# k& B$ `7 l" r/ B
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);
     2 {- j5 z' X0 L# t' a0 W
155. CString   request;
     ; V$ ~% E  G1 Z( M, ^1 p) ?# S- n( i
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);
     / I6 n7 P' i8 V: G
159. }
     ( F/ s2 X+ ?6 n" Z6 |9 x& g
160. }
     
161. 
     - ]% Q& j$ H  J2 A2 B& S
162. Sleep(10);
     , Y. |  d3 ?  f8 |
163. 
     
164. char   buffer[10240];
     ) J: Z! I) f2 R- |: L2 v% K; r' ~
165. rlen   =   recv(s,   buffer,   sizeof(buffer),   0);
     ( K5 X+ v! b! R) h/ }! M5 d, z5 r
166. if   (rlen   <=   0)   continue;
       x, k; R: ?% c5 m# w
167. closesocket(s);
     6 e2 E8 o5 f3 X
168. 
     
169. CString   response   =   CString(CStringA(buffer,   rlen));
     * @9 h9 ^& u0 f) a
170. CString   result;
     . ?- @0 T$ Z, j  j2 `5 M
171. if   (!parseHTTPResponse(response,   result))   return   false;
     3 B2 l" d$ R- H  k- s% _$ f  s% q
172. 
     ; z7 F$ Q# ]  }
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);
     0 F* X9 F7 f: R* T" J+ q
175. if   (result.Find(m_name)   > =   0)   {
     * ]1 K' h, g$ g) D9 B+ Y1 M' f
176. for   (int   pos   =   0;;)   {
     % K% L- L: T/ i$ [: t) E/ m2 |& h
177. CString   line   =   result.Tokenize(_T( "\r\n "),   pos);
     
178. if   (line.IsEmpty())   return   false;
     * T* Z+ O7 k# J$ w
179. CString   name   =   line.Mid(0,   9);
     
180. name.MakeUpper();
     8 q0 e' A( K4 i3 `& h
181. if   (name   ==   _T( "LOCATION: "))   {
     ( `6 Y' ]$ w2 D2 O4 R- D4 x
182. line.Delete(0,   9);
     . s1 h0 G6 n( ~( |( T
183. m_description   =   line;
     + R# c% h, M& |% Y2 J
184. m_description.Trim();
       `" y8 r% e' `  v5 Y
185. return   GetDescription();
       ]+ b+ v- I" D- L# I
186. }
     , N' b; E* N7 Z& X/ q
187. }
     ; t& o2 M4 A# ^" d, T7 D8 b! Q
188. }
     
189. }
     2 w+ Z% L# s8 g; y
190. }
     , e2 D+ u" O3 w3 }6 |. Q" z
191. closesocket(s);
     / @  F# o; T# m+ |( ^! l: Y) a  N3 B# \
192. 
     $ z. `5 v: Z( G. d4 t
193. return   false;
     - I5 h6 E# h/ i- p( N" Z
194. }
     ) v# w, F* g% Y* c

复制代码

天云上人

以下有关upnp的接口来自emule，
. W( w9 G9 L* \' z2 |
5 N  X$ g5 _* |3 R" M! u9 N9 t, G
///////////////////////////////////////////
: q3 K# T2 ^8 i# m5 }/ M# C//upnp的抽象接口类,只需调用这些接口即可实现端口映射功能.
7 j2 B$ I' j- [% f/ K) B
# z5 L' r1 V% n/ q% {
#pragma once
#include <exception>
7 a2 B: F' O; u# B* Z/ J
* t. @% c# _$ V* i- I" G# D1 q
$ n  _5 M" S; E* K! V  enum TRISTATE{
        TRIS_FALSE,
        TRIS_UNKNOWN,
        TRIS_TRUE
! S$ g- w9 [# a1 Q& O};

5 T, o# Q0 {  s( C; z
; M+ v4 ]* V, U4 _' P" Ienum UPNP_IMPLEMENTATION{
        UPNP_IMPL_WINDOWSERVICE = 0,
+ |6 z/ |- H' Y  M        UPNP_IMPL_MINIUPNPLIB,
$ t! q) R) C" y; b2 ~$ l8 y        UPNP_IMPL_NONE /*last*/
};

9 c6 c! m& y$ p) Y; X
$ w# o1 J9 b  k  S7 q' t$ Y

class CUPnPImpl
2 n! Z2 }# R' }& f4 L7 [$ ^{
public:
" E) A' C+ o1 `( w( }* V* a( j7 Y        CUPnPImpl();
        virtual ~CUPnPImpl();
        struct UPnPError : std::exception {};
. L/ y' Y" W( X+ `2 k% B        enum {
2 }  N( K- U( E! d4 O, y                UPNP_OK,
                UPNP_FAILED,
2 w& P# j/ s) C5 ]4 ]                UPNP_TIMEOUT
        };
. J: I9 w  ?  T& ~! H' J% [
7 e% u1 K% R# k9 q' d+ c# A
1 q  |/ S; }: e        virtual void        StartDiscovery(uint16 nTCPPort, uint16 nUDPPort, uint16 nTCPWebPort) = 0;
        virtual bool        CheckAndRefresh() = 0;
8 L/ g. p) M; C4 ?2 l# G7 k2 b        virtual void        StopAsyncFind() = 0;
5 \: _; D, v+ v, n3 Q9 h0 u        virtual void        DeletePorts() = 0;
        virtual bool        IsReady() = 0;
        virtual int                GetImplementationID() = 0;
8 T1 H" v5 j% Y6 E       
        void                        LateEnableWebServerPort(uint16 nPort); // Add Webserverport on already installed portmapping

9 B, ?& P- D/ l2 P2 {' Q
        void                        SetMessageOnResult(HWND hWindow, UINT nMessageID);
        TRISTATE                ArePortsForwarded() const                                                                { return m_bUPnPPortsForwarded; }
        uint16                        GetUsedTCPPort()                                                                                { return m_nTCPPort; }
        uint16                        GetUsedUDPPort()                                                                                { return m_nUDPPort; }       
0 s+ n( j& n  N7 `8 l
" [5 M! M; K" ]& m
. Q) Y7 n' D  r* `* ?/ @% p// Implementation
6 c3 l6 d3 B0 w3 Q( i) b' Xprotected:
# ]/ T' f" ~) C# F5 [) ^$ S        volatile TRISTATE        m_bUPnPPortsForwarded;
, H2 N, P2 S# Y3 \        void                                SendResultMessage();
        uint16                                m_nUDPPort;
" ?6 y1 c8 A4 V6 {& I        uint16                                m_nTCPPort;
        uint16                                m_nTCPWebPort;
* U9 m% s% O% z" B        bool                                m_bCheckAndRefresh;

6 S7 a  \% m6 d
( F7 l2 {' E% V3 Gprivate:
2 O. e8 O/ R0 \2 A' ~6 U        HWND        m_hResultMessageWindow;
  ~6 }$ D4 e+ V" {1 Z3 }        UINT        m_nResultMessageID;
7 k& S" }+ v2 j. L8 g5 D) Q% E$ c0 s
- v6 r9 ^1 ~" F# F1 z+ q+ ]
};


// Dummy Implementation to be used when no other implementation is available
class CUPnPImplNone: public CUPnPImpl
{
public:
        virtual void        StartDiscovery(uint16, uint16, uint16)                                        { ASSERT( false ); }
2 @! S+ t' g' n( p* }+ p, f        virtual bool        CheckAndRefresh()                                                                                { return false; }
5 W/ u2 f# i$ A+ d        virtual void        StopAsyncFind()                                                                                        { }
        virtual void        DeletePorts()                                                                                        { }
/ |* `: O0 U7 o2 q* ^        virtual bool        IsReady()                                                                                                { return false; }
: F! B3 s/ q; M/ G0 f# B        virtual int                GetImplementationID()                                                                        { return UPNP_IMPL_NONE; }
};


/////////////////////////////////////
6 H% s; R5 I( X8 u3 p3 q% J//下面是使用windows操作系统自带的UPNP功能的子类
. y* n, X& a) G+ N* {1 p9 I
1 n8 z+ O1 ?- y- Z, F4 e
+ A+ i! F; U' q% f' V3 ^#pragma once
#pragma warning( disable: 4355 )

3 j2 F1 S8 M5 T; M9 z. N8 {
, ]( c9 d9 y. P; I1 S# S0 j#include "UPnPImpl.h"
. b; k  ^+ H5 A: s( m#include <upnp.h>
7 j8 c* v7 k' k5 R; R2 v#include <iphlpapi.h>
#include <comdef.h>
#include <winsvc.h>
6 T3 ^4 b) `1 ?0 F3 f

#include <vector>
#include <exception>
#include <functional>
5 I2 {* z) r% C5 \0 D. Z



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;
* S6 M3 G4 q9 q' u+ dtypedef        _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;
- [' y. `* [  H: N* M: x2 K' E
  P. h% Z" p8 ?$ j8 j: D
typedef DWORD (WINAPI* TGetBestInterface) (
$ E# q2 p+ R. S2 s$ g  IPAddr dwDestAddr,
- l/ ?# z8 Z! x  k6 u. F) Y5 d  PDWORD pdwBestIfIndex
6 P2 {: i. A5 Q);


* b. E% D/ Q5 H: {* X4 C, N) dtypedef DWORD (WINAPI* TGetIpAddrTable) (
  PMIB_IPADDRTABLE pIpAddrTable,
  PULONG pdwSize,
  BOOL bOrder
);


typedef DWORD (WINAPI* TGetIfEntry) (
  PMIB_IFROW pIfRow
& e$ ?8 W" V) F9 ~3 \);


CString translateUPnPResult(HRESULT hr);
HRESULT UPnPMessage(HRESULT hr);
7 x  M2 r+ k8 \/ d; A
/ S4 S- t: x  w* L, w+ G; r
class CUPnPImplWinServ: public CUPnPImpl
{
        friend class CDeviceFinderCallback;
        friend class CServiceCallback;
// Construction
' M( G7 h# U& Z( E2 R+ wpublic:
        virtual ~CUPnPImplWinServ();
        CUPnPImplWinServ();


        virtual void        StartDiscovery(uint16 nTCPPort, uint16 nUDPPort, uint16 nTCPWebPort)                { StartDiscovery(nTCPPort, nUDPPort, nTCPWebPort, false); }
        virtual void        StopAsyncFind();
        virtual void        DeletePorts();
        virtual bool        IsReady();
        virtual int                GetImplementationID()                                                                        { return UPNP_IMPL_WINDOWSERVICE; }
" f4 o. N' V; m$ B
7 o. y. D) A& b1 t- Z+ u
        // No Support for Refreshing on this  (fallback) implementation yet - in many cases where it would be needed (router reset etc)
/ ^! [+ X# E. n) r7 J4 Z        // the windows side of the implementation tends to get bugged untill reboot anyway. Still might get added later
        virtual bool        CheckAndRefresh()                                                                                { return false; };

* \( }1 c9 A1 `4 w
protected:
% r+ O% f$ g; Y$ p* Y4 q        void        StartDiscovery(uint16 nTCPPort, uint16 nUDPPort, uint16 nTCPWebPort, bool bSecondTry);
        void        AddDevice(DevicePointer pDevice, bool bAddChilds, int nLevel = 0);
/ P: V* @0 n* y7 s% S& f3 x. S        void        RemoveDevice(CComBSTR bsUDN);
% |" D4 f2 G- I* A, |$ Q; z, \        bool        OnSearchComplete();
# \. W' t7 p" O5 A& M        void        Init();


6 m+ Q6 N- C2 P        inline bool IsAsyncFindRunning()
/ r; [8 ?9 k2 m        {
                if ( m_pDeviceFinder != NULL && m_bAsyncFindRunning && GetTickCount() - m_tLastEvent > 10000 )
                {
                        m_pDeviceFinder->CancelAsyncFind( m_nAsyncFindHandle );
                        m_bAsyncFindRunning = false;
) j1 R) T% }) n( y                }
$ G( ?, a1 u- X3 F0 g# E                MSG msg;
( c% w/ I2 N9 z; ?8 u3 n                while ( PeekMessage( &msg, NULL, 0, 0, PM_REMOVE ) )
                {
" B1 j' j' |7 \) o                        TranslateMessage( &msg );
                        DispatchMessage( &msg );
                }
                return m_bAsyncFindRunning;
) t  l/ U) ~1 @) {        }


7 ~3 }5 \5 R# R) t% Q5 K        TRISTATE                        m_bUPnPDeviceConnected;

. r- t# B) S3 c" f# z, u
* O) H  w# r2 D1 F. r// Implementation
        // API functions
! T; c8 {/ h! C. r: t        SC_HANDLE (WINAPI *m_pfnOpenSCManager)(LPCTSTR, LPCTSTR, DWORD);
8 X# b- M; w6 n: ?) s5 I* ?8 [        SC_HANDLE (WINAPI *m_pfnOpenService)(SC_HANDLE, LPCTSTR, DWORD);
9 z, |4 K( g0 g        BOOL (WINAPI *m_pfnQueryServiceStatusEx)(SC_HANDLE, SC_STATUS_TYPE, LPBYTE, DWORD, LPDWORD);
% K3 n) b1 L: q        BOOL (WINAPI *m_pfnCloseServiceHandle)(SC_HANDLE);
        BOOL (WINAPI *m_pfnStartService)(SC_HANDLE, DWORD, LPCTSTR*);
        BOOL (WINAPI *m_pfnControlService)(SC_HANDLE, DWORD, LPSERVICE_STATUS);
$ U# }+ a' e& N

        TGetBestInterface                m_pfGetBestInterface;
  e  k; b7 i2 A! b1 V: I        TGetIpAddrTable                        m_pfGetIpAddrTable;
        TGetIfEntry                                m_pfGetIfEntry;


        static FinderPointer CreateFinderInstance();
        struct FindDevice : std::unary_function< DevicePointer, bool >
9 K1 ]8 t% U) k        {
) x$ _8 U. V9 a$ ?: B                FindDevice(const CComBSTR& udn) : m_udn( udn ) {}
                result_type operator()(argument_type device) const
2 R& Q1 r4 m2 B+ o                {
2 m, i3 `8 a2 D( w                        CComBSTR deviceName;
                        HRESULT hr = device->get_UniqueDeviceName( &deviceName );

* j: p0 t2 Y) `- z6 u& i8 s
                        if ( FAILED( hr ) )
. J1 I8 X. q" |4 |                                return UPnPMessage( hr ), false;
5 ^. h" y( T9 [2 Q! M

8 [% A7 [; t" O% {, ~                        return wcscmp( deviceName.m_str, m_udn ) == 0;
                }
                CComBSTR m_udn;
6 h. R; R0 r/ u9 r1 B0 t! o5 T        };
       
        void        ProcessAsyncFind(CComBSTR bsSearchType);
( u# V3 ]9 b4 i" o$ Z' s) ]# y        HRESULT        GetDeviceServices(DevicePointer pDevice);
        void        StartPortMapping();
, t# T5 p$ V  i6 P        HRESULT        MapPort(const ServicePointer& service);
4 z% \  Z; l% h; l2 V- X        void        DeleteExistingPortMappings(ServicePointer pService);
        void        CreatePortMappings(ServicePointer pService);
        HRESULT SaveServices(EnumUnknownPtr pEU, const LONG nTotalItems);
' x! H" m! Q* H. z. f- u+ M        HRESULT InvokeAction(ServicePointer pService, CComBSTR action,
! g5 K  e' ~5 s* \                LPCTSTR pszInArgString, CString& strResult);
3 |4 ~0 V$ {" z# k8 i6 b. B! J9 C        void        StopUPnPService();


% G# r- }7 q8 L7 B! J6 j        // Utility functions
' ^) F6 k) I0 `1 y        HRESULT CreateSafeArray(const VARTYPE vt, const ULONG nArgs, SAFEARRAY** ppsa);
        INT_PTR CreateVarFromString(const CString& strArgs, VARIANT*** pppVars);
2 B( _7 ?: Q/ k: L, e        INT_PTR        GetStringFromOutArgs(const VARIANT* pvaOutArgs, CString& strArgs);
+ o7 n/ A$ S) G1 C# U/ A! F        void        DestroyVars(const INT_PTR nCount, VARIANT*** pppVars);
        HRESULT GetSafeArrayBounds(SAFEARRAY* psa, LONG* pLBound, LONG* pUBound);
        HRESULT GetVariantElement(SAFEARRAY* psa, LONG pos, VARIANT* pvar);
; [; O9 o/ w' ]6 U* Y        CString        GetLocalRoutableIP(ServicePointer pService);


6 a$ A/ {+ x& x: e3 k2 G- ^" n// Private members
6 h) {: r" e7 m, t+ p1 ?$ xprivate:
        DWORD        m_tLastEvent;        // When the last event was received?
        std::vector< DevicePointer >  m_pDevices;
9 Z- h5 S8 X& P        std::vector< ServicePointer > m_pServices;
$ M% g7 J+ c; H; \        FinderPointer                        m_pDeviceFinder;
        DeviceFinderCallback        m_pDeviceFinderCallback;
( [0 E" N* a6 N; T6 A  ~  k        ServiceCallback                        m_pServiceCallback;


        LONG        m_nAsyncFindHandle;
7 J6 y: L& a: p3 h        bool        m_bCOM;
        bool        m_bPortIsFree;
        CString m_sLocalIP;
9 j! n9 U7 Y+ G9 i; Y0 d        CString m_sExternalIP;
        bool        m_bADSL;                // Is the device ADSL?
  V: w9 X" D+ z) N, r3 i        bool        m_ADSLFailed;        // Did port mapping failed for the ADSL device?
  L* ~& ^4 U1 w3 @5 ]: C        bool        m_bInited;
        bool        m_bAsyncFindRunning;
        HMODULE m_hADVAPI32_DLL;
- \0 I. I( G: e        HMODULE        m_hIPHLPAPI_DLL;
        bool        m_bSecondTry;
        bool        m_bServiceStartedByEmule;
        bool        m_bDisableWANIPSetup;
        bool        m_bDisableWANPPPSetup;


3 @: V& ?2 b0 M0 _% l- Y};
# K! T7 ]2 u' s8 B/ b

0 M4 b* [% u1 D5 k& p* e; v// DeviceFinder Callback
class CDeviceFinderCallback
5 T! g* Y5 `9 ^  u        : public IUPnPDeviceFinderCallback
{
public:
        CDeviceFinderCallback(CUPnPImplWinServ& instance)
! _: u1 Z& d: K1 u" p                : m_instance( instance )
        { m_lRefCount = 0; }


   STDMETHODIMP QueryInterface(REFIID iid, LPVOID* ppvObject);
   STDMETHODIMP_(ULONG) AddRef();
7 Q0 O, P& a3 S  r2 K7 W   STDMETHODIMP_(ULONG) Release();
; P, b7 ]5 {% L1 y6 Z0 Y9 w

// implementation
8 w8 Y( ~0 E2 k! r8 J1 X/ N6 V5 b8 T+ eprivate:
& C7 s& J2 y% l4 M8 z9 e        HRESULT __stdcall DeviceAdded(LONG nFindData, IUPnPDevice* pDevice);
        HRESULT __stdcall DeviceRemoved(LONG nFindData, BSTR bsUDN);
        HRESULT __stdcall SearchComplete(LONG nFindData);
! ?$ T. [, H" T7 Q: G- ]. W
# `' s$ E; G/ Y# ~
private:
        CUPnPImplWinServ& m_instance;
        LONG m_lRefCount;
$ J( _; E- A& W- n6 \2 `};


// Service Callback
: D) W8 P) M" t. Cclass CServiceCallback
* D: V! S- I4 y) E" R# D, \  }! t" A        : public IUPnPServiceCallback
{
4 c: E+ q! `5 Npublic:
        CServiceCallback(CUPnPImplWinServ& instance)
                : m_instance( instance )
* p6 x; j4 R. s# ^' ?        { m_lRefCount = 0; }
" R3 Q9 p) q. S: S) c6 v   
3 Y5 D. c& `3 g& X5 D; W   STDMETHODIMP QueryInterface(REFIID iid, LPVOID* ppvObject);
   STDMETHODIMP_(ULONG) AddRef();
   STDMETHODIMP_(ULONG) Release();


// implementation
: ~! \8 _" [7 I. x2 Gprivate:
        HRESULT __stdcall StateVariableChanged(IUPnPService* pService, LPCWSTR pszStateVarName, VARIANT varValue);
        HRESULT __stdcall ServiceInstanceDied(IUPnPService* pService);

6 N1 K  \: _' l. O
. m! m1 X9 G/ J# mprivate:
" y9 p( l3 O0 {& W% y9 \        CUPnPImplWinServ& m_instance;
        LONG m_lRefCount;
};

; N& |  R* l; }" F0 E
: c' d3 H  V+ Y: Y/ }2 g* Y) k3 T/////////////////////////////////////////////////


使用时只需要使用抽象类的接口。
% Z, M6 N8 a( x: [' I: DCUPnPImpl::SetMessageOnResult设置需要接受UPNP端口映射是否成功的窗口句柄和消息ID.
$ g* v' V& e" }& Q5 _# H/ kCUPnPImpl::StartDiscovery将开启一个异步设备查找并进行端口映射,其参数为需要映射的内网端口.
1 @- i" [7 B' E& h1 cCUPnPImpl::StopAsyncFind停止设备查找.
6 ~0 a7 B, L! M% c9 Q" b9 H3 XCUPnPImpl::DeletePorts删除端口映射.