UPnP

天云上人

/*uPnP.h*/

1. 
   0 [  g' W: E0 r' C& n
2. #ifndef   MYUPNP_H_
   
3. 
   
4. #pragma   once
   9 b: L, i' f# f9 K5 t( y4 k! D
5. 
   0 H$ T; S0 J7 f7 I0 b' x
6. typedef   unsigned   long   ulong;
   
7. 
   " N5 R3 q  i# \0 t
8. class   MyUPnP
   
9. {
   
10. public:
    
11. typedef   enum{
    
12. UNAT_OK, //   Successfull
    5 p9 t0 R  m$ b, h
13. UNAT_ERROR, //   Error,   use   GetLastError()   to   get   an   error   description
    5 _2 S/ t1 V8 W7 E9 _/ K' p+ R. h
14. UNAT_NOT_OWNED_PORTMAPPING, //   Error,   you   are   trying   to   remove   a   port   mapping   not   owned   by   this   class
    % E0 b, M+ q" r. [" W
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;
    
18. 
    
19. typedef   enum{
    
20. UNAT_TCP, //   TCP   Protocol
    : K) c# e4 ^" ^7 L
21. UNAT_UDP //   UDP   Protocol
    3 ?! L# A5 ?* d$ q6 C2 U' J! j! ^
22. }   UPNPNAT_PROTOCOL;
    
23. 
    7 z9 r- E  ?6 ^" D4 E
24. typedef   struct{
    
25. WORD   internalPort; //   Port   mapping   internal   port
    
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
    
29. }   UPNPNAT_MAPPING;
    # \( q( g7 V2 a! \% S
30. 
    8 Y. o/ a; V9 u$ m' {
31. MyUPnP();
    
32. ~MyUPnP();
    
33. 
    
34. UPNPNAT_RETURN   AddNATPortMapping(UPNPNAT_MAPPING   *mapping,   bool   tryRandom   =   false);
    
35. UPNPNAT_RETURN   RemoveNATPortMapping(UPNPNAT_MAPPING   mapping,   bool   removeFromList   =   true);
    
36. void   clearNATPortMapping();
    ; N0 ~4 Q) S' A0 K8 w4 ?0 r
37. 
    
38. CString GetLastError();
    
39. CString GetLocalIPStr();
    - u( y6 z$ ~3 Q2 t1 q7 a
40. WORD GetLocalIP();
    
41. bool IsLANIP(WORD   nIP);
    
42. 
    8 k( a# A! Q4 E3 ?' O
43. protected:
    
44. void InitLocalIP();
    
45. void SetLastError(CString   error);
    
46. 
    * a: h  Q6 z9 b/ F& m! S. G2 H/ [
47. bool   addPortmap(int   eport,   int   iport,   const   CString&   iclient,
    
48.       const   CString&   descri,   const   CString&   type);
    " \5 r1 Q$ f5 s* z
49. bool   deletePortmap(int   eport,   const   CString&   type);
    
50. 
    
51. bool isComplete()   const   {   return   !m_controlurl.IsEmpty();   }
    % w% O# S1 o( T9 j" U
52. 
    7 n( @# W. N& v! u
53. bool Search(int   version=1);
    % ~8 w5 G. K( j( Q  s, T
54. bool GetDescription();
    2 G6 H+ ?2 r' b$ M$ z
55. CString GetProperty(const   CString&   name,   CString&   response);
    ! p0 s+ H7 @4 `+ g' e* }+ }1 ]. @+ H
56. bool InvokeCommand(const   CString&   name,   const   CString&   args);
    
57. 
    
58. bool Valid()const{return   (!m_name.IsEmpty()&&!m_description.IsEmpty());}
    
59. bool InternalSearch(int   version);
    
60. CString m_devicename;
      h% ]! M: p! L3 p
61. CString m_name;
    2 e( J3 H# B5 ?4 L; @4 d; c
62. CString m_description;
      W# x+ I7 Y* L* f
63. CString m_baseurl;
    ; s- D/ A3 }) }$ Z5 J; q& D
64. CString m_controlurl;
    5 f2 c. M5 q6 M
65. CString m_friendlyname;
    5 g: [0 J, ^$ H# z8 s( u, W. A7 ~
66. CString m_modelname;
    
67. int m_version;
    
68. 
    
69. private:
    8 _' x5 e( {0 V4 r0 `" M
70. CList <UPNPNAT_MAPPING,   UPNPNAT_MAPPING>   m_Mappings;
    7 R, e' ^  F) ]5 g
71. 
    % r/ G! r  a) q
72. CString m_slocalIP;
    
73. CString m_slastError;
    ' r2 j/ x% J( z- S) {; Q
74. WORD m_uLocalIP;
    
75. 
    
76. bool isSearched;
    
77. };
    . K: g  _2 e& k7 _2 ]
78. #endif

复制代码

天云上人

/*UPnP.cpp*/

1. 
   0 }' f/ u& _% W- W6 P
2. #include   "stdafx.h "
   ' V* N  J1 @2 U: p3 [0 k
3. 
   . k* K8 _' \1 j
4. #include   "upnp.h "
   0 S5 V6 y# z& @6 ]: D- h
5. 
   
6. #define   UPNPPORTMAP0       _T( "WANIPConnection ")
   + |$ \+ k+ H6 x/ f& {
7. #define   UPNPPORTMAP1       _T( "WANPPPConnection ")
   
8. #define   UPNPGETEXTERNALIP   _T( "GetExternalIPAddress "),_T( "NewExternalIPAddress ")
   8 _& U" Q0 _7 i, S. E' T# F8 J4 ^
9. #define   UPNPADDPORTMAP   _T( "AddPortMapping ")
   % ~8 n) k: D) x& t
10. #define   UPNPDELPORTMAP   _T( "DeletePortMapping ")
    % n: S. z1 `# s: J0 v+ d( ~* p
11. 
    , @+ c) O% b) [: ]$ p' h. K; H3 B
12. static   const   ulong UPNPADDR   =   0xFAFFFFEF;
      c, N& N7 H0 j9 |9 t( J
13. static   const   int UPNPPORT   =   1900;
    
14. static   const   CString URNPREFIX   =   _T( "urn:schemas-upnp-org: ");
    1 d0 M, t) e. c% W1 y% L/ b
15. 
    " C9 w+ C: |" J! W, C1 i: G
16. const   CString   getString(int   i)
    3 g: U- l6 ?* Z# F" f, {7 z3 j
17. {
    , ?3 Z% X6 }- X8 T
18. CString   s;
    2 z6 ^3 u& D4 f8 E1 B( r, |3 l
19. 
    8 T7 C1 P" a! i4 V; K* r! V
20. s.Format(_T( "%d "),   i);
    
21. 
    
22. return   s;
    
23. }
      ]& `% d# J2 L2 j
24. 
    
25. const   CString   GetArgString(const   CString&   name,   const   CString&   value)
    
26. {
    0 r1 e9 b1 d* @% L7 U4 X: W
27. return   _T( " < ")   +   name   +   _T( "> ")   +   value   +   _T( " </ ")   +   name   +   _T( "> ");
    
28. }
    4 `' [0 W2 Y0 Y& e+ @9 N: m( w
29. 
    " I) x+ k$ ^& O: z# o% M3 n
30. const   CString   GetArgString(const   CString&   name,   int   value)
    
31. {
    % t( m) [# u/ x- W
32. return   _T( " < ")   +   name   +   _T( "> ")   +   getString(value)   +   _T( " </ ")   +   name   +   _T( "> ");
    ) b" e: @$ `! v8 S
33. }
    
34. 
    # g5 N6 ^' p* m+ G
35. bool   SOAP_action(CString   addr,   uint16   port,   const   CString   request,   CString   &response)
    # w5 ^# I6 N  ~( l+ c; }
36. {
    ' j  k3 l) t: C8 a4 ]: C- c
37. char   buffer[10240];
    9 O$ \7 f9 f1 B6 ]
38. 
    : C7 q' G3 G* U7 k/ y2 c$ ^6 K! {; P
39. const   CStringA   sa(request);
    # _: D$ k9 K5 I9 C3 u- x
40. int   length   =   sa.GetLength();
    
41. strcpy(buffer,   (const   char*)sa);
    
42. 
    8 P6 ~  D6 C. `0 j) m4 F( l8 |
43. uint32   ip   =   inet_addr(CStringA(addr));
    " m1 ^* ~' q2 y1 h
44. struct   sockaddr_in   sockaddr;
    8 W9 u) R  L1 c4 V" p* b
45. memset(&sockaddr,   0,   sizeof(sockaddr));
    
46. sockaddr.sin_family   =   AF_INET;
    
47. sockaddr.sin_port   =   htons(port);
    
48. sockaddr.sin_addr.S_un.S_addr   =   ip;
    
49. int   s   =   socket(AF_INET,   SOCK_STREAM,   0);
    - a& \. j5 u* m
50. u_long   lv   =   1;
    
51. ioctlsocket(s,   FIONBIO,   &lv);
    3 h2 r* q- U+ R2 C% A, K9 B6 t
52. connect(s,   (struct   sockaddr   *)&sockaddr,   sizeof(sockaddr));
    # f/ l" ~# v- x. Z
53. Sleep(20);
    
54. int   n   =   send(s,   buffer,   length,   0);
    4 U- R) c7 O! U3 D0 V
55. Sleep(100);
    
56. int   rlen   =   recv(s,   buffer,   sizeof(buffer),   0);
    
57. closesocket(s);
    3 g9 _& H( V7 U# ^
58. if   (rlen   ==   SOCKET_ERROR)   return   false;
    
59. if   (!rlen)   return   false;
    8 |: |( _0 U( M  _- }
60. 
    
61. response   =   CString(CStringA(buffer,   rlen));
    
62. 
    
63. return   true;
    
64. }
    / l' {* z; O/ W+ d! x7 h
65. 
    
66. int   SSDP_sendRequest(int   s,   uint32   ip,   uint16   port,   const   CString&   request)
    
67. {
    1 e, F6 J: |: F# G. q! b3 [
68. char   buffer[10240];
    
69. 
      l0 d0 f& ~4 p$ K" H! q$ q" @
70. const   CStringA   sa(request);
    - ^0 n% d' n3 D: ^4 y
71. int   length   =   sa.GetLength();
    
72. strcpy(buffer,   (const   char*)sa);
    
73. 
    1 F2 E0 {3 J3 ?* g( k( [# h
74. struct   sockaddr_in   sockaddr;
    
75. memset(&sockaddr,   0,   sizeof(sockaddr));
    
76. sockaddr.sin_family   =   AF_INET;
    
77. sockaddr.sin_port   =   htons(port);
    ! D1 ]6 Z; V0 s7 W- K/ w  \! b' o% v
78. sockaddr.sin_addr.S_un.S_addr   =   ip;
    
79. 
    % Y: \* l. c  L4 x* y* [# {
80. return   sendto(s,   buffer,   length,   0,   (struct   sockaddr   *)&sockaddr,   sizeof(sockaddr));
    & H$ n$ w- T( |% i% c
81. }
    
82. 
    - C" B( I1 P' s: a& u! c
83. bool   parseHTTPResponse(const   CString&   response,   CString&   result)
    
84. {
    
85. int   pos   =   0;
    ' M9 I! Y& y' J) I4 I* H& C
86. 
    
87. CString   status   =   response.Tokenize(_T( "\r\n "),   pos);
    * Q5 ^. b) ^% f" n% G# }
88. 
    " L, ?8 u1 Z: }  K2 Y' D( d5 F
89. result   =   response;
    3 Q3 M  I5 b* Z- C8 ?
90. result.Delete(0,   pos);
    # j5 K% q* R8 R/ ^$ R/ b
91. 
    
92. pos   =   0;
    
93. status.Tokenize(_T( "   "),   pos);
    8 h# ]& J) v, ~( l9 N, ^. j
94. status   =   status.Tokenize(_T( "   "),   pos);
    
95. if   (status.IsEmpty()   ||   status[0]!= '2 ')   return   false;
    % s6 l& e/ _( V
96. return   true;
    6 z  V& T5 l9 r5 `
97. }
    5 t, U5 O( Y4 X" \! Z& y3 L9 _4 K0 L
98. 
    
99. const   CString   getProperty(const   CString&   all,   const   CString&   name)
    
100. {
     ! q6 m& g* \) _9 F, F
101. CString   startTag   =   ' < '   +   name   +   '> ';
     
102. CString   endTag   =   _T( " </ ")   +   name   +   '> ';
     * L! d$ Q" Z. r
103. CString   property;
       v9 F. Q# L9 }# t# d
104. 
     
105. int   posStart   =   all.Find(startTag);
     
106. if   (posStart <0)   return   CString();
     + t# u: o" |+ T6 T; S. J( r; j; V
107. 
     - V5 H- b9 U  G% }
