UPnP

天云上人

/*uPnP.h*/

1. 
   
2. #ifndef   MYUPNP_H_
   - a# A! V9 g1 `1 h, ?" a
3. 
   + Z7 }' W0 T  m, f* q6 z) P( a: o
4. #pragma   once
   
5. 
   
6. typedef   unsigned   long   ulong;
   & x1 M5 r* Z" ]& m/ y. z
7. 
   
8. class   MyUPnP
   
9. {
   + G$ g. y+ i$ X, }+ R) l
10. public:
    
11. typedef   enum{
    8 S3 Z5 }5 }& @; `! m: ^6 _! ?
12. UNAT_OK, //   Successfull
    6 }7 j5 @" F2 X  ~1 h
13. UNAT_ERROR, //   Error,   use   GetLastError()   to   get   an   error   description
    , r* ]' Y1 b8 m! ~9 S  K
14. UNAT_NOT_OWNED_PORTMAPPING, //   Error,   you   are   trying   to   remove   a   port   mapping   not   owned   by   this   class
    , @4 J" e+ O* R; \/ X9 }8 T( }4 j
15. UNAT_EXTERNAL_PORT_IN_USE, //   Error,   you   are   trying   to   add   a   port   mapping   with   an   external   port   in   use
    " ^- \* B3 w9 @! q" E, `# D" p- ~
16. UNAT_NOT_IN_LAN //   Error,   you   aren 't   in   a   LAN   ->   no   router   or   firewall
    
17. }   UPNPNAT_RETURN;
    2 x4 _: ?$ u% G
18. 
    + M7 |$ I' R$ e' x( g) f
19. typedef   enum{
    
20. UNAT_TCP, //   TCP   Protocol
    
21. UNAT_UDP //   UDP   Protocol
    " u6 }, n1 {, P2 D& B
22. }   UPNPNAT_PROTOCOL;
    
23. 
    
24. typedef   struct{
    + O( a) v( F2 X: D& ~% s# G
25. WORD   internalPort; //   Port   mapping   internal   port
    
26. WORD   externalPort; //   Port   mapping   external   port
    # ~1 b0 w& p( X1 m. k' P/ U
27. UPNPNAT_PROTOCOL   protocol; //   Protocol->   TCP   (UPNPNAT_PROTOCOL:UNAT_TCP)   ||   UDP   (UPNPNAT_PROTOCOL:UNAT_UDP)
    * R1 x  A& d. @4 I
28. CString   description; //   Port   mapping   description
    % p: C) f# h1 G6 j. Y/ }; Y
29. }   UPNPNAT_MAPPING;
    
30. 
    
