UPnP

天云上人

/*uPnP.h*/

1. 
   
2. #ifndef   MYUPNP_H_
   , m, S# x, q3 Q  Y% g  r& ]; p
3. 
   ) {  s; d* {; H! l3 W6 I; g
4. #pragma   once
   0 W& @3 z( F! k4 d% `
5. 
   
6. typedef   unsigned   long   ulong;
   
7. 
   : w+ q4 J1 g5 g' L
8. class   MyUPnP
   , q7 s0 G) y; Z& W  Z
9. {
   
10. public:
    5 q2 B0 H0 F0 k( ]
11. typedef   enum{
    
12. UNAT_OK, //   Successfull
    2 V) f& m& f, Y& L$ b: B" x' m
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
    7 S# u5 o# \: X0 ]) r# h$ y: g* w/ r1 F
16. UNAT_NOT_IN_LAN //   Error,   you   aren 't   in   a   LAN   ->   no   router   or   firewall
    * {6 i5 _  X( {1 t9 N( z
17. }   UPNPNAT_RETURN;
    3 k  k! a  ~8 ^( |# d- V3 H
18. 
    
19. typedef   enum{
    ( ]! G: v  H. m7 x
20. UNAT_TCP, //   TCP   Protocol
    0 z/ u: {# o/ [
21. UNAT_UDP //   UDP   Protocol
    0 v+ y4 c. x# q; o+ S
22. }   UPNPNAT_PROTOCOL;
    
23. 
    
24. typedef   struct{
    
25. WORD   internalPort; //   Port   mapping   internal   port
    
26. WORD   externalPort; //   Port   mapping   external   port
    # U0 b& V: G# s1 Z" b
27. UPNPNAT_PROTOCOL   protocol; //   Protocol->   TCP   (UPNPNAT_PROTOCOL:UNAT_TCP)   ||   UDP   (UPNPNAT_PROTOCOL:UNAT_UDP)
      S( s" M2 \6 r1 F
28. CString   description; //   Port   mapping   description
    
29. }   UPNPNAT_MAPPING;
    " p1 @/ u; y# M# U
30. 
    , C8 P2 Y/ R# u  @+ q
31. MyUPnP();
    ' Z% y3 i2 V9 W" i8 G5 m
32. ~MyUPnP();
    
33. 
    : U" U, I  Z) A! K
34. UPNPNAT_RETURN   AddNATPortMapping(UPNPNAT_MAPPING   *mapping,   bool   tryRandom   =   false);
    & J! u+ Y( \0 e$ ]4 w
35. UPNPNAT_RETURN   RemoveNATPortMapping(UPNPNAT_MAPPING   mapping,   bool   removeFromList   =   true);
    9 J# @& n- G5 [
36. void   clearNATPortMapping();
    # w7 u6 X: x! h1 a4 g) P2 {& Q
37. 
    
38. CString GetLastError();
    ; g0 l8 \- |" e' p: f9 [
39. CString GetLocalIPStr();
    ( E6 i% a0 ^- J' i) u6 _* T! Y3 u, @
40. WORD GetLocalIP();
    
41. bool IsLANIP(WORD   nIP);
    ' m  T$ T; @1 d# c% W" l
42. 
    
43. protected:
    
44. void InitLocalIP();
    ( g" s$ Y8 T0 J" J
45. void SetLastError(CString   error);
    + `3 q5 C% [' m1 J: F
46. 
    
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);
    
50. 
    
51. bool isComplete()   const   {   return   !m_controlurl.IsEmpty();   }
    
52. 
    
53. bool Search(int   version=1);
    3 m% l. B/ r8 w1 ^
54. bool GetDescription();
    $ U8 ~9 r. `6 ]" T
55. CString GetProperty(const   CString&   name,   CString&   response);
    , a$ {/ g" t$ B
56. bool InvokeCommand(const   CString&   name,   const   CString&   args);
    
57. 
    
58. bool Valid()const{return   (!m_name.IsEmpty()&&!m_description.IsEmpty());}
    " A  P( G/ Q4 q6 u6 i+ C7 J1 j
59. bool InternalSearch(int   version);
    
60. CString m_devicename;
    
61. CString m_name;
    / o1 i5 \, I5 m1 F9 ]
62. CString m_description;
    
63. CString m_baseurl;
    0 D) r( V0 N. h
64. CString m_controlurl;
    5 G5 c/ p; V, r! m4 y! o
65. CString m_friendlyname;
    
66. CString m_modelname;
    / w8 @. X6 P7 x2 x
67. int m_version;
    9 f7 M6 `- Y0 C( V- P1 v) c( O) g
68. 
    ! s0 W+ h: y4 [, y% O3 |
69. private:
    + L5 t* f# [) U* X1 W8 {. a
70. CList <UPNPNAT_MAPPING,   UPNPNAT_MAPPING>   m_Mappings;
    ! o6 g* d: Q6 T4 R8 a
71. 
    
72. CString m_slocalIP;
    
73. CString m_slastError;
    
74. WORD m_uLocalIP;
    9 F7 G& f5 M! T3 G
75. 
    
76. bool isSearched;
    
77. };
    & ?7 f/ H2 }" O. x; o& S* L
78. #endif

复制代码

天云上人

/*UPnP.cpp*/

1. 
   
2. #include   "stdafx.h "
   * _5 D2 l! S8 y
3. 
   
4. #include   "upnp.h "
   
5. 
   9 B2 _! u1 f% |; u
6. #define   UPNPPORTMAP0       _T( "WANIPConnection ")
   
7. #define   UPNPPORTMAP1       _T( "WANPPPConnection ")
   
8. #define   UPNPGETEXTERNALIP   _T( "GetExternalIPAddress "),_T( "NewExternalIPAddress ")
   8 z0 I6 P3 w6 H4 b  Z
9. #define   UPNPADDPORTMAP   _T( "AddPortMapping ")
   . w( [( c! N( W' ?
10. #define   UPNPDELPORTMAP   _T( "DeletePortMapping ")
    
11. 
    
12. static   const   ulong UPNPADDR   =   0xFAFFFFEF;
      I  M7 P3 G+ [# _$ R" r+ Q. G
13. static   const   int UPNPPORT   =   1900;
    9 Z" l- v$ E' W& M9 k  f+ J# x
14. static   const   CString URNPREFIX   =   _T( "urn:schemas-upnp-org: ");
    
15. 
    ( H/ g+ I$ w' B: \- \
16. const   CString   getString(int   i)
    2 q4 M( n1 c7 d) h8 X
17. {
    
18. CString   s;
    
19. 
    
20. s.Format(_T( "%d "),   i);
    
21. 
    
22. return   s;
    4 z7 G) `+ P) U  i  H  D
23. }
    
24. 
    # Z6 c7 [( q; \9 V! R
25. const   CString   GetArgString(const   CString&   name,   const   CString&   value)
    
26. {
    2 q7 s, a1 `2 Q' A* [; }9 W
27. return   _T( " < ")   +   name   +   _T( "> ")   +   value   +   _T( " </ ")   +   name   +   _T( "> ");
    - f0 v1 `6 c) s/ a& C3 N
28. }
    
29. 
    
30. const   CString   GetArgString(const   CString&   name,   int   value)
    $ r- C$ r* @- N
31. {
    
32. return   _T( " < ")   +   name   +   _T( "> ")   +   getString(value)   +   _T( " </ ")   +   name   +   _T( "> ");
    & c! z  H+ f2 }3 s0 `
33. }
    : t8 K/ X+ w8 O; ^
34. 
    . I/ p- {5 |2 J2 V, W( I" ~1 F# ~
35. bool   SOAP_action(CString   addr,   uint16   port,   const   CString   request,   CString   &response)
    ( x8 T, m' e% `
36. {
    
37. char   buffer[10240];
    : h5 n$ v  m7 Z* i0 `; _
38. 
    ) f# `8 G  g4 y& ?  w
39. const   CStringA   sa(request);
    
40. int   length   =   sa.GetLength();
    2 B' L% ?0 m: L1 Z* f; Z
41. strcpy(buffer,   (const   char*)sa);
    . o+ c) H0 n3 V/ i8 R- T
42. 
    1 J+ J' P; D0 G4 ?
43. uint32   ip   =   inet_addr(CStringA(addr));
    + r2 ~7 E( f* s% S, G. e
44. struct   sockaddr_in   sockaddr;
    7 v1 u& Q' K, @# n2 H
45. memset(&sockaddr,   0,   sizeof(sockaddr));
    . B4 |( I+ g  K
46. sockaddr.sin_family   =   AF_INET;
    
47. sockaddr.sin_port   =   htons(port);
    
48. sockaddr.sin_addr.S_un.S_addr   =   ip;
      y2 K4 W7 u! T- z8 `8 l  W
49. int   s   =   socket(AF_INET,   SOCK_STREAM,   0);
    
50. u_long   lv   =   1;
    5 [, R" T. f; P9 A
51. ioctlsocket(s,   FIONBIO,   &lv);
    
52. connect(s,   (struct   sockaddr   *)&sockaddr,   sizeof(sockaddr));
      e5 L0 ]4 A. J# c
53. Sleep(20);
    $ X' z/ N1 _( i/ ^$ T) d- r/ q
54. int   n   =   send(s,   buffer,   length,   0);
    - A+ Q9 W  h/ ^9 X* }0 t( a- X
55. Sleep(100);
    
56. int   rlen   =   recv(s,   buffer,   sizeof(buffer),   0);
    
57. closesocket(s);
    - }6 D6 J0 D- X' f
58. if   (rlen   ==   SOCKET_ERROR)   return   false;
    ) q1 l/ n# G+ \, T7 [9 o9 n/ ]4 z
59. if   (!rlen)   return   false;
    
60. 
    . u8 F- ]9 [" t' ]( _  L4 K
61. response   =   CString(CStringA(buffer,   rlen));
    4 ^! I' f/ _$ U- B# M2 u
62. 
    
63. return   true;
    4 G9 F& q9 E% d  ]* Z
64. }
    
65. 
    
66. int   SSDP_sendRequest(int   s,   uint32   ip,   uint16   port,   const   CString&   request)
    " W( F, @0 J( j
67. {
    
68. char   buffer[10240];
    $ O  O" j! X% f+ @9 O
69. 
    ! g# ?! x; a# Z5 |
70. const   CStringA   sa(request);
    
71. int   length   =   sa.GetLength();
    
72. strcpy(buffer,   (const   char*)sa);
    ! ?  X- I; M/ a1 _% u
73. 
    
74. struct   sockaddr_in   sockaddr;
    
75. memset(&sockaddr,   0,   sizeof(sockaddr));
    , P  Y- e3 B# V& K
76. sockaddr.sin_family   =   AF_INET;
    : f" P) X: N# S2 i$ X
77. sockaddr.sin_port   =   htons(port);
    
78. sockaddr.sin_addr.S_un.S_addr   =   ip;
    
79. 
      `0 l8 F3 G* b( |
80. return   sendto(s,   buffer,   length,   0,   (struct   sockaddr   *)&sockaddr,   sizeof(sockaddr));
      Y0 m: Z* x- W  _) i" g* u- G
81. }
    & L$ v! w% H* S  f
82. 
    
83. bool   parseHTTPResponse(const   CString&   response,   CString&   result)
    
84. {
    1 g* E% Z5 B0 W& N
85. int   pos   =   0;
    
86. 
    : @* A- K  F* B, h1 S# j
87. CString   status   =   response.Tokenize(_T( "\r\n "),   pos);
    & r+ o( T, f* g  s* H$ k" F
88. 
    5 Z# ~. Z- p4 m- z* `& l
89. result   =   response;
    1 U0 A( q* J3 B2 m+ w7 U
90. result.Delete(0,   pos);
    
91. 
    
92. pos   =   0;
    % \* ?% o+ A" z  u& o6 v
93. status.Tokenize(_T( "   "),   pos);
    
94. status   =   status.Tokenize(_T( "   "),   pos);
    $ S8 R  d) d- W4 W& C8 z
95. if   (status.IsEmpty()   ||   status[0]!= '2 ')   return   false;
    
96. return   true;
    
97. }
    
98. 
    ; u5 w' M3 K3 }" A- j* U) Y, W
99. const   CString   getProperty(const   CString&   all,   const   CString&   name)
    
100. {
     
101. CString   startTag   =   ' < '   +   name   +   '> ';
     
102. CString   endTag   =   _T( " </ ")   +   name   +   '> ';
     - f6 X( H. J+ ^. x; j
103. CString   property;
     
104. 
     
105. int   posStart   =   all.Find(startTag);
     
106. if   (posStart <0)   return   CString();
     
107. 
     + A# M! E! M# n
108. int   posEnd   =   all.Find(endTag,   posStart);
     $ L# v( ]  r4 u8 ?
109. if   (posStart> =posEnd)   return   CString();
     / t8 x' t8 x4 k! M9 y5 A8 g) h1 S
110. 
     
111. return   all.Mid(posStart   +   startTag.GetLength(),   posEnd   -   posStart   -   startTag.GetLength());
     
112. }
     , W# z$ I" I# l1 v1 g7 g3 j2 t( T
113. 
     ( m: n! i" j! Z! M, C
114. MyUPnP::MyUPnP()
     9 O* A; ]. y" C& m
115. :   m_version(1)
     ; F! C0 \- e+ X- @4 |
116. {
     4 [& K7 ^4 I9 [. P- G3 H' {7 B
117. m_uLocalIP   =   0;
     $ O/ a, q9 w3 l6 O4 s) o, n
118. isSearched   =   false;
     ; j  I) [" x' H" A. h. B- p  A
119. }
     
120. 
     
121. MyUPnP::~MyUPnP()
     
122. {
     0 L8 E% `5 M# x/ a& ?- W2 m% w6 N
123. UPNPNAT_MAPPING   search;
     
124. POSITION   pos   =   m_Mappings.GetHeadPosition();
     9 y! S8 _. ?. D- j/ ~4 A+ j* Z
125. while(pos){
     ) m+ G* B! ?5 \# U' B
126. search   =   m_Mappings.GetNext(pos);
     / ^. U% W: \3 Y" a( U  f# K0 [6 d
127. RemoveNATPortMapping(search,   false);
     / J$ S8 l8 I4 o& z
128. }
     ) F6 g/ p$ [3 `5 H, z# o$ t5 n
129. 
     8 W% _$ \% Y+ d8 g6 k$ B
130. m_Mappings.RemoveAll();
     ; Q6 o9 z! `, S8 @
131. }
     
132. 
     
133. 
     8 x1 B6 I. a  Y8 T
134. bool   MyUPnP::InternalSearch(int   version)
     
135. {
     
136. if(version <=0)version   =   1;
     
137. m_version   =   version;
     
138. 
     
139. #define   NUMBEROFDEVICES 2
     0 f) l/ s: G% v. T% Y% z# ?
140. CString   devices[][2]   =   {
     
141. {UPNPPORTMAP1,   _T( "service ")},
     
142. {UPNPPORTMAP0,   _T( "service ")},
     7 w, e; ?: Q3 E: u- e( _6 ]' O% @
143. {_T( "InternetGatewayDevice "),   _T( "device ")},
     3 j# A) w+ f: }. ?* j! N" U. y
144. };
     0 H- {* [6 h1 e- q* r* i/ `
145. 
     
146. int   s   =   socket(AF_INET,   SOCK_DGRAM,   0);
     4 X/ ?& E+ C6 I
147. u_long   lv   =   1;
     
148. ioctlsocket(s,   FIONBIO,   &lv);
     
149. 
     0 {, _: `* y- U
150. int   rlen   =   0;
     & `6 O4 ?- O; P( |& {. \
151. for   (int   i=0;   rlen <=0   &&   i <500;   i++)   {
     
152. if   (!(i%100))   {
     6 O' b9 h* g8 B$ }& W" l1 P: J! N0 j
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;
     8 ~9 y7 Z+ b) u( X& `7 D# c% T# A
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. }
     8 P5 v1 I5 R" B6 W2 t
161. 
     
162. Sleep(10);
     3 W6 u' M# g' q4 [8 a  r
163. 
     
164. char   buffer[10240];
     9 ~( ~# X8 ^3 \! j1 S8 n
165. rlen   =   recv(s,   buffer,   sizeof(buffer),   0);
     
166. if   (rlen   <=   0)   continue;
     
167. closesocket(s);
     : c& e; x( b* M( x- c3 ^
168. 
     / E/ g$ T5 I% O: Z& t% E
169. CString   response   =   CString(CStringA(buffer,   rlen));
     ; F4 r3 w# U# a  a! p
170. CString   result;
     2 X  j0 L8 v- B; W2 A
171. if   (!parseHTTPResponse(response,   result))   return   false;
       A; z7 H9 @6 q1 Z
172. 
     + O% W7 o# D: u9 Z: g
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)   {
     3 A. G" ]( n: B3 k6 d7 l
176. for   (int   pos   =   0;;)   {
     * J6 m1 B/ c: i  A- |
177. CString   line   =   result.Tokenize(_T( "\r\n "),   pos);
     
178. if   (line.IsEmpty())   return   false;
     
179. CString   name   =   line.Mid(0,   9);
     , Z, Y3 {! M& t- U) p4 H
180. name.MakeUpper();
     
181. if   (name   ==   _T( "LOCATION: "))   {
     ) Z8 C3 s; O" _. A, T# V
182. line.Delete(0,   9);
     
183. m_description   =   line;
     
184. m_description.Trim();
     6 O. s$ ?8 `( @7 M
185. return   GetDescription();
     # Y* W$ ]+ P3 W- O2 V6 S3 d
186. }
     # u0 d  c$ W6 d+ t) z
187. }
     8 p* q* P5 ]$ r0 F
188. }
     
189. }
     1 \" H2 K9 r4 u- I$ {0 r) u
190. }
     , J2 j/ I! _5 ?+ R& T8 P; S1 I
191. closesocket(s);
     - v& X& k; B* m
192. 
     3 F& N8 J) A) m; p8 ]  V0 ?* n; @, N
193. return   false;
     
194. }
     ) ~/ R! n$ T* u  z( o2 R

复制代码

天云上人

以下有关upnp的接口来自emule，
) Z+ ^6 r! I, e% A8 L

///////////////////////////////////////////
//upnp的抽象接口类,只需调用这些接口即可实现端口映射功能.


4 `% ]; Z, `2 ?% w" d8 D#pragma once
#include <exception>

' }: |( ?/ D5 q7 B& b5 {' _/ Q8 z
  enum TRISTATE{
0 f# C. F# [6 k- |0 t        TRIS_FALSE,
% e- d7 O2 y- G5 @% }; _2 p: p% M        TRIS_UNKNOWN,
        TRIS_TRUE
( g3 I- n- F: M& O9 U+ }3 h};

6 V! J+ w2 L9 D5 F1 [8 w
enum UPNP_IMPLEMENTATION{
6 y+ c/ _- m5 [4 Y0 b- `# ~9 N        UPNP_IMPL_WINDOWSERVICE = 0,
        UPNP_IMPL_MINIUPNPLIB,
        UPNP_IMPL_NONE /*last*/
};


! e6 E; g$ y8 c9 V* v. Y# n4 v

8 G- a# z8 m3 mclass CUPnPImpl
% J5 l- ]3 q8 W4 e{
1 F0 }% y( z# s: Upublic:
        CUPnPImpl();
        virtual ~CUPnPImpl();
        struct UPnPError : std::exception {};
        enum {
                UPNP_OK,
                UPNP_FAILED,
& Z- H! b6 ]2 |5 f                UPNP_TIMEOUT
0 V& Q, z3 H) r* ]4 F        };
+ n: y$ I" j! {2 C; \4 Q
8 [- H( h& R  i+ k/ X9 x7 @
        virtual void        StartDiscovery(uint16 nTCPPort, uint16 nUDPPort, uint16 nTCPWebPort) = 0;
( b7 t* Z" Q1 i# y  F( |( q        virtual bool        CheckAndRefresh() = 0;
% D  @9 q" \& H* B        virtual void        StopAsyncFind() = 0;
8 G7 K; {# v/ z3 s& @& n& X        virtual void        DeletePorts() = 0;
  F+ ?8 t( Z6 |        virtual bool        IsReady() = 0;
* i2 H( J: O  a9 H6 ]/ n        virtual int                GetImplementationID() = 0;
       
        void                        LateEnableWebServerPort(uint16 nPort); // Add Webserverport on already installed portmapping


        void                        SetMessageOnResult(HWND hWindow, UINT nMessageID);
0 ^% [( l0 f. K8 e        TRISTATE                ArePortsForwarded() const                                                                { return m_bUPnPPortsForwarded; }
        uint16                        GetUsedTCPPort()                                                                                { return m_nTCPPort; }
2 V( R) i/ q6 r" g1 y        uint16                        GetUsedUDPPort()                                                                                { return m_nUDPPort; }       
' l' D6 b- C9 d/ U" H' m1 X; y
  U7 Z) O* `" z/ t
; o2 @. w3 ~7 ~+ r6 S% ]// Implementation
protected:
4 ?* a0 V. ]3 v7 P% G; K" e/ d        volatile TRISTATE        m_bUPnPPortsForwarded;
1 H5 D* Y+ }# }8 t- {9 b# v        void                                SendResultMessage();
        uint16                                m_nUDPPort;
* w/ Q% l! l0 y9 c  d+ h" n1 r        uint16                                m_nTCPPort;
' S- u7 C9 e4 Y, {  w        uint16                                m_nTCPWebPort;
        bool                                m_bCheckAndRefresh;

1 H' p! H6 E2 [
5 s: ]) m. V# O9 |9 Sprivate:
4 M1 h+ {- j- i7 u: h        HWND        m_hResultMessageWindow;
5 ^9 u9 w! }8 J4 n# R! O! M( e7 }        UINT        m_nResultMessageID;

6 l: t4 f* e0 L
3 s: _+ F5 l$ N" @* }0 s};
7 z& w# T5 _4 L/ S% P: `
0 n+ d2 E5 I+ j2 B/ \. l* l) @0 P
2 H( A; e( c6 s3 \0 C1 X// Dummy Implementation to be used when no other implementation is available
/ E+ \$ j  @3 E" v+ p! }class CUPnPImplNone: public CUPnPImpl
{
public:
        virtual void        StartDiscovery(uint16, uint16, uint16)                                        { ASSERT( false ); }
. p7 y) I7 V, h2 U4 Z* @        virtual bool        CheckAndRefresh()                                                                                { return false; }
( V7 S4 W3 O! G) Q; ?        virtual void        StopAsyncFind()                                                                                        { }
        virtual void        DeletePorts()                                                                                        { }
        virtual bool        IsReady()                                                                                                { return false; }
        virtual int                GetImplementationID()                                                                        { return UPNP_IMPL_NONE; }
};
7 ^9 _7 o. W  A$ q& k

1 D  i. }& R' j/////////////////////////////////////
: N5 a, k) x- B% P3 d//下面是使用windows操作系统自带的UPNP功能的子类

$ z) G9 a; b/ G  G8 x0 X4 J. z  P
* m/ t( D( x' o3 I#pragma once
1 N2 `, F) u8 F& y#pragma warning( disable: 4355 )
( R: v" x* n0 e
  t, c6 N& [5 W6 H
#include "UPnPImpl.h"
  }. t1 A/ R7 f9 g8 s5 n) h#include <upnp.h>
#include <iphlpapi.h>
#include <comdef.h>
7 I; |# R6 y: O; c' U* [#include <winsvc.h>


# @, r0 d1 d2 t#include <vector>
8 Q# b- e% E" m#include <exception>
  |# w7 L$ o& p0 T+ h5 N' A#include <functional>
% Z- _8 c" C  q0 N; |& \' K% v3 T



- M' L6 a) m8 z2 u; l- w+ Xtypedef _com_ptr_t<_com_IIID<IUPnPDeviceFinder,&IID_IUPnPDeviceFinder> >        FinderPointer;
typedef _com_ptr_t<_com_IIID<IUPnPDevice,&IID_IUPnPDevice>        >                                DevicePointer;
4 O$ z% |8 ~' S$ I8 Atypedef _com_ptr_t<_com_IIID<IUPnPService,&IID_IUPnPService> >                                ServicePointer;
; v3 v6 }- F7 z) ]typedef        _com_ptr_t<_com_IIID<IUPnPDeviceFinderCallback,&IID_IUPnPDeviceFinderCallback> > DeviceFinderCallback;
typedef        _com_ptr_t<_com_IIID<IUPnPServiceCallback,&IID_IUPnPServiceCallback> >                        ServiceCallback;
typedef _com_ptr_t<_com_IIID<IEnumUnknown,&IID_IEnumUnknown> >                                EnumUnknownPtr;
typedef _com_ptr_t<_com_IIID<IUnknown,&IID_IUnknown> >                                                UnknownPtr;
: u. t7 w  O8 ]
! i" H+ E& [$ S2 q4 o* n- z; J( \3 {3 t
6 ^7 m+ u2 E$ Htypedef DWORD (WINAPI* TGetBestInterface) (
  IPAddr dwDestAddr,
  PDWORD pdwBestIfIndex
- N. A1 L' v4 l1 p: H$ p);


/ U2 ?6 F: t# x5 k9 {% btypedef DWORD (WINAPI* TGetIpAddrTable) (
! x# O* }* h% r9 Y& g- t  PMIB_IPADDRTABLE pIpAddrTable,
2 x8 p3 x1 z. i( S: a1 U  s6 c  PULONG pdwSize,
) G( R1 K& y' T$ C. \2 n# }: i% a  BOOL bOrder
);

; R: c3 r) _- u; z+ Q
2 Z2 F! X1 a$ O  m5 atypedef DWORD (WINAPI* TGetIfEntry) (
1 J( O4 K- e7 s( j' L7 k4 \! G( L) @5 \  PMIB_IFROW pIfRow
);

/ e5 u" y6 z/ i& T, T2 f$ B
" ~% l! {0 C6 K% E% r1 ZCString translateUPnPResult(HRESULT hr);
HRESULT UPnPMessage(HRESULT hr);
& ?( W( A! o2 @4 l, h% y

6 D) e; Y5 r/ Vclass CUPnPImplWinServ: public CUPnPImpl
{
        friend class CDeviceFinderCallback;
3 r$ j( |1 ?7 N& d        friend class CServiceCallback;
// Construction
2 `& s6 |+ I5 m! X3 Q9 k5 zpublic:
& d; ^7 x% {4 N! x6 p) n* H% ]        virtual ~CUPnPImplWinServ();
3 m/ a5 T( @9 a  ?/ r8 i. l        CUPnPImplWinServ();


        virtual void        StartDiscovery(uint16 nTCPPort, uint16 nUDPPort, uint16 nTCPWebPort)                { StartDiscovery(nTCPPort, nUDPPort, nTCPWebPort, false); }
0 Z4 e6 f' K: y# c        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)
" ?0 S4 _$ W% D$ k0 u1 R        // the windows side of the implementation tends to get bugged untill reboot anyway. Still might get added later
        virtual bool        CheckAndRefresh()                                                                                { return false; };

' N# l+ A# b' ], i2 M( @+ J
protected:
( [* C$ \" Z6 K! M2 E        void        StartDiscovery(uint16 nTCPPort, uint16 nUDPPort, uint16 nTCPWebPort, bool bSecondTry);
% ^) V- g8 f# {8 h# [        void        AddDevice(DevicePointer pDevice, bool bAddChilds, int nLevel = 0);
' `- m% y! C" I# b. f- A) p        void        RemoveDevice(CComBSTR bsUDN);
( Q) k4 N8 a; k+ n        bool        OnSearchComplete();
        void        Init();
6 d1 W. K6 _% b- `) [! {/ L

        inline bool IsAsyncFindRunning()
        {
: V% p8 C  j: h& B                if ( m_pDeviceFinder != NULL && m_bAsyncFindRunning && GetTickCount() - m_tLastEvent > 10000 )
5 ]+ ?% h* f2 \" ~/ n" h2 l: S6 z                {
6 ~' M  ]  A$ z6 q                        m_pDeviceFinder->CancelAsyncFind( m_nAsyncFindHandle );
                        m_bAsyncFindRunning = false;
/ X0 t* M0 N: i! w9 f) O                }
4 a+ {3 w, T: _0 w  T                MSG msg;
  m9 k* a  S7 Z3 ~# ]; K- Y& B                while ( PeekMessage( &msg, NULL, 0, 0, PM_REMOVE ) )
1 ~: t, S& u. m, x5 U" r2 f# A6 E                {
7 _& c6 C  W( E, a/ t                        TranslateMessage( &msg );
                        DispatchMessage( &msg );
                }
                return m_bAsyncFindRunning;
6 r5 o3 V9 ^- o( k' S* n6 w5 U        }
: {) ^/ ]+ ]$ D: o, W

        TRISTATE                        m_bUPnPDeviceConnected;

7 G  E1 F: a' a: F7 f! E
// Implementation
        // API functions
        SC_HANDLE (WINAPI *m_pfnOpenSCManager)(LPCTSTR, LPCTSTR, DWORD);
        SC_HANDLE (WINAPI *m_pfnOpenService)(SC_HANDLE, LPCTSTR, DWORD);
5 H7 I( G4 \1 r+ n# a3 F1 G  X; H/ @        BOOL (WINAPI *m_pfnQueryServiceStatusEx)(SC_HANDLE, SC_STATUS_TYPE, LPBYTE, DWORD, LPDWORD);
( w3 b& b7 m" K        BOOL (WINAPI *m_pfnCloseServiceHandle)(SC_HANDLE);
! O) }/ D9 d& ]  C8 R4 `        BOOL (WINAPI *m_pfnStartService)(SC_HANDLE, DWORD, LPCTSTR*);
1 h" Q+ q* M! a6 r        BOOL (WINAPI *m_pfnControlService)(SC_HANDLE, DWORD, LPSERVICE_STATUS);
$ z+ j1 s& ^: J8 p2 s6 }

# _& V8 H$ R* m6 L        TGetBestInterface                m_pfGetBestInterface;
        TGetIpAddrTable                        m_pfGetIpAddrTable;
        TGetIfEntry                                m_pfGetIfEntry;

& \, A+ i& H3 O: y5 g# `
        static FinderPointer CreateFinderInstance();
        struct FindDevice : std::unary_function< DevicePointer, bool >
        {
4 H1 s/ E$ e- V+ n! E( U                FindDevice(const CComBSTR& udn) : m_udn( udn ) {}
                result_type operator()(argument_type device) const
; ^1 U# {5 D$ i+ T                {
& p  ]' H( m# a+ T! z5 B- U9 L                        CComBSTR deviceName;
                        HRESULT hr = device->get_UniqueDeviceName( &deviceName );
* ~: ]1 s4 Y  E+ Q" O9 x

                        if ( FAILED( hr ) )
                                return UPnPMessage( hr ), false;
5 g  ]5 F9 _/ g+ s
, z% ]! u/ G+ R3 i# q4 T+ n1 a
                        return wcscmp( deviceName.m_str, m_udn ) == 0;
                }
  |" I1 n+ Q, w2 ?3 q% B! L                CComBSTR m_udn;
, J' [3 m; T6 K        };
! `) ~0 `, O" B: {, o2 f       
7 n) M9 e" N' }( T6 l        void        ProcessAsyncFind(CComBSTR bsSearchType);
        HRESULT        GetDeviceServices(DevicePointer pDevice);
, D* C: L$ p" b        void        StartPortMapping();
        HRESULT        MapPort(const ServicePointer& service);
        void        DeleteExistingPortMappings(ServicePointer pService);
        void        CreatePortMappings(ServicePointer pService);
        HRESULT SaveServices(EnumUnknownPtr pEU, const LONG nTotalItems);
        HRESULT InvokeAction(ServicePointer pService, CComBSTR action,
3 S$ V9 `9 I6 Q( C  r" r9 `                LPCTSTR pszInArgString, CString& strResult);
! c8 f& \  g( Q9 J, T2 o" w1 V5 E. Q2 u8 a        void        StopUPnPService();

# N" |7 P" ?" A5 A/ }5 S
        // Utility functions
        HRESULT CreateSafeArray(const VARTYPE vt, const ULONG nArgs, SAFEARRAY** ppsa);
$ e+ L* ^; r9 x& P/ ~; q& o0 a4 M        INT_PTR CreateVarFromString(const CString& strArgs, VARIANT*** pppVars);
, d3 g( i6 j% }- O4 [1 C        INT_PTR        GetStringFromOutArgs(const VARIANT* pvaOutArgs, CString& strArgs);
- m  s$ E. n! @) P/ h1 c        void        DestroyVars(const INT_PTR nCount, VARIANT*** pppVars);
& u% n# s& ]/ o( [* y        HRESULT GetSafeArrayBounds(SAFEARRAY* psa, LONG* pLBound, LONG* pUBound);
        HRESULT GetVariantElement(SAFEARRAY* psa, LONG pos, VARIANT* pvar);
        CString        GetLocalRoutableIP(ServicePointer pService);
9 \  @- E  t1 q4 {2 h7 ?  P

: P0 N1 ?3 s2 \" m8 a// Private members
private:
        DWORD        m_tLastEvent;        // When the last event was received?
# r/ Q6 z& f2 j( F& F- E        std::vector< DevicePointer >  m_pDevices;
% Y) M+ K' h% T9 Z        std::vector< ServicePointer > m_pServices;
        FinderPointer                        m_pDeviceFinder;
        DeviceFinderCallback        m_pDeviceFinderCallback;
- _; [" T2 }; k% `        ServiceCallback                        m_pServiceCallback;


/ h6 Z5 O- Z" U$ i3 [        LONG        m_nAsyncFindHandle;
        bool        m_bCOM;
  l/ m' x( g; W3 w1 N, g' F        bool        m_bPortIsFree;
. I3 q' D# g. p! j2 E, X6 h        CString m_sLocalIP;
        CString m_sExternalIP;
        bool        m_bADSL;                // Is the device ADSL?
4 V/ B3 Q( ^9 ^        bool        m_ADSLFailed;        // Did port mapping failed for the ADSL device?
# `% n, E+ E1 [* S; S! W        bool        m_bInited;
        bool        m_bAsyncFindRunning;
        HMODULE m_hADVAPI32_DLL;
        HMODULE        m_hIPHLPAPI_DLL;
' R$ k  D4 c# f5 S$ |        bool        m_bSecondTry;
        bool        m_bServiceStartedByEmule;
+ y% d2 T1 e( o& {        bool        m_bDisableWANIPSetup;
        bool        m_bDisableWANPPPSetup;
0 h$ W! H4 a  m2 x  |

};
+ d7 _  e: L( P& Q& K

// DeviceFinder Callback
% F0 \$ E: D' L9 w1 C! K! mclass CDeviceFinderCallback
        : public IUPnPDeviceFinderCallback
* o5 B6 e/ I6 M: v{
- `0 v; ?: A; h" m7 y8 ypublic:
        CDeviceFinderCallback(CUPnPImplWinServ& instance)
                : m_instance( instance )
        { m_lRefCount = 0; }


   STDMETHODIMP QueryInterface(REFIID iid, LPVOID* ppvObject);
2 ~3 _1 S; `9 D( X; p# p   STDMETHODIMP_(ULONG) AddRef();
$ M0 c1 E. I8 p' m9 f$ _   STDMETHODIMP_(ULONG) Release();
$ W$ V9 P0 t* e# s- W% K. Q+ G
4 a5 V- ]! m$ s  ~; p* j- X. Z+ v
) ]* B7 j  W: h8 v/ s// implementation
" _2 _. t  R  b7 z' ?/ q% Mprivate:
6 b4 g, C4 I7 {8 N! u3 x        HRESULT __stdcall DeviceAdded(LONG nFindData, IUPnPDevice* pDevice);
        HRESULT __stdcall DeviceRemoved(LONG nFindData, BSTR bsUDN);
$ h$ `) c8 X6 z1 ^        HRESULT __stdcall SearchComplete(LONG nFindData);

$ b/ r& I) v# r& o, }9 T9 x
/ r. X' @8 u. q: B, eprivate:
        CUPnPImplWinServ& m_instance;
        LONG m_lRefCount;
! F5 Z6 s, L- |2 Q3 I9 m};
$ H) O8 K7 y9 L

// Service Callback
7 A, E- v% |' J! t, w+ Xclass CServiceCallback
        : public IUPnPServiceCallback
5 {6 q9 H8 a- S1 O* _# F) V9 ~{
public:
        CServiceCallback(CUPnPImplWinServ& instance)
& B6 B6 \! ?' Z                : m_instance( instance )
        { m_lRefCount = 0; }
   
   STDMETHODIMP QueryInterface(REFIID iid, LPVOID* ppvObject);
   STDMETHODIMP_(ULONG) AddRef();
   STDMETHODIMP_(ULONG) Release();
/ @2 ^) R; F6 N+ e8 I
# w, M; ~6 S( \& _) n
// implementation
$ F. u! E- \/ p  ]private:
        HRESULT __stdcall StateVariableChanged(IUPnPService* pService, LPCWSTR pszStateVarName, VARIANT varValue);
5 o& A) C3 n6 O( X- t* {7 F& a        HRESULT __stdcall ServiceInstanceDied(IUPnPService* pService);


private:
9 e8 J7 d) B& S" Z; u% R        CUPnPImplWinServ& m_instance;
        LONG m_lRefCount;
};


8 u. G$ N5 X' Y# \/////////////////////////////////////////////////
  Y6 Y- Z& l) ?; L  b0 c7 h( K

使用时只需要使用抽象类的接口。
3 B* P4 k* A! y. d! ]! nCUPnPImpl::SetMessageOnResult设置需要接受UPNP端口映射是否成功的窗口句柄和消息ID.
$ u% @0 O7 E/ m1 C$ HCUPnPImpl::StartDiscovery将开启一个异步设备查找并进行端口映射,其参数为需要映射的内网端口.
% L( z8 a$ c( r. @CUPnPImpl::StopAsyncFind停止设备查找.
' g* J0 M2 _! `" S2 PCUPnPImpl::DeletePorts删除端口映射.