108. int   posEnd   =   all.Find(endTag,   posStart);
     
109. if   (posStart> =posEnd)   return   CString();
     2 G: d3 F$ Y! U, q. T# d+ i1 K
110. 
     
111. return   all.Mid(posStart   +   startTag.GetLength(),   posEnd   -   posStart   -   startTag.GetLength());
     ) l% a% Q& g( u1 K& A
112. }
     9 H) W: C1 d% M" ?
113. 
     3 a. @" }* w$ u2 @0 k
114. MyUPnP::MyUPnP()
     
115. :   m_version(1)
     
116. {
     / K" n( j8 p+ w, @' s4 x: G
117. m_uLocalIP   =   0;
     
118. isSearched   =   false;
     
119. }
     - @* ]# l& Y) u9 y/ [$ H; z2 i
120. 
     
121. MyUPnP::~MyUPnP()
     
122. {
     
123. UPNPNAT_MAPPING   search;
     % {, l, v" L% _6 J+ w  Y
124. POSITION   pos   =   m_Mappings.GetHeadPosition();
     / N: d9 G$ `+ h! R. X
125. while(pos){
     ' ^3 K+ N. ?$ Q$ V: ~8 Y
126. search   =   m_Mappings.GetNext(pos);
     
127. RemoveNATPortMapping(search,   false);
     $ V) Z7 O7 O: L# R( ?0 z( k
128. }
     
129. 
     
130. m_Mappings.RemoveAll();
     * L; N3 F2 F  \8 c! }
131. }
     - M/ ^4 o& t* X
132. 
     
133. 
     ! o& Z( J: k$ V! @  P
134. bool   MyUPnP::InternalSearch(int   version)
     
135. {
     
136. if(version <=0)version   =   1;
     1 ^9 h; d& A: l+ P) u
137. m_version   =   version;
     9 D* f7 L5 O/ q# v. O
138. 
     ; W$ {1 F/ z3 H- Q' L/ d
139. #define   NUMBEROFDEVICES 2
     ) Q; t$ G8 U+ K2 v. C
140. CString   devices[][2]   =   {
     " {5 I9 d+ p, [# f; \4 c
141. {UPNPPORTMAP1,   _T( "service ")},
     
142. {UPNPPORTMAP0,   _T( "service ")},
     - s- f# l( g# E6 F# N& y
143. {_T( "InternetGatewayDevice "),   _T( "device ")},
     
144. };
     4 i' l& b+ z) s4 O% @0 n
145. 
     
146. int   s   =   socket(AF_INET,   SOCK_DGRAM,   0);
     * c; H9 u4 J" z% r# Y2 v
147. u_long   lv   =   1;
     7 ?' J1 [4 N8 \2 x1 R" ]1 \- R
148. ioctlsocket(s,   FIONBIO,   &lv);
     0 ~; c. e' L2 N5 w8 |: D
149. 
     - G& ~0 @: o+ m
150. int   rlen   =   0;
     
151. for   (int   i=0;   rlen <=0   &&   i <500;   i++)   {
     8 Z6 R. A; B0 d0 M! |
152. if   (!(i%100))   {
     - ]. g+ v: K3 E# Y- p
153. for   (int   i=0;   i <NUMBEROFDEVICES;   i++)   {
     , t+ a9 F* S8 k5 p/ t7 v
154. m_name.Format(_T( "%s%s:%s:%d "),   URNPREFIX,   devices[i][1],   devices[i][0],   version);
     7 M: D- E4 b1 ^4 `, W
155. CString   request;
     " {- E2 X# K9 J$ N- 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. }
     8 N! Y# V7 r. k8 g
160. }
     6 T: ^: P, [, y! n5 l0 u
161. 
     
162. Sleep(10);
     ) ^- h. U  t1 Y0 F8 t. R  D* R  X8 C
163. 
     : y6 ]2 U  n! v" ~