31. MyUPnP();
    ( m1 o+ _$ P% {8 d, b
32. ~MyUPnP();
    4 r0 V8 W6 G" ~/ F
33. 
    
34. UPNPNAT_RETURN   AddNATPortMapping(UPNPNAT_MAPPING   *mapping,   bool   tryRandom   =   false);
    
35. UPNPNAT_RETURN   RemoveNATPortMapping(UPNPNAT_MAPPING   mapping,   bool   removeFromList   =   true);
    
36. void   clearNATPortMapping();
    ) z, L' n! p1 [) b0 u+ G( J
37. 
    ; m. z5 r! V" H4 P2 i8 y
38. CString GetLastError();
    ; H3 ~0 a9 D, m
39. CString GetLocalIPStr();
    
40. WORD GetLocalIP();
    7 @1 T$ i; b4 q4 S: {1 X
41. bool IsLANIP(WORD   nIP);
    2 S! V% [) U+ T8 v
42. 
    
43. protected:
    
44. void InitLocalIP();
    
45. void SetLastError(CString   error);
    
46. 
    0 l8 v, y* _! ?% {4 n8 v
47. bool   addPortmap(int   eport,   int   iport,   const   CString&   iclient,
    
48.       const   CString&   descri,   const   CString&   type);
    
49. bool   deletePortmap(int   eport,   const   CString&   type);
    8 n- T) A" l) @1 g2 l1 e
50. 
    
51. bool isComplete()   const   {   return   !m_controlurl.IsEmpty();   }
    3 K& Y# a( d" ^5 E. l
52. 
    / A6 [0 B; \1 [/ Y6 ]  ~; z
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. 
    3 ?, R. h+ f- _: j, {3 k
58. bool Valid()const{return   (!m_name.IsEmpty()&&!m_description.IsEmpty());}
    6 R' k8 k. o4 B4 U0 r; u- ~6 _
59. bool InternalSearch(int   version);
    # @/ I. N& W& G# s0 w' T$ f
60. CString m_devicename;
    0 k$ ]! k( _0 k2 q2 M; V% j( m, Q9 x
61. CString m_name;
    ' I$ c8 n0 l% D9 H; J
62. CString m_description;
    3 h7 ^& k8 t% ~8 f/ D# M6 _
63. CString m_baseurl;
    / R5 F+ s+ H; n( I
64. CString m_controlurl;
    . x# b1 q* u8 Y3 G/ \
65. CString m_friendlyname;
    . r! g2 k7 O, R3 n7 u4 H) h
66. CString m_modelname;
    - T8 c2 _! {, d- b0 x/ f0 Z* r; W
67. int m_version;
    5 q9 f( Y4 z/ Z( f3 B* J, `! k
68. 
    
69. private:
    
70. CList <UPNPNAT_MAPPING,   UPNPNAT_MAPPING>   m_Mappings;
    4 D* w7 c" o  r0 f; i( q
71. 
    
72. CString m_slocalIP;
    
73. CString m_slastError;
    ( l5 ^8 r3 {$ j4 H; p& Y% A7 c; I
74. WORD m_uLocalIP;
    / e# o8 N4 }+ ^
75. 
    2 k+ Y  H6 i9 {1 g8 \# P
76. bool isSearched;
      M  n5 i5 o! m: I. _7 A1 J
77. };
    ' v% G, r3 L" u( @3 K/ A/ _) S) y9 j: {
78. #endif

复制代码

天云上人

/*UPnP.cpp*/

1. 
   , X1 Q( @6 F$ Y
2. #include   "stdafx.h "
   
3. 
   ( u0 h* Z  L8 N& L2 }
4. #include   "upnp.h "
   
5. 
   : g. |! N1 B5 \
6. #define   UPNPPORTMAP0       _T( "WANIPConnection ")
   
7. #define   UPNPPORTMAP1       _T( "WANPPPConnection ")
   
8. #define   UPNPGETEXTERNALIP   _T( "GetExternalIPAddress "),_T( "NewExternalIPAddress ")
   & v: L  F" X' _
9. #define   UPNPADDPORTMAP   _T( "AddPortMapping ")
   
10. #define   UPNPDELPORTMAP   _T( "DeletePortMapping ")
    * C8 X- P; ~& [! a
11. 
    
12. static   const   ulong UPNPADDR   =   0xFAFFFFEF;
      ]0 [  ~! X2 ]7 f
13. static   const   int UPNPPORT   =   1900;
    
14. static   const   CString URNPREFIX   =   _T( "urn:schemas-upnp-org: ");
    # o9 D% q( a2 s8 e
15. 
    3 F4 v% z8 ]% _. q- Q
16. const   CString   getString(int   i)
    
17. {
    # u6 Z2 `4 X. \$ X
18. CString   s;
    
19. 
    ! ?9 b$ }: Y  v( A
20. s.Format(_T( "%d "),   i);
    ; D% \0 C4 \! g! G
21. 
    
22. return   s;
    * p0 F8 x$ ~7 Z' R9 O. k
23. }
    5 K% i7 L, X0 `  W0 _
24. 
    + S: X. p  v+ F8 d9 k# h4 C; {' G2 x
25. const   CString   GetArgString(const   CString&   name,   const   CString&   value)
    6 ]; `- }0 K" y+ {, Q' _* ^
26. {
    2 v$ N7 Z9 C' T7 q! U, ^8 H( u
27. return   _T( " < ")   +   name   +   _T( "> ")   +   value   +   _T( " </ ")   +   name   +   _T( "> ");
    
28. }
    , s, T3 @, P: q- d
29. 
    2 J. {# ], G2 N; a( J) ~; j
30. const   CString   GetArgString(const   CString&   name,   int   value)
    9 w. y3 z. I6 \3 n) f) E9 B5 @
31. {
    
32. return   _T( " < ")   +   name   +   _T( "> ")   +   getString(value)   +   _T( " </ ")   +   name   +   _T( "> ");
    5 o- i& G! Q; J
33. }
    
34. 
    
35. bool   SOAP_action(CString   addr,   uint16   port,   const   CString   request,   CString   &response)
    
36. {
    
37. char   buffer[10240];
    
38. 
    $ D) h- r* s: J
39. const   CStringA   sa(request);
    
40. int   length   =   sa.GetLength();
    
41. strcpy(buffer,   (const   char*)sa);
    
42. 
    
43. uint32   ip   =   inet_addr(CStringA(addr));
    
44. struct   sockaddr_in   sockaddr;
    8 ?8 I7 w  c7 a
45. memset(&sockaddr,   0,   sizeof(sockaddr));
    . ~, z+ z" K$ }6 U. c) D
46. sockaddr.sin_family   =   AF_INET;
    
47. sockaddr.sin_port   =   htons(port);
    
48. sockaddr.sin_addr.S_un.S_addr   =   ip;
    $ q/ W* x/ L! p( y1 D$ Y6 ^5 N
49. int   s   =   socket(AF_INET,   SOCK_STREAM,   0);
    4 i& q" W0 w5 ~% @, o7 Q
50. u_long   lv   =   1;
    5 c( H3 h# F% z- S; B" [
51. ioctlsocket(s,   FIONBIO,   &lv);
    
52. connect(s,   (struct   sockaddr   *)&sockaddr,   sizeof(sockaddr));
    3 i) @/ _7 c# s# c" y
53. Sleep(20);
    9 ]  b6 B: u4 E) `7 G# n; n5 V
54. int   n   =   send(s,   buffer,   length,   0);
    1 @2 X2 P$ o/ J* K: X2 y' H
55. Sleep(100);
    - ?- L. t; d" v# x
56. int   rlen   =   recv(s,   buffer,   sizeof(buffer),   0);
    * N3 {* a0 S7 T0 R6 T3 V3 u* K" O
57. closesocket(s);
    
58. if   (rlen   ==   SOCKET_ERROR)   return   false;
    + `& @; Z- L2 ^7 h& j' w
59. if   (!rlen)   return   false;
    4 W* C0 |9 Z# q) ?9 e
60. 
    5 h4 |4 e& U) p5 N% J
61. response   =   CString(CStringA(buffer,   rlen));
    3 D- Y/ {3 N: h) f% y3 n
62. 
    ) O9 {  r3 c/ K: ?5 t
63. return   true;
    3 A8 ~2 C) i3 U7 Q- V1 u# X8 R
64. }
    0 ^9 j. g  D- K
65. 
    5 J' Z* n" W1 D: r! [  v
66. int   SSDP_sendRequest(int   s,   uint32   ip,   uint16   port,   const   CString&   request)
    1 [9 C8 \9 @, u* r; R
67. {
    , J7 R- ]/ t; t/ X- H
68. char   buffer[10240];
    
69. 
    & D3 i8 Y, w9 k& Q# {2 ^1 a' M
70. const   CStringA   sa(request);
    ; ]$ t; ?, H% ?( @+ Q- `# ^
71. int   length   =   sa.GetLength();
    
72. strcpy(buffer,   (const   char*)sa);
    
73. 
    
74. struct   sockaddr_in   sockaddr;
    8 `5 m$ r6 L; A' |; Z' u3 i; O
75. memset(&sockaddr,   0,   sizeof(sockaddr));
    
76. sockaddr.sin_family   =   AF_INET;
    # Y9 }# `4 e- d  W8 Y9 W9 J
77. sockaddr.sin_port   =   htons(port);
    
78. sockaddr.sin_addr.S_un.S_addr   =   ip;
    7 ]! \. P5 _$ D8 V
79. 
    ' j8 p- q: _) K
80. return   sendto(s,   buffer,   length,   0,   (struct   sockaddr   *)&sockaddr,   sizeof(sockaddr));
    % h  e, [$ o$ C& C+ O
81. }
    2 @- p, J  J# y) B# G9 {  w& D4 }% `
82. 
    
83. bool   parseHTTPResponse(const   CString&   response,   CString&   result)
    
84. {
    
85. int   pos   =   0;
    " b% u6 ^# H& s9 `& R$ D# k4 q9 g
86. 
    / W, l2 P8 T7 P
87. CString   status   =   response.Tokenize(_T( "\r\n "),   pos);
    - |6 L# f" D7 \
88. 
    * V/ K9 }$ l- P( d! `5 Q) M
89. result   =   response;
    . [$ p$ p: s+ l6 Y3 Y
90. result.Delete(0,   pos);
    2 a1 M! V+ l8 ~! r' x/ Q5 e
91. 
    
92. pos   =   0;
    6 a: E4 r' D  `; F& X& w
93. status.Tokenize(_T( "   "),   pos);
    
94. status   =   status.Tokenize(_T( "   "),   pos);
    8 k  z8 P$ w& Q3 p5 Q) t0 z4 {( a
95. if   (status.IsEmpty()   ||   status[0]!= '2 ')   return   false;
    
96. return   true;
    1 ~: M6 P6 H6 o, V
97. }
    & V  A0 n$ h) K
98. 
    
99. const   CString   getProperty(const   CString&   all,   const   CString&   name)
    , J7 k  N2 h, ?: F) c* a* X, Z5 J8 M$ y