164. char   buffer[10240];
     
165. rlen   =   recv(s,   buffer,   sizeof(buffer),   0);
     ) B( u& J" O( [+ G# V5 a
166. if   (rlen   <=   0)   continue;
     
167. closesocket(s);
     . _! r1 f' C$ t) X0 C" s# {+ Q( B
168. 
     " N% v2 _3 ]+ `  K1 P/ F
169. CString   response   =   CString(CStringA(buffer,   rlen));
     / ?8 Y+ g& m$ e1 r0 Z
170. CString   result;
     ! i- r; U( T5 I6 z
171. if   (!parseHTTPResponse(response,   result))   return   false;
     
172. 
     0 h8 u% z% {6 d  H# p
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);
     ( Q3 ]/ M0 [5 V7 \* [, J! @5 B+ B
175. if   (result.Find(m_name)   > =   0)   {
     . C, a/ m; K, l) @/ p$ [% d$ ~# W
176. for   (int   pos   =   0;;)   {
     
177. CString   line   =   result.Tokenize(_T( "\r\n "),   pos);
     ) ?8 E: x1 I/ k- R$ g
178. if   (line.IsEmpty())   return   false;
     
179. CString   name   =   line.Mid(0,   9);
     4 u- K) L' J; ^6 Y) Q- Z1 ^( Q2 d
180. name.MakeUpper();
     
181. if   (name   ==   _T( "LOCATION: "))   {
       g7 ]" p; R# M- J1 J% W$ y
182. line.Delete(0,   9);
     
183. m_description   =   line;
     
184. m_description.Trim();
     
185. return   GetDescription();
     
186. }
     
187. }
     
188. }
     
189. }
     
190. }
     
191. closesocket(s);
       @6 W+ y# C' `) `
192. 
     # X% m4 Q  s; v6 D* r6 U. M; a
193. return   false;
     
194. }
     ' V7 i* C# |/ U: Z! M. k

复制代码

天云上人

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


, f2 Y$ c" K/ I' y' }7 M///////////////////////////////////////////
//upnp的抽象接口类,只需调用这些接口即可实现端口映射功能.

. C* g. B3 j) g
#pragma once
#include <exception>

/ \* A# E# y: E& C7 ?4 S) B1 U+ p4 r
  enum TRISTATE{
2 \# G' o. W- I8 T' a: V        TRIS_FALSE,
        TRIS_UNKNOWN,
5 [4 h1 \! M8 O, p; X        TRIS_TRUE
* r0 R9 v( E, ^' W; @" _};
( a  Q* j5 ?  }. x  H0 o

enum UPNP_IMPLEMENTATION{
        UPNP_IMPL_WINDOWSERVICE = 0,
6 ?- S) L6 t, h        UPNP_IMPL_MINIUPNPLIB,
0 l+ M: G3 ]  X        UPNP_IMPL_NONE /*last*/
  ?9 ?6 q* O  J: _+ B4 S) P};
1 U) ]. g' v' E  \6 X- k7 n
& ]+ B5 A8 [6 i9 S3 p" r' k0 n4 n9 S
# E. C) \- q9 l" Z; {
7 G! J- B* Y( z2 R1 s# c
0 G) b/ m9 ?. E/ aclass CUPnPImpl
; l* e7 w9 r, o  L{
public:
        CUPnPImpl();
        virtual ~CUPnPImpl();
        struct UPnPError : std::exception {};
- Z+ b4 K2 r5 [' Z9 [, `        enum {
                UPNP_OK,
                UPNP_FAILED,
8 W4 E. }* S0 w9 w6 c) C% M                UPNP_TIMEOUT
$ H7 o2 N5 k# a: x) c        };
' H9 Z  n, u$ g" G  Q, y) Y2 b" S

8 f" ^3 l/ @: C9 R1 t, |$ n        virtual void        StartDiscovery(uint16 nTCPPort, uint16 nUDPPort, uint16 nTCPWebPort) = 0;
        virtual bool        CheckAndRefresh() = 0;
        virtual void        StopAsyncFind() = 0;
3 ~4 E; u, Z* Y7 u- [2 }! B        virtual void        DeletePorts() = 0;
        virtual bool        IsReady() = 0;
        virtual int                GetImplementationID() = 0;
       
, @. f: c5 ^9 t, L3 e3 z. O& z        void                        LateEnableWebServerPort(uint16 nPort); // Add Webserverport on already installed portmapping

  K5 s( D& e9 u+ C4 x9 y, n
3 [; U3 z5 s$ `% U; F5 w        void                        SetMessageOnResult(HWND hWindow, UINT nMessageID);
# ?4 {& B$ ~  f4 u7 R0 y/ {        TRISTATE                ArePortsForwarded() const                                                                { return m_bUPnPPortsForwarded; }
. V% [$ o) i- H" ~3 W: T( o        uint16                        GetUsedTCPPort()                                                                                { return m_nTCPPort; }
        uint16                        GetUsedUDPPort()                                                                                { return m_nUDPPort; }       
* y2 t/ K# x; {; K# h

( J( \! D% n  E! H// Implementation
7 j9 z  q0 S, d! D' Nprotected:
5 u, C7 X9 h# E        volatile TRISTATE        m_bUPnPPortsForwarded;
3 R  V+ E" R+ h% g$ H        void                                SendResultMessage();
  e5 @1 z) s, S        uint16                                m_nUDPPort;
        uint16                                m_nTCPPort;
        uint16                                m_nTCPWebPort;
/ S, A+ {# m) Y3 |        bool                                m_bCheckAndRefresh;


private:
        HWND        m_hResultMessageWindow;
2 M0 m* O$ a; g( K1 r        UINT        m_nResultMessageID;


};


// Dummy Implementation to be used when no other implementation is available
" I6 F8 }9 I* i6 D9 Mclass CUPnPImplNone: public CUPnPImpl
) O/ g; j8 K: M/ _8 N- u9 ^5 y{
public:
        virtual void        StartDiscovery(uint16, uint16, uint16)                                        { ASSERT( false ); }
        virtual bool        CheckAndRefresh()                                                                                { return false; }
, W. ]2 E$ f5 d: `$ L/ e        virtual void        StopAsyncFind()                                                                                        { }
1 a" V; J8 [% g+ T        virtual void        DeletePorts()                                                                                        { }
        virtual bool        IsReady()                                                                                                { return false; }
/ D) e% Y1 Q9 y8 f        virtual int                GetImplementationID()                                                                        { return UPNP_IMPL_NONE; }
: m' V5 b7 U: j* t+ Y};
+ Q( }7 G+ M- l/ h# H
2 a5 }0 f2 }7 T: R% H
& z: P6 T+ E' j+ J( b/////////////////////////////////////
//下面是使用windows操作系统自带的UPNP功能的子类
( Y$ X5 y- A+ m: l7 Q: [
: c& W3 K0 }* p" `) \" g
#pragma once
$ G. I/ C, Y( _/ ~" E7 V#pragma warning( disable: 4355 )
2 q0 w2 R4 g) L5 G: p

/ p' {6 j( g0 X/ O  s9 a, X#include "UPnPImpl.h"
#include <upnp.h>
#include <iphlpapi.h>
& X* ?6 W2 i  N#include <comdef.h>
#include <winsvc.h>
2 c! h. w8 I- F4 ^0 `7 \

#include <vector>
' K+ o8 b3 @( a( ~/ ^$ T#include <exception>
7 x: @: a5 d0 f" K; |) m3 p; r/ S#include <functional>
2 P! U1 M% K/ u9 n* I
+ i8 R, G4 h( _! v" O


typedef _com_ptr_t<_com_IIID<IUPnPDeviceFinder,&IID_IUPnPDeviceFinder> >        FinderPointer;
typedef _com_ptr_t<_com_IIID<IUPnPDevice,&IID_IUPnPDevice>        >                                DevicePointer;
' C3 a6 I& D& e. H2 b6 A$ Jtypedef _com_ptr_t<_com_IIID<IUPnPService,&IID_IUPnPService> >                                ServicePointer;
typedef        _com_ptr_t<_com_IIID<IUPnPDeviceFinderCallback,&IID_IUPnPDeviceFinderCallback> > DeviceFinderCallback;
7 [% \* E8 |1 x+ W8 \typedef        _com_ptr_t<_com_IIID<IUPnPServiceCallback,&IID_IUPnPServiceCallback> >                        ServiceCallback;
typedef _com_ptr_t<_com_IIID<IEnumUnknown,&IID_IEnumUnknown> >                                EnumUnknownPtr;
0 H8 A5 |5 t& `7 I4 ftypedef _com_ptr_t<_com_IIID<IUnknown,&IID_IUnknown> >                                                UnknownPtr;
' V# a; S& P) H' Q
  U8 W% j7 v9 D. g' J9 F
5 m% Y- U* j% i8 N, y$ ~typedef DWORD (WINAPI* TGetBestInterface) (
* T4 Z1 `( t* [) R$ i  IPAddr dwDestAddr,
& C* W3 |7 Y9 F  PDWORD pdwBestIfIndex
. `0 y1 L6 a% M$ v4 N4 }3 Q);