100. {
     
101. CString   startTag   =   ' < '   +   name   +   '> ';
     
102. CString   endTag   =   _T( " </ ")   +   name   +   '> ';
     
103. CString   property;
     : f$ n3 ~* {' R
104. 
     . P& [4 h' ?: [( C: a3 q6 M$ Q
105. int   posStart   =   all.Find(startTag);
     
106. if   (posStart <0)   return   CString();
     1 A1 P. i" I! D$ X; w3 U$ w8 Z
107. 
     9 T' j4 T- D, k6 s, @' B$ L+ j
108. int   posEnd   =   all.Find(endTag,   posStart);
     
109. if   (posStart> =posEnd)   return   CString();
     
110. 
     * B) D8 l6 _8 \) n1 U: |! F1 z
111. return   all.Mid(posStart   +   startTag.GetLength(),   posEnd   -   posStart   -   startTag.GetLength());
     
112. }
     
113. 
     6 Q6 F7 \  G2 r, E7 r! v
114. MyUPnP::MyUPnP()
     , n. b  {- c) `
115. :   m_version(1)
     
116. {
     ! c% o# d7 P' d; ]8 Q" V$ q
117. m_uLocalIP   =   0;
     
118. isSearched   =   false;
     2 s- |- l, u9 p) J* N2 f: u
119. }
     
120. 
     
121. MyUPnP::~MyUPnP()
     
122. {
     & k2 P, g3 O" _: u6 [9 ^/ j
123. UPNPNAT_MAPPING   search;
     
124. POSITION   pos   =   m_Mappings.GetHeadPosition();
     - j' D6 V- J+ {" q% ?# A  m
125. while(pos){
     3 }) _4 v% P4 B7 _+ ?! }& T
126. search   =   m_Mappings.GetNext(pos);
     / k. o4 c: E+ E4 h$ \+ w; {- j' s
127. RemoveNATPortMapping(search,   false);
     $ B$ w& i+ z) q
128. }
     
129. 
     * f) ~% @1 u6 j. v8 I
130. m_Mappings.RemoveAll();
     , k3 s: t8 i4 a
131. }
     % z2 n* L& |; O) r6 F7 _
132. 
     : Y; I- ^+ n2 j. g  M- _
133. 
     ) ?! [& h: O2 K  `