( S( ~; b6 S; T; B( v3 a7 rtypedef DWORD (WINAPI* TGetIpAddrTable) (
  PMIB_IPADDRTABLE pIpAddrTable,
1 {  j+ A7 G6 I$ M. s9 y  PULONG pdwSize,
( f+ x! H$ G# @; o6 t  BOOL bOrder
, h: D6 \0 X/ B( P);

3 b) D* O! o( ~' b# R2 ~
typedef DWORD (WINAPI* TGetIfEntry) (
( ]9 L0 {, J; n# L  PMIB_IFROW pIfRow
+ a$ R; u) H) I# Y);
) B8 @+ J: q9 X. J; ^
! W7 W7 u' [2 V2 ]8 m
7 q8 r7 y) \7 J" qCString translateUPnPResult(HRESULT hr);
' e% Z9 B% t& v: M% z( A$ AHRESULT UPnPMessage(HRESULT hr);


1 M9 U* z0 }% m3 K5 Iclass CUPnPImplWinServ: public CUPnPImpl
{
        friend class CDeviceFinderCallback;
        friend class CServiceCallback;
* r- K" p0 k* [+ T2 x// Construction
1 x" p9 E' i1 e' q6 u, n% Upublic:
        virtual ~CUPnPImplWinServ();
        CUPnPImplWinServ();


0 i1 G: B- X: M6 t. V& A6 }* {        virtual void        StartDiscovery(uint16 nTCPPort, uint16 nUDPPort, uint16 nTCPWebPort)                { StartDiscovery(nTCPPort, nUDPPort, nTCPWebPort, false); }
: X9 a' s3 j" ?# u$ ?        virtual void        StopAsyncFind();
        virtual void        DeletePorts();
+ X" l9 {! e( k7 e        virtual bool        IsReady();
3 m  y6 ~: D. |) n        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)
$ F" d, c+ k' F. T  |( w        // the windows side of the implementation tends to get bugged untill reboot anyway. Still might get added later
5 h* i  {% O& e* i. y        virtual bool        CheckAndRefresh()                                                                                { return false; };
1 ?! o" J$ B; H, s. x! C, X
+ a& e9 W% Z9 H0 y8 a4 w
7 g* F+ {- I: z- n/ X- a" a4 uprotected:
        void        StartDiscovery(uint16 nTCPPort, uint16 nUDPPort, uint16 nTCPWebPort, bool bSecondTry);
        void        AddDevice(DevicePointer pDevice, bool bAddChilds, int nLevel = 0);
        void        RemoveDevice(CComBSTR bsUDN);
7 s; {" X8 z/ d9 R# |        bool        OnSearchComplete();
        void        Init();
" m/ D2 Q% ?6 u
5 ]# D- H1 j, c. |( n
        inline bool IsAsyncFindRunning()
1 c+ ?* P- s. B        {
1 j$ E: R2 D3 n2 h- V- o' L2 s  D                if ( m_pDeviceFinder != NULL && m_bAsyncFindRunning && GetTickCount() - m_tLastEvent > 10000 )
                {
                        m_pDeviceFinder->CancelAsyncFind( m_nAsyncFindHandle );
+ i1 O" E+ s* g8 |                        m_bAsyncFindRunning = false;
                }
                MSG msg;
                while ( PeekMessage( &msg, NULL, 0, 0, PM_REMOVE ) )
! a1 a) o; z& S7 v5 ^8 q( P                {
                        TranslateMessage( &msg );
7 Y' Z# \1 R3 m8 ^, V0 ]% d/ I# I                        DispatchMessage( &msg );
; K- ^* c4 M8 X: x                }
1 m% d5 E0 e2 X" c# {) M6 W                return m_bAsyncFindRunning;
) E* k7 W- X- d2 g$ Q        }
& d5 Q4 N6 J) T- E& G& c( b
( S; K4 @# \3 `5 `! D- t" q; S2 M
6 [; Q* P4 C7 p. U        TRISTATE                        m_bUPnPDeviceConnected;


// Implementation
6 x8 d) J: ^6 T$ T& _        // API functions
9 J* A& M" y; \        SC_HANDLE (WINAPI *m_pfnOpenSCManager)(LPCTSTR, LPCTSTR, DWORD);
# `8 E; J. |' M        SC_HANDLE (WINAPI *m_pfnOpenService)(SC_HANDLE, LPCTSTR, DWORD);
! ~: F3 {2 {( W# h: J% w, ]  Q        BOOL (WINAPI *m_pfnQueryServiceStatusEx)(SC_HANDLE, SC_STATUS_TYPE, LPBYTE, DWORD, LPDWORD);
8 g' B9 a0 `' ]( _8 t5 L0 M        BOOL (WINAPI *m_pfnCloseServiceHandle)(SC_HANDLE);
& l1 f, ?3 K+ F: Y: V7 R. t        BOOL (WINAPI *m_pfnStartService)(SC_HANDLE, DWORD, LPCTSTR*);
; v( k3 `8 @0 w% e# q        BOOL (WINAPI *m_pfnControlService)(SC_HANDLE, DWORD, LPSERVICE_STATUS);
* w) f  K7 [1 X
5 M$ O+ o1 ]: C( M
        TGetBestInterface                m_pfGetBestInterface;
! w9 H- }, S$ q" z) M        TGetIpAddrTable                        m_pfGetIpAddrTable;
' x* Y/ L' ]; G6 ]7 Q        TGetIfEntry                                m_pfGetIfEntry;

( ^$ J: G  x1 j$ R7 y
% X' @, B) B" N, T        static FinderPointer CreateFinderInstance();
3 R0 l9 Y. f7 \+ h$ r; b        struct FindDevice : std::unary_function< DevicePointer, bool >
* _8 g$ X+ ]- e. D        {
& K/ u, w. o  B6 z( R7 F                FindDevice(const CComBSTR& udn) : m_udn( udn ) {}
3 H6 H+ R4 ~% X                result_type operator()(argument_type device) const
                {
                        CComBSTR deviceName;
0 h* ^: R5 _3 r/ @! |- b                        HRESULT hr = device->get_UniqueDeviceName( &deviceName );


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

% m" T# n! T! A. z8 }4 \; K6 ~
& p/ A% u2 C2 Q) o                        return wcscmp( deviceName.m_str, m_udn ) == 0;
" r8 b& \. Y/ y                }
                CComBSTR m_udn;
, K% \% z# F: G8 K2 c- C        };
2 ]/ h5 l9 t% D3 z       
- ^% E$ y! ~1 v( d3 p4 \* o, U        void        ProcessAsyncFind(CComBSTR bsSearchType);
        HRESULT        GetDeviceServices(DevicePointer pDevice);
        void        StartPortMapping();
        HRESULT        MapPort(const ServicePointer& service);
        void        DeleteExistingPortMappings(ServicePointer pService);
6 r3 L" o# [( c- M        void        CreatePortMappings(ServicePointer pService);
        HRESULT SaveServices(EnumUnknownPtr pEU, const LONG nTotalItems);
        HRESULT InvokeAction(ServicePointer pService, CComBSTR action,
                LPCTSTR pszInArgString, CString& strResult);
        void        StopUPnPService();
$ X! \# j7 [. {1 F) g
- j4 }4 r) ]8 y: R
$ q% s' I6 |' F9 W' i6 y$ G        // 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);
: t- a; W% S% J8 F1 s* V8 x        void        DestroyVars(const INT_PTR nCount, VARIANT*** pppVars);
        HRESULT GetSafeArrayBounds(SAFEARRAY* psa, LONG* pLBound, LONG* pUBound);
" Z/ W4 q* L2 [5 c& o2 Y, _        HRESULT GetVariantElement(SAFEARRAY* psa, LONG pos, VARIANT* pvar);
        CString        GetLocalRoutableIP(ServicePointer pService);


// Private members
: c6 w8 w4 v: c- n7 R4 B3 q& W5 kprivate:
" |+ k- _/ ]( Q/ D/ P! u# _2 s  l        DWORD        m_tLastEvent;        // When the last event was received?
  e- k# w8 R* r        std::vector< DevicePointer >  m_pDevices;
        std::vector< ServicePointer > m_pServices;
        FinderPointer                        m_pDeviceFinder;
        DeviceFinderCallback        m_pDeviceFinderCallback;
' @1 v& B9 b6 i+ ^: J' J$ i/ ]* v        ServiceCallback                        m_pServiceCallback;

* ^: ~4 @5 g  v! t+ M" \8 Q
* Q9 k3 f/ }2 h8 m        LONG        m_nAsyncFindHandle;
6 o2 j/ {; X- c2 d; y* ]6 H* T: d        bool        m_bCOM;
        bool        m_bPortIsFree;
        CString m_sLocalIP;
4 E) _/ R1 P5 n        CString m_sExternalIP;
        bool        m_bADSL;                // Is the device ADSL?
        bool        m_ADSLFailed;        // Did port mapping failed for the ADSL device?
        bool        m_bInited;
        bool        m_bAsyncFindRunning;
% J$ \4 z8 V1 U" m        HMODULE m_hADVAPI32_DLL;
        HMODULE        m_hIPHLPAPI_DLL;
        bool        m_bSecondTry;
        bool        m_bServiceStartedByEmule;
3 p" i, Y! ~. E5 r* M! s+ ?1 ~3 H        bool        m_bDisableWANIPSetup;
0 l7 `4 n( b) a/ w/ m        bool        m_bDisableWANPPPSetup;
7 ^. z7 r7 B2 H* \6 N8 m8 {
7 H& {( K+ h8 l" ]" _% p( T! [
};
  [! l# \& v4 ?3 x
: t. }) z1 F5 p! n; T5 _
& i) e' u- i; c( d) H8 @7 f// DeviceFinder Callback
class CDeviceFinderCallback
* \; _3 b* ^8 W        : public IUPnPDeviceFinderCallback
( t, q( b3 ]  N% R& c{
5 `- U8 {3 v* s# ?public:
7 |' C4 c& \( A/ X        CDeviceFinderCallback(CUPnPImplWinServ& instance)
' O! ^) `$ u1 v4 G7 D/ v                : m_instance( instance )
. V( v" h. S. \  [        { m_lRefCount = 0; }


" z: B# L( l9 t- h9 [. P   STDMETHODIMP QueryInterface(REFIID iid, LPVOID* ppvObject);
6 {6 M- a7 S+ t- |, ], P* `' y   STDMETHODIMP_(ULONG) AddRef();
   STDMETHODIMP_(ULONG) Release();

( I# d) m1 i; ]' V& j  }9 X
1 [: y7 o9 P7 z// implementation
private:
. O. T, z1 \% n7 p        HRESULT __stdcall DeviceAdded(LONG nFindData, IUPnPDevice* pDevice);
        HRESULT __stdcall DeviceRemoved(LONG nFindData, BSTR bsUDN);
/ y! h% q# E3 u8 P        HRESULT __stdcall SearchComplete(LONG nFindData);

0 e) ~, v- R/ d6 a$ A& B5 e" Q
2 e0 F2 V$ p+ Y: }0 gprivate:
        CUPnPImplWinServ& m_instance;
6 k" f0 Q4 p$ f# M5 P$ R        LONG m_lRefCount;
) G; r6 \) O$ f2 U( r};

; K: B# h1 R+ T0 o8 m% ~5 ?% F- @
// Service Callback
class CServiceCallback
: g  Z8 B+ b# B1 A- g        : public IUPnPServiceCallback
{
public:
* D  _' B7 l6 j        CServiceCallback(CUPnPImplWinServ& instance)
                : m_instance( instance )
3 C; O7 N3 d5 `  e; ?        { m_lRefCount = 0; }
; U1 x3 _, a' }6 E9 r/ g; N( |$ n   
   STDMETHODIMP QueryInterface(REFIID iid, LPVOID* ppvObject);
   STDMETHODIMP_(ULONG) AddRef();
   STDMETHODIMP_(ULONG) Release();

9 Y" H1 Y0 Y0 i- S
// implementation
private:
, J; g2 c3 v" \6 o) S" o        HRESULT __stdcall StateVariableChanged(IUPnPService* pService, LPCWSTR pszStateVarName, VARIANT varValue);
        HRESULT __stdcall ServiceInstanceDied(IUPnPService* pService);
6 k# J% v- o, A, E1 i

% P, a, |, [7 I5 T& y5 \$ \private:
$ K- {) N& d% I1 }% U  c: Z        CUPnPImplWinServ& m_instance;
8 C; z% F3 k: L. |        LONG m_lRefCount;
};
7 e5 \4 z6 X4 q& x+ I' n' k

/ `$ g7 k5 d# \7 @% B/////////////////////////////////////////////////

7 _5 i* E% g  T7 T. X
使用时只需要使用抽象类的接口。
9 p8 `" ]: e5 L: k9 ~. S% U3 I" CCUPnPImpl::SetMessageOnResult设置需要接受UPNP端口映射是否成功的窗口句柄和消息ID.
2 }8 S2 a3 L+ _1 c( b& KCUPnPImpl::StartDiscovery将开启一个异步设备查找并进行端口映射,其参数为需要映射的内网端口.
CUPnPImpl::StopAsyncFind停止设备查找.
CUPnPImpl::DeletePorts删除端口映射.