134. bool   MyUPnP::InternalSearch(int   version)
     
135. {
     
136. if(version <=0)version   =   1;
     
137. m_version   =   version;
     
138. 
     
139. #define   NUMBEROFDEVICES 2
     - M) ~0 V8 t) ^/ F
140. CString   devices[][2]   =   {
     " W/ [7 ~9 A' g  b% a, a
141. {UPNPPORTMAP1,   _T( "service ")},
     
142. {UPNPPORTMAP0,   _T( "service ")},
     4 ~) i6 d7 {* N4 D# m+ b
143. {_T( "InternetGatewayDevice "),   _T( "device ")},
     
144. };
     . |! w$ w# L2 M
145. 
     ) k/ R  W. L4 i$ W, W6 \) W
146. int   s   =   socket(AF_INET,   SOCK_DGRAM,   0);
     
147. u_long   lv   =   1;
     + A- `$ `8 |, j
148. ioctlsocket(s,   FIONBIO,   &lv);
     
149. 
     8 D8 h1 J/ ~- o* k+ `  I
150. int   rlen   =   0;
     
151. for   (int   i=0;   rlen <=0   &&   i <500;   i++)   {
     
152. if   (!(i%100))   {
     
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);
     
155. CString   request;
     , n3 ]5 w9 n1 z" U% d* X
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. }
     
160. }
     
161. 
     % g6 b6 s, P7 z- ]" l# {3 q
162. Sleep(10);
     
163. 
     / D4 P2 M( C6 Y$ _
164. char   buffer[10240];
     
165. rlen   =   recv(s,   buffer,   sizeof(buffer),   0);
     $ i4 o4 p/ r6 x  g
166. if   (rlen   <=   0)   continue;
     
167. closesocket(s);
     
168. 
     
169. CString   response   =   CString(CStringA(buffer,   rlen));
     
170. CString   result;
     ) v( H9 y/ Q' t* X
171. if   (!parseHTTPResponse(response,   result))   return   false;
     9 R2 }2 u2 M& E! m5 @
172. 
     
173. for   (int   d=0;   d <NUMBEROFDEVICES;   d++)   {
     . K- P+ z' Z! G+ W  P5 ?
174. m_name.Format(_T( "%s%s:%s:%d "),   URNPREFIX,   devices[d][1],   devices[d][0],   version);
     9 y9 f2 [% B+ N7 E& r) x
175. if   (result.Find(m_name)   > =   0)   {
     
176. for   (int   pos   =   0;;)   {
     6 Q' H3 S3 q' c4 U' P
177. CString   line   =   result.Tokenize(_T( "\r\n "),   pos);
     
178. if   (line.IsEmpty())   return   false;
     7 [+ C+ G9 F& H+ X" Y' g4 J$ W6 r2 A
179. CString   name   =   line.Mid(0,   9);
     ( B# U$ ^) {4 y2 z4 g7 j. q9 I4 E
180. name.MakeUpper();
     7 m1 z$ A# y8 g7 ?1 h( C
181. if   (name   ==   _T( "LOCATION: "))   {
     
182. line.Delete(0,   9);
     6 i6 l# u! e; ]9 P
183. m_description   =   line;
       T$ I( u1 V8 T0 V$ ^7 y% M1 F
184. m_description.Trim();
     
185. return   GetDescription();
     
186. }
     3 |5 D3 E$ ^( u8 O1 Q" ?' h
187. }
     
188. }
     $ E# w, t1 B" ?) ?; V+ E. J
189. }
     ) x6 i4 w8 W) G4 |( r( F
190. }
     $ U5 b9 u4 Y+ G% Q1 D& T) V
191. closesocket(s);
     ; W' [& g( C. @. h0 p2 B
192. 
     
193. return   false;
     $ p$ Z* Y: ~" ~- ?/ N( O! q7 b
194. }
     " k1 S. A1 U! ]- H- k5 B

复制代码

天云上人

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

, i+ ~+ p6 C5 r" r; U7 O) ]
6 z' p# f* W- c9 T, }9 H9 j///////////////////////////////////////////
//upnp的抽象接口类,只需调用这些接口即可实现端口映射功能.
  e( n$ L) \; d/ U& J

! F5 G5 {9 s" }* X3 k#pragma once
#include <exception>
  p2 z/ P; [. L1 F& m- ^0 w

  enum TRISTATE{
4 a! r$ Z( `* t- M        TRIS_FALSE,
        TRIS_UNKNOWN,
        TRIS_TRUE
6 C- t1 \) L: p4 |};
2 b: E: W6 N9 W8 _2 o
9 X' p7 s- v2 z- y  X  y- S
& ^* v2 f# W/ d" U! r2 n; Menum UPNP_IMPLEMENTATION{
' f9 C& V) `* O/ ~& k- ?9 p        UPNP_IMPL_WINDOWSERVICE = 0,
) y) E- {/ r' R  N# Y5 ?3 n) ]) _* ^5 w        UPNP_IMPL_MINIUPNPLIB,
        UPNP_IMPL_NONE /*last*/
};
0 g+ ^( ^/ X+ n* g
3 P2 {2 @1 I& v# H: q
: v/ V& ]: G, i( J
1 g  V! A3 a, v) t- J# e
class CUPnPImpl
{
. ^6 X1 |  ^. P6 z5 Q( q) n/ M, D2 U! Mpublic:
/ n" \3 w' j( C9 {/ K4 P        CUPnPImpl();
        virtual ~CUPnPImpl();
        struct UPnPError : std::exception {};
        enum {
& z0 ^( a  h0 o" i. o: T. A' i6 Z                UPNP_OK,
( c6 O4 }2 G$ w6 X! ^8 G                UPNP_FAILED,
* q5 o1 {- f; g0 s                UPNP_TIMEOUT
$ s3 A8 }3 m1 O5 R, [( ?) z        };


, K  y3 S8 r: ~; \) p) Y, l9 N        virtual void        StartDiscovery(uint16 nTCPPort, uint16 nUDPPort, uint16 nTCPWebPort) = 0;
        virtual bool        CheckAndRefresh() = 0;
' E; G$ w& P3 \  ]! r, a1 E        virtual void        StopAsyncFind() = 0;
        virtual void        DeletePorts() = 0;
        virtual bool        IsReady() = 0;
        virtual int                GetImplementationID() = 0;
- d# D% m" z5 {* b8 A       
0 h3 r0 J8 g; P7 }        void                        LateEnableWebServerPort(uint16 nPort); // Add Webserverport on already installed portmapping
' q* W% i* x$ I- G

; G% l/ A1 h! I$ N$ U        void                        SetMessageOnResult(HWND hWindow, UINT nMessageID);
        TRISTATE                ArePortsForwarded() const                                                                { return m_bUPnPPortsForwarded; }
, ~; p6 z" C& a; Q: j" }( J        uint16                        GetUsedTCPPort()                                                                                { return m_nTCPPort; }
7 G' D/ m3 j/ ~. @( g: B% P        uint16                        GetUsedUDPPort()                                                                                { return m_nUDPPort; }       
5 Q+ P3 f/ ]! `' W

// Implementation
protected:
        volatile TRISTATE        m_bUPnPPortsForwarded;
3 R7 V/ l. e4 m6 `  S        void                                SendResultMessage();
* u  j  d) E3 |        uint16                                m_nUDPPort;
' m1 @- P% I5 r- s; c$ c% Y. D        uint16                                m_nTCPPort;
        uint16                                m_nTCPWebPort;
        bool                                m_bCheckAndRefresh;
6 k+ m4 t# I8 Y& j
) ?  a7 I, o9 d
  ?1 L# G, Z" l% e2 \. @6 M' u* Rprivate:
        HWND        m_hResultMessageWindow;
        UINT        m_nResultMessageID;
/ H/ M9 z7 E7 m% Q
0 g# R* S& _' s* W8 @
};
% v* S0 F- W) N- @4 m1 S
3 A4 k6 K! E' T+ V. c. ]( ~9 s
, Q7 A& v8 ~) c7 }0 n$ [, w// Dummy Implementation to be used when no other implementation is available
' h( I2 M) {  E/ b) wclass CUPnPImplNone: public CUPnPImpl
) ]* A6 A0 E9 M2 i+ Y* n* `{
+ [* g4 w6 E, Vpublic:
" i- G& x  z0 I& K, ^        virtual void        StartDiscovery(uint16, uint16, uint16)                                        { ASSERT( false ); }
        virtual bool        CheckAndRefresh()                                                                                { return false; }
        virtual void        StopAsyncFind()                                                                                        { }
9 q' A& e6 h" ?5 W$ t        virtual void        DeletePorts()                                                                                        { }
        virtual bool        IsReady()                                                                                                { return false; }
& I& r8 v$ P7 z" D4 k        virtual int                GetImplementationID()                                                                        { return UPNP_IMPL_NONE; }
};

5 O& v, S8 c4 \% L- M# c
/////////////////////////////////////
' Z! @. ]! y% G% R8 U. w9 l//下面是使用windows操作系统自带的UPNP功能的子类

; m7 \# q, }. Z/ ?# Z4 u! X+ Y9 |
7 l5 u( T9 A$ m( X+ [#pragma once
#pragma warning( disable: 4355 )


( Y6 ]0 N* y& X, n3 y; a6 R#include "UPnPImpl.h"
! N  K) [" j8 n/ g- q5 ]. y, B#include <upnp.h>
2 T" Q6 i: R$ F1 a" Y6 S/ L#include <iphlpapi.h>
#include <comdef.h>
5 z# f" v% s8 L2 ~. E2 q#include <winsvc.h>

+ J8 [5 c) R/ Q. x, J' o/ k
0 s5 a% H9 D) S#include <vector>
#include <exception>
#include <functional>


; l: x# ~9 X" W" |( n9 P9 J. D' o

typedef _com_ptr_t<_com_IIID<IUPnPDeviceFinder,&IID_IUPnPDeviceFinder> >        FinderPointer;
" r: R( A3 s% z! O0 @* Htypedef _com_ptr_t<_com_IIID<IUPnPDevice,&IID_IUPnPDevice>        >                                DevicePointer;
. k9 f# o% a9 ftypedef _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;
  H5 g  L: e' u/ p" w1 W" k# `typedef _com_ptr_t<_com_IIID<IUnknown,&IID_IUnknown> >                                                UnknownPtr;

# X3 n/ C5 k7 |+ q- B+ F. {
typedef DWORD (WINAPI* TGetBestInterface) (
: b# y4 l; u! y. [' v, |  IPAddr dwDestAddr,
0 u4 P+ \; G  f  PDWORD pdwBestIfIndex
);


: I* q6 b$ E( Ttypedef DWORD (WINAPI* TGetIpAddrTable) (
' Z: H# y( c5 i$ S2 t3 j, J  PMIB_IPADDRTABLE pIpAddrTable,
8 i3 e2 L. M: b$ U( v  PULONG pdwSize,
- d9 k2 @( U, M0 I  u. H2 `# k1 A  S  BOOL bOrder
3 G  {" T1 ~! B$ a);
) a, Y3 Z: i8 O" v
+ [: o" r$ u+ b$ D6 P2 O1 }
typedef DWORD (WINAPI* TGetIfEntry) (
6 ]4 k/ V. s' {+ I  PMIB_IFROW pIfRow
);

0 Y# D; P% c4 \5 }( G
* D) z  Z* p# rCString translateUPnPResult(HRESULT hr);
6 k/ S2 A- A+ pHRESULT UPnPMessage(HRESULT hr);
. e# m$ y& O' u

class CUPnPImplWinServ: public CUPnPImpl
. @1 s) w! S% H6 h$ y2 Q5 w7 V{
        friend class CDeviceFinderCallback;
        friend class CServiceCallback;
$ Y) |- R- M. U) I( I* U2 C+ j% r// Construction
( A; d' N1 X  L& d1 jpublic:
0 m/ W- |& S, {1 E% z/ r        virtual ~CUPnPImplWinServ();
. ^6 N( a5 D$ g        CUPnPImplWinServ();

( P, @# f8 ^1 g; }; J
        virtual void        StartDiscovery(uint16 nTCPPort, uint16 nUDPPort, uint16 nTCPWebPort)                { StartDiscovery(nTCPPort, nUDPPort, nTCPWebPort, false); }
0 m% F& i! F* t9 M& o, L6 d* E* q        virtual void        StopAsyncFind();
        virtual void        DeletePorts();
        virtual bool        IsReady();
        virtual int                GetImplementationID()                                                                        { return UPNP_IMPL_WINDOWSERVICE; }


        // No Support for Refreshing on this  (fallback) implementation yet - in many cases where it would be needed (router reset etc)
& g0 e! g5 b! r5 x        // the windows side of the implementation tends to get bugged untill reboot anyway. Still might get added later
) Q  n; [5 X/ i$ a4 M& J        virtual bool        CheckAndRefresh()                                                                                { return false; };
# n- \( o; o  Q: [+ i) Z7 {5 A- Y
( v/ V6 a+ H( M" y
' h! S. e& y% S1 Sprotected:
        void        StartDiscovery(uint16 nTCPPort, uint16 nUDPPort, uint16 nTCPWebPort, bool bSecondTry);
$ q' Y6 o' N: s9 f, p- u2 d        void        AddDevice(DevicePointer pDevice, bool bAddChilds, int nLevel = 0);
        void        RemoveDevice(CComBSTR bsUDN);
# e7 Y8 h: W: M6 x' Q        bool        OnSearchComplete();
8 ?7 \. U/ W) ?/ B  y        void        Init();
: f& n8 d' O8 N  R  y; b& W
! ]( r* y- V; g+ b% Q- D
! Z+ r  G- U* N; X! S( v& e        inline bool IsAsyncFindRunning()
- x5 Y, m: U" H, {! n        {
0 @) o  d8 J) B3 A' W                if ( m_pDeviceFinder != NULL && m_bAsyncFindRunning && GetTickCount() - m_tLastEvent > 10000 )
* k7 G# E0 E3 m+ G& }                {
1 Y9 a% _/ I$ g* r3 e7 V; E                        m_pDeviceFinder->CancelAsyncFind( m_nAsyncFindHandle );
                        m_bAsyncFindRunning = false;
                }
9 f. M% h. b4 I. S, O                MSG msg;
                while ( PeekMessage( &msg, NULL, 0, 0, PM_REMOVE ) )
                {
% F+ w: I2 z. J  z4 w                        TranslateMessage( &msg );
5 L2 K6 L4 n6 c4 j6 [                        DispatchMessage( &msg );
                }
+ X8 w1 i7 M$ }4 W' L  H, h                return m_bAsyncFindRunning;
        }

3 H! a! C4 Q0 v5 H' }' a3 X) v! E6 H
# E/ ^  g& u) m' q2 @0 |        TRISTATE                        m_bUPnPDeviceConnected;
5 S% E2 e  V: t: \
0 i3 S# i5 _9 K5 W6 A# Y
& c+ G' Z& ?# N+ [4 z# n// Implementation
        // API functions
        SC_HANDLE (WINAPI *m_pfnOpenSCManager)(LPCTSTR, LPCTSTR, DWORD);
" X/ z: v1 R3 V7 ?+ p        SC_HANDLE (WINAPI *m_pfnOpenService)(SC_HANDLE, LPCTSTR, DWORD);
: K+ H, ~2 Q$ j$ j2 y4 d- C        BOOL (WINAPI *m_pfnQueryServiceStatusEx)(SC_HANDLE, SC_STATUS_TYPE, LPBYTE, DWORD, LPDWORD);
! R6 O% j4 H" z, s        BOOL (WINAPI *m_pfnCloseServiceHandle)(SC_HANDLE);
        BOOL (WINAPI *m_pfnStartService)(SC_HANDLE, DWORD, LPCTSTR*);
) r% Z$ u1 `( O$ X4 ?' g) R        BOOL (WINAPI *m_pfnControlService)(SC_HANDLE, DWORD, LPSERVICE_STATUS);
4 T6 s0 e1 M2 h. w
9 N  E4 i8 J" t; V% }
        TGetBestInterface                m_pfGetBestInterface;
        TGetIpAddrTable                        m_pfGetIpAddrTable;
4 T4 v/ R8 @: S& m, |        TGetIfEntry                                m_pfGetIfEntry;


* n4 F9 q- g! g$ \. S" x! {        static FinderPointer CreateFinderInstance();
        struct FindDevice : std::unary_function< DevicePointer, bool >
0 }( d& {$ r9 v7 ^7 r% j2 f        {
; n4 W0 S) V* {6 e, A1 V1 |+ Z                FindDevice(const CComBSTR& udn) : m_udn( udn ) {}
                result_type operator()(argument_type device) const
% T5 H6 ]: M+ u' u  ^! W8 k4 c% ?                {
                        CComBSTR deviceName;
                        HRESULT hr = device->get_UniqueDeviceName( &deviceName );


                        if ( FAILED( hr ) )
% f: I- p  G& t  L) Z8 P* k                                return UPnPMessage( hr ), false;
7 H: F2 f, n0 L  K2 ^3 i

6 M( i$ y; |  A- f7 {  \                        return wcscmp( deviceName.m_str, m_udn ) == 0;
  J" i2 s7 J. A. V) O0 J* X4 s                }
' ^( N2 I" `4 i" P( F. V                CComBSTR m_udn;
+ f3 Z- o- Z  ]' g6 `9 i( g# P        };
       
        void        ProcessAsyncFind(CComBSTR bsSearchType);
9 O& Y5 U# v% w! ?' m( O        HRESULT        GetDeviceServices(DevicePointer pDevice);
+ ^6 P/ u2 b* U: J" A% I        void        StartPortMapping();
1 ~$ \2 c. W: A; G        HRESULT        MapPort(const ServicePointer& service);
        void        DeleteExistingPortMappings(ServicePointer pService);
  K" |. q, I. k        void        CreatePortMappings(ServicePointer pService);
        HRESULT SaveServices(EnumUnknownPtr pEU, const LONG nTotalItems);
        HRESULT InvokeAction(ServicePointer pService, CComBSTR action,
% y& Z; I+ B+ C; X& \/ T                LPCTSTR pszInArgString, CString& strResult);
        void        StopUPnPService();
: t7 u3 h# z* A5 d- u, w  \% h

        // Utility functions
6 O# [7 b0 O6 n) L* T4 k# d+ ]        HRESULT CreateSafeArray(const VARTYPE vt, const ULONG nArgs, SAFEARRAY** ppsa);
; Q2 p2 ]9 H* T: c3 c0 v' v. \        INT_PTR CreateVarFromString(const CString& strArgs, VARIANT*** pppVars);
/ F# b: N# w9 j. S- |* W" }  R. m        INT_PTR        GetStringFromOutArgs(const VARIANT* pvaOutArgs, CString& strArgs);
5 M( E! @0 @& I, u0 q        void        DestroyVars(const INT_PTR nCount, VARIANT*** pppVars);
* \$ {9 T4 n( u$ X4 }- p, }        HRESULT GetSafeArrayBounds(SAFEARRAY* psa, LONG* pLBound, LONG* pUBound);
        HRESULT GetVariantElement(SAFEARRAY* psa, LONG pos, VARIANT* pvar);
4 K# ]9 _! [3 [- ^! }+ O) f        CString        GetLocalRoutableIP(ServicePointer pService);


// Private members
private:
        DWORD        m_tLastEvent;        // When the last event was received?
/ ^/ [% q& c* _5 v        std::vector< DevicePointer >  m_pDevices;
: \2 \% n) g/ H$ B5 ~$ ]( J& T        std::vector< ServicePointer > m_pServices;
- y; C2 @1 u# _, {        FinderPointer                        m_pDeviceFinder;
  w" H. }: `) `2 {- W. |1 h        DeviceFinderCallback        m_pDeviceFinderCallback;
8 Y8 H2 H# u% Q$ _$ q+ U; l        ServiceCallback                        m_pServiceCallback;

0 R; t7 T) C7 U
        LONG        m_nAsyncFindHandle;
        bool        m_bCOM;
        bool        m_bPortIsFree;
        CString m_sLocalIP;
  T3 I  F3 C8 a& s% O! R% p        CString m_sExternalIP;
1 z3 F1 e9 z; ?# }        bool        m_bADSL;                // Is the device ADSL?
* K. {# ?7 F/ V- ]. K        bool        m_ADSLFailed;        // Did port mapping failed for the ADSL device?
        bool        m_bInited;
5 F: o$ _- @/ r8 o0 B; b6 U        bool        m_bAsyncFindRunning;
        HMODULE m_hADVAPI32_DLL;
        HMODULE        m_hIPHLPAPI_DLL;
% F% q& \+ Y7 h1 l        bool        m_bSecondTry;
        bool        m_bServiceStartedByEmule;
        bool        m_bDisableWANIPSetup;
        bool        m_bDisableWANPPPSetup;
  L" M+ n# l( D; \( a" C! y. Y( C

};


4 @1 D0 Z3 G* B# B1 s// DeviceFinder Callback
1 H$ f: B3 ]1 t8 j# qclass CDeviceFinderCallback
        : public IUPnPDeviceFinderCallback
& V1 n* x5 T6 T: g. V{
public:
        CDeviceFinderCallback(CUPnPImplWinServ& instance)
1 x+ s% ^2 R' D. U) E  q* |) i) G  g                : m_instance( instance )
        { m_lRefCount = 0; }


# ~7 Z; E) v0 f' O, C; w* o   STDMETHODIMP QueryInterface(REFIID iid, LPVOID* ppvObject);
1 W: ^/ b4 z6 z' n7 l   STDMETHODIMP_(ULONG) AddRef();
7 g. q0 l7 r0 n! g   STDMETHODIMP_(ULONG) Release();


// implementation
private:
- F, S" M6 i; ~; Q        HRESULT __stdcall DeviceAdded(LONG nFindData, IUPnPDevice* pDevice);
8 C2 A# E* D, A; M; g- [$ v        HRESULT __stdcall DeviceRemoved(LONG nFindData, BSTR bsUDN);
  H3 M5 J- I, q$ J# v        HRESULT __stdcall SearchComplete(LONG nFindData);

' Y. g" }& G2 E4 v
private:
+ c: j1 q0 A- t  w% V: k+ @        CUPnPImplWinServ& m_instance;
1 {9 Z# }+ p, }" i, k        LONG m_lRefCount;
};


8 m- l" u) j5 B6 c// Service Callback
4 P- Z5 j7 H, y) tclass CServiceCallback
        : public IUPnPServiceCallback
% F# ]) i/ w1 A4 I. _{
  L2 T4 V4 s8 \) m5 cpublic:
5 z# h' K! c) l( L        CServiceCallback(CUPnPImplWinServ& instance)
" P' q4 X9 O7 A4 ?) m( T2 Q                : m_instance( instance )
# V& D5 `) O8 K% ~        { m_lRefCount = 0; }
   
   STDMETHODIMP QueryInterface(REFIID iid, LPVOID* ppvObject);
5 T; q# K* w4 B" D$ _' \+ d   STDMETHODIMP_(ULONG) AddRef();
   STDMETHODIMP_(ULONG) Release();

7 o6 ~$ a, K7 H: N' V) E
' ]5 E; T% G- ]/ s4 ~// implementation
' Z/ U/ I$ W3 z5 F8 J+ m- eprivate:
( L$ R- F+ o) H: D9 ?7 m' t1 j6 D        HRESULT __stdcall StateVariableChanged(IUPnPService* pService, LPCWSTR pszStateVarName, VARIANT varValue);
  m1 ~. N) H2 q% C# H+ Z/ D        HRESULT __stdcall ServiceInstanceDied(IUPnPService* pService);


$ o8 F" S8 E4 h- Bprivate:
        CUPnPImplWinServ& m_instance;
( a, O) O$ s: d# @+ D        LONG m_lRefCount;
4 m. g) p5 [- T. v) }( m};
7 n9 K" ^% Y* Z! S3 m

- S& S, f' C( g/////////////////////////////////////////////////
; f7 q6 Z" M. r: O

8 h5 k3 }$ C6 J使用时只需要使用抽象类的接口。
% N7 H1 Z+ U3 dCUPnPImpl::SetMessageOnResult设置需要接受UPNP端口映射是否成功的窗口句柄和消息ID.
8 o) W$ S: O1 _" Y" ]7 Z, m8 _, @CUPnPImpl::StartDiscovery将开启一个异步设备查找并进行端口映射,其参数为需要映射的内网端口.
, ]- u( O/ X$ u0 y: k) Z) |CUPnPImpl::StopAsyncFind停止设备查找.
CUPnPImpl::DeletePorts删除端口映射.