UPnP

天云上人

/*uPnP.h*/

1. 
   ' k' S% G9 P) W$ S& ?
2. #ifndef   MYUPNP_H_
   
3. 
   ! N+ i1 Y5 q; s4 |, N) {+ l5 j/ U- ~
4. #pragma   once
   / [! Z/ l+ l$ F2 L+ X
5. 
   6 _8 d" Z5 g$ R4 v4 w6 h
6. typedef   unsigned   long   ulong;
   5 V& c8 R$ J3 T7 D- K' N6 k7 B
7. 
   
8. class   MyUPnP
   6 X' O# K, f4 {; h. z$ v
9. {
   9 O; X! q/ @' }9 M3 F# i
10. public:
    * J, @3 m9 x' t5 n# R
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
    7 _' F. w7 i+ F
16. UNAT_NOT_IN_LAN //   Error,   you   aren 't   in   a   LAN   ->   no   router   or   firewall
    7 D8 w3 o8 G3 w  n! r
17. }   UPNPNAT_RETURN;
    
18. 
    
19. typedef   enum{
    
20. UNAT_TCP, //   TCP   Protocol
    : a. s$ L, p# z& o6 u, s
21. UNAT_UDP //   UDP   Protocol
    7 U% g5 h( |6 ?/ a. x0 T
22. }   UPNPNAT_PROTOCOL;
    . s+ l% j+ \$ h7 @
23. 
    5 V6 [$ ]( w+ `2 j5 l: G$ `& B
24. typedef   struct{
    
25. WORD   internalPort; //   Port   mapping   internal   port
    $ L* D9 S; u1 r- ^4 ^
26. WORD   externalPort; //   Port   mapping   external   port
    7 ]5 e& ]& f% {5 G" g
27. UPNPNAT_PROTOCOL   protocol; //   Protocol->   TCP   (UPNPNAT_PROTOCOL:UNAT_TCP)   ||   UDP   (UPNPNAT_PROTOCOL:UNAT_UDP)
    
28. CString   description; //   Port   mapping   description
    , d$ w! E* O2 Y5 r
29. }   UPNPNAT_MAPPING;
    
30. 
    
31. MyUPnP();
    . d* T. ^6 r0 Z
32. ~MyUPnP();
    
33. 
    6 i9 E: D2 [9 q% a' b
34. UPNPNAT_RETURN   AddNATPortMapping(UPNPNAT_MAPPING   *mapping,   bool   tryRandom   =   false);
    2 |# H# q0 T9 e* G  @( J
35. UPNPNAT_RETURN   RemoveNATPortMapping(UPNPNAT_MAPPING   mapping,   bool   removeFromList   =   true);
    
36. void   clearNATPortMapping();
    5 o+ w# }7 \+ |4 u5 c/ {
37. 
    
38. CString GetLastError();
    " ]: x+ H2 m# }$ U4 z
39. CString GetLocalIPStr();
    
40. WORD GetLocalIP();
    
41. bool IsLANIP(WORD   nIP);
    
42. 
    * [; B3 x3 j$ C' l. w8 J
43. protected:
    
44. void InitLocalIP();
    
45. void SetLastError(CString   error);
    ( I; ~% x, C! A+ l
46. 
    7 Q3 D/ _( `6 F
47. bool   addPortmap(int   eport,   int   iport,   const   CString&   iclient,
    * _( P$ \0 w1 _; E* {7 _# X
48.       const   CString&   descri,   const   CString&   type);
    " g5 O' \( h" u* T5 m
49. bool   deletePortmap(int   eport,   const   CString&   type);
    
50. 
    - i, }* U+ {* p: J$ L4 x
51. bool isComplete()   const   {   return   !m_controlurl.IsEmpty();   }
    
52. 
    
53. bool Search(int   version=1);
    6 o9 O  {7 _" [
54. bool GetDescription();
    4 {7 @9 r4 i, ^# A
55. CString GetProperty(const   CString&   name,   CString&   response);
    6 X, f. t$ K& O* b
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);
    ' e' G2 O: V  _
60. CString m_devicename;
    # Y, Q' @$ r/ ?( s, @6 f: A
61. CString m_name;
    
62. CString m_description;
    4 A, k" }% N. @7 w' V: O; }9 C. f
63. CString m_baseurl;
    
64. CString m_controlurl;
    $ T2 C+ c' C- R% t
65. CString m_friendlyname;
    1 U. n' T: K" b
66. CString m_modelname;
    ) ~5 u- c9 Q, p: B- x0 A- T7 J
67. int m_version;
    
68. 
    1 H4 l; O5 C* s
69. private:
    + V  X- P- q( l  d
70. CList <UPNPNAT_MAPPING,   UPNPNAT_MAPPING>   m_Mappings;
    . D, m9 Q5 R1 s, _# k
71. 
    
72. CString m_slocalIP;
    
73. CString m_slastError;
    ; f4 ]0 w7 y7 ?4 D* `
74. WORD m_uLocalIP;
    ; W8 {! N! r+ @; s3 P
75. 
    ! K6 O9 p0 N/ w1 X8 s% r
76. bool isSearched;
    * q9 U' U5 A- K/ i# p6 _/ Q3 I
77. };
    * A' g2 ?' `6 ^8 W" c, p
78. #endif

复制代码

天云上人

/*UPnP.cpp*/

1. 
   
2. #include   "stdafx.h "
   
3. 
   # Q  u& E1 l7 Q1 `3 x. A5 k! v
4. #include   "upnp.h "
   
5. 
   
6. #define   UPNPPORTMAP0       _T( "WANIPConnection ")
   
7. #define   UPNPPORTMAP1       _T( "WANPPPConnection ")
   9 b6 j7 f! _! z( u
8. #define   UPNPGETEXTERNALIP   _T( "GetExternalIPAddress "),_T( "NewExternalIPAddress ")
   
9. #define   UPNPADDPORTMAP   _T( "AddPortMapping ")
   
10. #define   UPNPDELPORTMAP   _T( "DeletePortMapping ")
    
11. 
    # D& A( x4 Z8 A3 \& n$ f/ V& @
12. static   const   ulong UPNPADDR   =   0xFAFFFFEF;
    
13. static   const   int UPNPPORT   =   1900;
    
14. static   const   CString URNPREFIX   =   _T( "urn:schemas-upnp-org: ");
    7 ?4 p1 O  O7 V0 d) `. W% J3 ^
15. 
    
16. const   CString   getString(int   i)
    5 e- e* ?) Q; M) P
17. {
    
18. CString   s;
    . `$ I  M  v8 ]& o0 T
19. 
    
20. s.Format(_T( "%d "),   i);
    + A5 g' y8 n( U( A
21. 
    4 ]  }5 U8 z: m: m4 O! Z& ?! C
22. return   s;
    1 I$ [) M" Z. ]! G! U6 h
23. }
    
24. 
    
25. const   CString   GetArgString(const   CString&   name,   const   CString&   value)
    ( w8 m" d/ E. K8 l8 u* ~; F
26. {
    # {9 d) C6 c8 B/ V
27. return   _T( " < ")   +   name   +   _T( "> ")   +   value   +   _T( " </ ")   +   name   +   _T( "> ");
    
28. }
    3 H/ p8 v& h' ^3 e) s2 b) Z3 n# I
29. 
    8 y9 q( y, x5 z; Z
30. const   CString   GetArgString(const   CString&   name,   int   value)
    
31. {
    3 J( U5 H5 X1 ?
32. return   _T( " < ")   +   name   +   _T( "> ")   +   getString(value)   +   _T( " </ ")   +   name   +   _T( "> ");
    " s) a, z$ M$ y% I8 N
33. }
    : T8 {6 {5 q( Q9 p% M
34. 
    - h; Y7 V  x: n* N; ^* \
35. bool   SOAP_action(CString   addr,   uint16   port,   const   CString   request,   CString   &response)
    
36. {
    
37. char   buffer[10240];
    
38. 
    % Y& O( r4 ~. T+ N4 Y* u
39. const   CStringA   sa(request);
    
40. int   length   =   sa.GetLength();
    8 D+ \  {/ u9 W$ w. _
41. strcpy(buffer,   (const   char*)sa);
    
42. 
    
43. uint32   ip   =   inet_addr(CStringA(addr));
    
44. struct   sockaddr_in   sockaddr;
    
45. memset(&sockaddr,   0,   sizeof(sockaddr));
    - r. w9 n7 x" \3 y
46. sockaddr.sin_family   =   AF_INET;
    & q) K9 B+ s6 k4 g) z  I) l$ k
47. sockaddr.sin_port   =   htons(port);
    
48. sockaddr.sin_addr.S_un.S_addr   =   ip;
    
49. int   s   =   socket(AF_INET,   SOCK_STREAM,   0);
      S/ k7 E+ `4 f
50. u_long   lv   =   1;
    
51. ioctlsocket(s,   FIONBIO,   &lv);
    
52. connect(s,   (struct   sockaddr   *)&sockaddr,   sizeof(sockaddr));
    
53. Sleep(20);
    % \* c' r% f) C/ _, V: }
54. int   n   =   send(s,   buffer,   length,   0);
    
55. Sleep(100);
    
56. int   rlen   =   recv(s,   buffer,   sizeof(buffer),   0);
    - }+ A4 K* S# \4 ^0 |6 |4 O
57. closesocket(s);
    2 j5 v& K! Z, m
58. if   (rlen   ==   SOCKET_ERROR)   return   false;
    
59. if   (!rlen)   return   false;
    ' m4 k1 f& i* u, F' n: H$ i# T
60. 
    # _+ B4 A4 w1 ]1 R5 y" o
61. response   =   CString(CStringA(buffer,   rlen));
    & G% q. _4 V- u( ~
62. 
    
63. return   true;
    $ v* e8 k2 G- T1 t. ?! k1 r
64. }
    
65. 
    
66. int   SSDP_sendRequest(int   s,   uint32   ip,   uint16   port,   const   CString&   request)
    ' p3 R9 F; n9 m2 V( b
67. {
    # I. E/ |" ?1 w/ n; M9 u, J
68. char   buffer[10240];
    
69. 
    1 H( g9 z" p) x! o
70. const   CStringA   sa(request);
    
71. int   length   =   sa.GetLength();
      y/ z2 m' E$ }! y2 f( b* i) y% E
72. strcpy(buffer,   (const   char*)sa);
    # r' g' {) `& E; R+ f
73. 
    ! T  a" l9 Y" W8 t, D) P
74. struct   sockaddr_in   sockaddr;
    
75. memset(&sockaddr,   0,   sizeof(sockaddr));
    
76. sockaddr.sin_family   =   AF_INET;
    
77. sockaddr.sin_port   =   htons(port);
    
78. sockaddr.sin_addr.S_un.S_addr   =   ip;
    ! r# G- E3 j9 p4 S( E
79. 
    
80. return   sendto(s,   buffer,   length,   0,   (struct   sockaddr   *)&sockaddr,   sizeof(sockaddr));
    
81. }
    6 t5 {- l% J$ p0 o
82. 
    * t  d9 w/ h( D- ?. R# d0 m9 u
83. bool   parseHTTPResponse(const   CString&   response,   CString&   result)
    ! t; A, {; J1 N+ H/ m, J
84. {
    / |# G- P3 X  |) E; i% M
85. int   pos   =   0;
    $ i9 X1 F/ d' ]" Z
86. 
    2 J/ Z$ u& |$ ^8 C" v; N- \
87. CString   status   =   response.Tokenize(_T( "\r\n "),   pos);
    
88. 
    4 i) P9 q* p/ |* h" C% s2 F
89. result   =   response;
    0 O0 }' p5 M; S+ J8 a5 s  h
90. result.Delete(0,   pos);
    
91. 
    9 B6 v, t( ?3 i: ^1 ~
92. pos   =   0;
    ) k1 {8 g( F+ I( u% L0 l4 K
93. status.Tokenize(_T( "   "),   pos);
    
94. status   =   status.Tokenize(_T( "   "),   pos);
    
95. if   (status.IsEmpty()   ||   status[0]!= '2 ')   return   false;
    5 {& {7 T1 {; v0 u' r. C/ e
96. return   true;
    
97. }
    0 ], ~! h3 p, v
98. 
    
99. const   CString   getProperty(const   CString&   all,   const   CString&   name)
    ( K1 w$ m, P) P8 s) A4 x
100. {
     2 O( w# \3 t1 f/ a$ O- w
101. CString   startTag   =   ' < '   +   name   +   '> ';
     : [: Q+ I* ]4 ~. ]; I
102. CString   endTag   =   _T( " </ ")   +   name   +   '> ';
     
103. CString   property;
     ( m4 W8 s' q2 h) U6 B) x
104. 
     
105. int   posStart   =   all.Find(startTag);
     " q+ [: P5 Z0 g( m2 ?- u5 F2 V" c) a
106. if   (posStart <0)   return   CString();
       {- |* |; v# T1 O/ H
107. 
     6 k5 B6 ^- }! t* o+ p0 k
108. int   posEnd   =   all.Find(endTag,   posStart);
     
109. if   (posStart> =posEnd)   return   CString();
     
110. 
     
111. return   all.Mid(posStart   +   startTag.GetLength(),   posEnd   -   posStart   -   startTag.GetLength());
     
112. }
     2 Q# ]+ B+ |7 i, h; ^+ D/ T
113. 
     / z. E5 r$ W; _
114. MyUPnP::MyUPnP()
     
115. :   m_version(1)
     * |* Q: }% B$ _3 x3 H+ E
116. {
     
117. m_uLocalIP   =   0;
     ; Y6 A6 x* {7 d$ Y* F& m
118. isSearched   =   false;
     % V$ A: A4 @2 h
119. }
     
120. 
     % }" t; l8 y0 M* F* G. k+ x* p
121. MyUPnP::~MyUPnP()
     
122. {
     
123. UPNPNAT_MAPPING   search;
     
124. POSITION   pos   =   m_Mappings.GetHeadPosition();
     
125. while(pos){
     
126. search   =   m_Mappings.GetNext(pos);
     
127. RemoveNATPortMapping(search,   false);
     ! `5 }  W* a" b
128. }
       c5 i, R/ R$ {' i
129. 
     ' p- j2 T2 t3 \  C, [
130. m_Mappings.RemoveAll();
     9 ~2 u. e2 b+ K  c
131. }
     
132. 
     
133. 
     
134. bool   MyUPnP::InternalSearch(int   version)
     4 U7 R# y5 I$ q" o$ Q* g
135. {
     
136. if(version <=0)version   =   1;
     
137. m_version   =   version;
     
138. 
     , n4 i* k1 E! W+ V, K) E
139. #define   NUMBEROFDEVICES 2
     : t, Q( Y5 v+ F0 E8 |( t  B
140. CString   devices[][2]   =   {
     ( Y; B- a5 _) c+ c5 ^
141. {UPNPPORTMAP1,   _T( "service ")},
     + L+ }( U5 O& r9 M# T
142. {UPNPPORTMAP0,   _T( "service ")},
     + t( k+ R. a/ r: Y7 X0 I9 Z
143. {_T( "InternetGatewayDevice "),   _T( "device ")},
     
144. };
     
145. 
     
146. int   s   =   socket(AF_INET,   SOCK_DGRAM,   0);
     2 V3 g; i. I6 Q/ p
147. u_long   lv   =   1;
     
148. ioctlsocket(s,   FIONBIO,   &lv);
     
149. 
     - E( e- g5 ]% D# V( @/ |
150. int   rlen   =   0;
     / D4 f6 [" E( Z# K+ W
151. for   (int   i=0;   rlen <=0   &&   i <500;   i++)   {
     ) [3 g* Z! d1 z/ y# u
152. if   (!(i%100))   {
     
153. for   (int   i=0;   i <NUMBEROFDEVICES;   i++)   {
     ) t" ^& |& K; R
154. m_name.Format(_T( "%s%s:%s:%d "),   URNPREFIX,   devices[i][1],   devices[i][0],   version);
     " _( u5 L$ I3 B8 e7 U7 P
155. CString   request;
     & W# O- W, |- k! [, W- M' S
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. }
     % b! W% O* C/ H7 ^
160. }
     2 y+ Z5 Y$ E* w8 V. R. R. _
161. 
     + E. \* ?( N) l' R' ~5 r
162. Sleep(10);
     
163. 
     % E; C8 s# B% [7 K0 _
164. char   buffer[10240];
     
165. rlen   =   recv(s,   buffer,   sizeof(buffer),   0);
     7 _0 l" L! ~2 c2 c$ F* W1 n
166. if   (rlen   <=   0)   continue;
     
167. closesocket(s);
     " ^: u; K5 G0 G- `' [  e
168. 
     
169. CString   response   =   CString(CStringA(buffer,   rlen));
     
170. CString   result;
     . B. @+ k  V9 v3 U
171. if   (!parseHTTPResponse(response,   result))   return   false;
     ! \% U9 O4 a, i0 p* ^8 Z8 T
172. 
     0 H' f" d0 }. t
173. for   (int   d=0;   d <NUMBEROFDEVICES;   d++)   {
     ' C* w/ }% l; E4 |0 v) j+ Y
174. m_name.Format(_T( "%s%s:%s:%d "),   URNPREFIX,   devices[d][1],   devices[d][0],   version);
     
175. if   (result.Find(m_name)   > =   0)   {
     
176. for   (int   pos   =   0;;)   {
     ( x0 F6 {+ X, H/ W
177. CString   line   =   result.Tokenize(_T( "\r\n "),   pos);
     
178. if   (line.IsEmpty())   return   false;
     2 Y5 e. ?0 E4 D8 k+ q0 Z
179. CString   name   =   line.Mid(0,   9);
     
180. name.MakeUpper();
     ' q; H+ B- P8 j1 [
181. if   (name   ==   _T( "LOCATION: "))   {
     
182. line.Delete(0,   9);
     & P- U& y$ e* O5 k0 Y5 J
183. m_description   =   line;
     
184. m_description.Trim();
     8 t# N3 P' ?- \* K$ p/ p* V
185. return   GetDescription();
     
186. }
     
187. }
     
188. }
     
189. }
     , F% W! q& q) N  U, o0 p
190. }
     * r8 C) n, S7 k+ {( }
191. closesocket(s);
     
192. 
     3 E8 i9 O" P1 ~0 m$ C, R# @7 `
193. return   false;
     ) X/ |  X& p$ {4 r, R( t
194. }
     8 J) m0 i- f! g4 w- z

复制代码

天云上人

以下有关upnp的接口来自emule，
9 ~/ |9 Z2 ]* C% U/ m: Q5 i; X3 |
1 `+ z+ f! W; S: L$ I
9 M+ t8 n( t" z% U! H/ G9 _///////////////////////////////////////////
9 `" I, e# \( d& `9 l2 W//upnp的抽象接口类,只需调用这些接口即可实现端口映射功能.
, a% w  |; |$ z/ e  S7 ]. G

, `+ L" `4 U* B" s2 E1 I#pragma once
& N) }1 [* C9 h; j! X5 C# e#include <exception>
1 ?5 K' X1 a( G  F" z4 I3 S5 a

  enum TRISTATE{
        TRIS_FALSE,
        TRIS_UNKNOWN,
2 E# A" P/ g- ^' N) C7 S        TRIS_TRUE
- J$ L5 Q8 h3 \! I; D: e};

* a: g0 J2 c/ F1 Z
enum UPNP_IMPLEMENTATION{
        UPNP_IMPL_WINDOWSERVICE = 0,
- s( n1 {0 p1 j, J, Q        UPNP_IMPL_MINIUPNPLIB,
        UPNP_IMPL_NONE /*last*/
};
+ N3 x+ k) e1 n; M



class CUPnPImpl
$ [/ Q& ?& A+ Y0 T* |) }2 P& ~- j{
public:
        CUPnPImpl();
        virtual ~CUPnPImpl();
        struct UPnPError : std::exception {};
        enum {
                UPNP_OK,
+ t& }. H6 @7 e" J! a                UPNP_FAILED,
/ ]$ E) Q7 B1 k% m4 V# N                UPNP_TIMEOUT
4 N. y" {" ]; f  s7 J        };
: }9 Z# o- v/ c0 o9 C; Q+ G" ?
1 O8 p( M6 B) R
        virtual void        StartDiscovery(uint16 nTCPPort, uint16 nUDPPort, uint16 nTCPWebPort) = 0;
6 O3 ?$ i  h0 B9 x        virtual bool        CheckAndRefresh() = 0;
        virtual void        StopAsyncFind() = 0;
0 m/ a/ A3 p. f        virtual void        DeletePorts() = 0;
" _$ R- i. ]6 J) g        virtual bool        IsReady() = 0;
        virtual int                GetImplementationID() = 0;
       
        void                        LateEnableWebServerPort(uint16 nPort); // Add Webserverport on already installed portmapping


8 e9 k  d- h# `0 p- k( m+ |# S( _; m        void                        SetMessageOnResult(HWND hWindow, UINT nMessageID);
        TRISTATE                ArePortsForwarded() const                                                                { return m_bUPnPPortsForwarded; }
        uint16                        GetUsedTCPPort()                                                                                { return m_nTCPPort; }
% y8 g$ J3 [! f6 x% B+ i$ F) w; Q  ?1 k        uint16                        GetUsedUDPPort()                                                                                { return m_nUDPPort; }       
( F. E, K% b: }0 B/ F+ @. g
5 Q$ p0 k3 N7 f- I
// Implementation
protected:
        volatile TRISTATE        m_bUPnPPortsForwarded;
$ v3 B. z) ?$ n3 \0 @& {2 R        void                                SendResultMessage();
  M9 x% n/ ~1 t# r* G! f. r        uint16                                m_nUDPPort;
        uint16                                m_nTCPPort;
        uint16                                m_nTCPWebPort;
/ z' M# g6 H: [0 ]2 B        bool                                m_bCheckAndRefresh;

3 ]1 f4 q5 X2 A2 m! M/ Y! _5 U
% _2 X: W, {  u; L: A* kprivate:
  e2 y- h; d8 {: s        HWND        m_hResultMessageWindow;
        UINT        m_nResultMessageID;


9 b- G( H, ]/ E- m* p5 P};
, [3 E% ]$ w% [! Q' a

( V/ l, |" F/ J' C: x- `// Dummy Implementation to be used when no other implementation is available
- g$ f+ }' f' }/ \$ g8 E1 w; V( _class CUPnPImplNone: public CUPnPImpl
, y) ?& b* c2 f# S+ D! `{
  U9 D8 U2 t1 P% e4 L) U$ _# ppublic:
        virtual void        StartDiscovery(uint16, uint16, uint16)                                        { ASSERT( false ); }
0 x& d+ ^1 Z, ]        virtual bool        CheckAndRefresh()                                                                                { return false; }
        virtual void        StopAsyncFind()                                                                                        { }
        virtual void        DeletePorts()                                                                                        { }
        virtual bool        IsReady()                                                                                                { return false; }
& Y0 O; r$ }& D- |/ a. N& l        virtual int                GetImplementationID()                                                                        { return UPNP_IMPL_NONE; }
};


/////////////////////////////////////
. A2 S+ r- y* O5 b- J& ^1 Y, H//下面是使用windows操作系统自带的UPNP功能的子类


+ \) w" w/ o) e2 g; r& e#pragma once
#pragma warning( disable: 4355 )

+ y: {/ U. \9 W" y- \
#include "UPnPImpl.h"
  P( Q& X/ E3 ^6 ]  v- f0 f3 X#include <upnp.h>
#include <iphlpapi.h>
#include <comdef.h>
#include <winsvc.h>
$ k* f3 i# u; Q; ~* t
+ D& n! t+ L" d3 r6 ?
, X% y' u. \; V$ Q9 e#include <vector>
: O# N4 |( F* F) g3 Z+ G#include <exception>
4 s& o2 q1 n0 }7 J; q#include <functional>
/ C$ @* r4 a, f/ C! |! e9 p
" P4 k3 d  K" v# k3 o4 t7 y
9 D0 ?9 W2 \0 I2 R; X8 D6 h
% J* Y* c. R' g' [& r! i0 E
typedef _com_ptr_t<_com_IIID<IUPnPDeviceFinder,&IID_IUPnPDeviceFinder> >        FinderPointer;
typedef _com_ptr_t<_com_IIID<IUPnPDevice,&IID_IUPnPDevice>        >                                DevicePointer;
: ^) \  x& d* {typedef _com_ptr_t<_com_IIID<IUPnPService,&IID_IUPnPService> >                                ServicePointer;
/ M  M% Y  z/ n4 ~7 L* w7 Vtypedef        _com_ptr_t<_com_IIID<IUPnPDeviceFinderCallback,&IID_IUPnPDeviceFinderCallback> > DeviceFinderCallback;
1 c. E0 i1 {7 C. {typedef        _com_ptr_t<_com_IIID<IUPnPServiceCallback,&IID_IUPnPServiceCallback> >                        ServiceCallback;
typedef _com_ptr_t<_com_IIID<IEnumUnknown,&IID_IEnumUnknown> >                                EnumUnknownPtr;
0 |' L9 d, U3 r0 ^# P8 l$ ntypedef _com_ptr_t<_com_IIID<IUnknown,&IID_IUnknown> >                                                UnknownPtr;
; x7 g$ s2 {8 p) L3 F+ d& D2 h& {
1 ^2 N6 d% ?1 ~5 P
# g1 E4 B# c# d- Ktypedef DWORD (WINAPI* TGetBestInterface) (
  IPAddr dwDestAddr,
  PDWORD pdwBestIfIndex
) F1 n9 `% I/ q1 O( B);
) K9 I: q1 \# E+ O+ N! Q
$ |/ x6 v% Q6 c  ^& s& E8 Q+ m
typedef DWORD (WINAPI* TGetIpAddrTable) (
  PMIB_IPADDRTABLE pIpAddrTable,
& i) L! o8 b6 ~. J/ E  PULONG pdwSize,
  BOOL bOrder
: G  c' w. N+ p# Q: `);
/ g3 U( e, w' u8 y: `
" F# P8 M. k( \$ m5 _$ J
typedef DWORD (WINAPI* TGetIfEntry) (
  PMIB_IFROW pIfRow
);
8 O( T  N  `. B* n6 E) i
! K, H/ A) `9 D* [9 d1 l& R
CString translateUPnPResult(HRESULT hr);
2 W4 T' T; @3 P& vHRESULT UPnPMessage(HRESULT hr);
/ H; A# ?+ V, i
' I9 x( _% X3 N& D
class CUPnPImplWinServ: public CUPnPImpl
: D  u& n; ^4 W0 k{
        friend class CDeviceFinderCallback;
        friend class CServiceCallback;
; B/ w  r& A. d' S% R// Construction
public:
        virtual ~CUPnPImplWinServ();
        CUPnPImplWinServ();
8 G3 [; C7 S8 L* R+ v

: z  B4 l8 W; C' Y) U. ~0 X; z        virtual void        StartDiscovery(uint16 nTCPPort, uint16 nUDPPort, uint16 nTCPWebPort)                { StartDiscovery(nTCPPort, nUDPPort, nTCPWebPort, false); }
        virtual void        StopAsyncFind();
- t/ S/ W  M2 l* ]% V+ Z        virtual void        DeletePorts();
/ Z+ P6 K! v5 x7 J3 L. V; ^        virtual bool        IsReady();
        virtual int                GetImplementationID()                                                                        { return UPNP_IMPL_WINDOWSERVICE; }
; _, `4 D. R& o+ M
8 s& a( m1 w8 _
2 Q& F" F1 l' o" S7 T        // No Support for Refreshing on this  (fallback) implementation yet - in many cases where it would be needed (router reset etc)
* a7 U  `1 D7 B# l. k- X8 g/ J2 k2 x        // the windows side of the implementation tends to get bugged untill reboot anyway. Still might get added later
        virtual bool        CheckAndRefresh()                                                                                { return false; };
9 S0 ]0 r8 k/ r. Z, o( u9 U
. g/ Z8 W. H' [8 L
( [9 x/ X" U$ [4 ^9 k7 @7 |/ q( y7 Fprotected:
" {) g, L4 E% a5 `( o        void        StartDiscovery(uint16 nTCPPort, uint16 nUDPPort, uint16 nTCPWebPort, bool bSecondTry);
( K5 e" q; A& J% x6 P6 w; n" W        void        AddDevice(DevicePointer pDevice, bool bAddChilds, int nLevel = 0);
( d2 g& P2 s: Q9 |) g- g        void        RemoveDevice(CComBSTR bsUDN);
        bool        OnSearchComplete();
0 L7 B8 |* a! a        void        Init();
# h, [5 V+ M$ M4 ?9 X; L) F
4 D$ G& u; r+ L
        inline bool IsAsyncFindRunning()
        {
                if ( m_pDeviceFinder != NULL && m_bAsyncFindRunning && GetTickCount() - m_tLastEvent > 10000 )
# [, \  b+ M! l5 B+ ~                {
2 {9 k& G& B+ t" k0 N                        m_pDeviceFinder->CancelAsyncFind( m_nAsyncFindHandle );
1 O' y$ t' M3 d3 f8 l                        m_bAsyncFindRunning = false;
                }
" K' f  ?$ U0 b2 g* ?+ [) V5 A                MSG msg;
                while ( PeekMessage( &msg, NULL, 0, 0, PM_REMOVE ) )
4 p. H5 o4 {7 t& N  U" q! j                {
! q4 N9 c7 n  h                        TranslateMessage( &msg );
" O2 Q' j6 b& t( A* |9 n/ {                        DispatchMessage( &msg );
                }
( G  x9 B$ A2 J( M/ D, D6 S. g                return m_bAsyncFindRunning;
4 B9 F2 i' D# X& u        }
6 h+ Z; K3 ^( O

/ U( {" C$ W- T1 [3 d        TRISTATE                        m_bUPnPDeviceConnected;
! Q! S3 @& F4 E
4 M, j$ S$ A/ h' y; O
// Implementation
        // API functions
        SC_HANDLE (WINAPI *m_pfnOpenSCManager)(LPCTSTR, LPCTSTR, DWORD);
6 [2 E! h" t  V. N        SC_HANDLE (WINAPI *m_pfnOpenService)(SC_HANDLE, LPCTSTR, DWORD);
  f: m9 C9 A, S        BOOL (WINAPI *m_pfnQueryServiceStatusEx)(SC_HANDLE, SC_STATUS_TYPE, LPBYTE, DWORD, LPDWORD);
        BOOL (WINAPI *m_pfnCloseServiceHandle)(SC_HANDLE);
" p9 i" L, ~2 {/ g1 Z7 d        BOOL (WINAPI *m_pfnStartService)(SC_HANDLE, DWORD, LPCTSTR*);
        BOOL (WINAPI *m_pfnControlService)(SC_HANDLE, DWORD, LPSERVICE_STATUS);


& s0 `7 U( ~! j        TGetBestInterface                m_pfGetBestInterface;
+ R. @3 p" `4 N( B- {        TGetIpAddrTable                        m_pfGetIpAddrTable;
        TGetIfEntry                                m_pfGetIfEntry;

, }# g# k* |9 b
        static FinderPointer CreateFinderInstance();
        struct FindDevice : std::unary_function< DevicePointer, bool >
+ U7 f/ C5 ~, w- F- X! }( C        {
3 q$ W5 h/ Q, Y& A! C                FindDevice(const CComBSTR& udn) : m_udn( udn ) {}
5 N) F% c: l, q9 P                result_type operator()(argument_type device) const
                {
                        CComBSTR deviceName;
) u2 _: K/ ^0 i8 f: Z6 y                        HRESULT hr = device->get_UniqueDeviceName( &deviceName );
% U: F+ @" |. h; d+ _. }
, s8 ?9 O) I4 J3 R# {2 a$ P/ }
4 S4 a" z0 x3 F                        if ( FAILED( hr ) )
                                return UPnPMessage( hr ), false;
3 h3 s$ f/ [) ~, H' n0 ?
7 T, f8 F0 n0 W2 i0 H
                        return wcscmp( deviceName.m_str, m_udn ) == 0;
3 V3 e: X, r4 a                }
" m8 b# n4 \+ a+ o2 H, X                CComBSTR m_udn;
        };
       
* I  E2 q" ]0 a' a        void        ProcessAsyncFind(CComBSTR bsSearchType);
5 @  Q) k6 C, a. ]) B$ q        HRESULT        GetDeviceServices(DevicePointer pDevice);
        void        StartPortMapping();
4 |( y7 d3 Y3 C5 r# f- C        HRESULT        MapPort(const ServicePointer& service);
        void        DeleteExistingPortMappings(ServicePointer pService);
        void        CreatePortMappings(ServicePointer pService);
        HRESULT SaveServices(EnumUnknownPtr pEU, const LONG nTotalItems);
* G- f. s5 k9 X7 S3 d& T        HRESULT InvokeAction(ServicePointer pService, CComBSTR action,
. V4 G- t" m: I$ f                LPCTSTR pszInArgString, CString& strResult);
        void        StopUPnPService();


        // Utility functions
        HRESULT CreateSafeArray(const VARTYPE vt, const ULONG nArgs, SAFEARRAY** ppsa);
  O  x' u* `, a" N7 G# B# C        INT_PTR CreateVarFromString(const CString& strArgs, VARIANT*** pppVars);
        INT_PTR        GetStringFromOutArgs(const VARIANT* pvaOutArgs, CString& strArgs);
        void        DestroyVars(const INT_PTR nCount, VARIANT*** pppVars);
        HRESULT GetSafeArrayBounds(SAFEARRAY* psa, LONG* pLBound, LONG* pUBound);
        HRESULT GetVariantElement(SAFEARRAY* psa, LONG pos, VARIANT* pvar);
/ x7 g9 T: o6 O( b        CString        GetLocalRoutableIP(ServicePointer pService);


// Private members
private:
        DWORD        m_tLastEvent;        // When the last event was received?
        std::vector< DevicePointer >  m_pDevices;
$ i0 C) X# w) g4 S4 M        std::vector< ServicePointer > m_pServices;
        FinderPointer                        m_pDeviceFinder;
3 [) X5 P: [) I7 h        DeviceFinderCallback        m_pDeviceFinderCallback;
        ServiceCallback                        m_pServiceCallback;
+ w( [4 L7 o4 l8 ~1 T* j

% k1 {7 f" r% `6 ?, |; t        LONG        m_nAsyncFindHandle;
* \) v7 r" D$ c& @" I3 a        bool        m_bCOM;
        bool        m_bPortIsFree;
5 }" S8 J9 w9 h. ^( _4 Q/ X        CString m_sLocalIP;
        CString m_sExternalIP;
7 o0 a6 a' @# b+ R7 T7 G        bool        m_bADSL;                // Is the device ADSL?
        bool        m_ADSLFailed;        // Did port mapping failed for the ADSL device?
        bool        m_bInited;
9 ^  y1 F! R, Z/ _0 B, K; X3 b        bool        m_bAsyncFindRunning;
; r7 W2 A& M. B+ }: Q$ M4 O" h        HMODULE m_hADVAPI32_DLL;
        HMODULE        m_hIPHLPAPI_DLL;
        bool        m_bSecondTry;
5 }  @# ^) Y  h: o  B  p$ q        bool        m_bServiceStartedByEmule;
3 u- R8 A; _: [. K5 Q2 `% R" @        bool        m_bDisableWANIPSetup;
        bool        m_bDisableWANPPPSetup;

  Q& h, i# b) D6 n, p! m
};

4 K: \, i+ ~7 q  l0 S- A5 r$ i" |9 ^
// DeviceFinder Callback
/ |9 y0 B- I0 E/ x& Wclass CDeviceFinderCallback
        : public IUPnPDeviceFinderCallback
" Q" P6 ]9 Q: G* \7 r" Y{
+ d& ]5 F4 ^3 V8 ?, a2 g! y: Epublic:
        CDeviceFinderCallback(CUPnPImplWinServ& instance)
                : m_instance( instance )
0 l2 X5 w* G1 ?* T+ [. s3 e6 ?        { m_lRefCount = 0; }
/ v! g+ y" ?$ X! Y" U

   STDMETHODIMP QueryInterface(REFIID iid, LPVOID* ppvObject);
   STDMETHODIMP_(ULONG) AddRef();
   STDMETHODIMP_(ULONG) Release();
# J: H8 U7 ^& n6 s

// implementation
" n! Y% W3 p' ~" Dprivate:
! k6 y) r4 W  s2 E5 f3 }( r        HRESULT __stdcall DeviceAdded(LONG nFindData, IUPnPDevice* pDevice);
        HRESULT __stdcall DeviceRemoved(LONG nFindData, BSTR bsUDN);
        HRESULT __stdcall SearchComplete(LONG nFindData);
" V& k- e8 o# E" A% b! y0 K2 Q
5 f2 V, j: @2 {* F; l
private:
        CUPnPImplWinServ& m_instance;
7 N% ]- B: x$ N( B* U        LONG m_lRefCount;
};
2 N* n4 U- y# `
5 F4 N# \! {6 L+ D/ P
. b1 O9 p& d/ a! g// Service Callback
class CServiceCallback
        : public IUPnPServiceCallback
{
2 |% S1 s2 B5 Q( n6 n9 mpublic:
8 Y$ @; G! z9 a* ]        CServiceCallback(CUPnPImplWinServ& instance)
                : m_instance( instance )
. X4 b3 B6 I( r, ~        { m_lRefCount = 0; }
" ]) u2 n" g; c2 r1 i2 V   
   STDMETHODIMP QueryInterface(REFIID iid, LPVOID* ppvObject);
   STDMETHODIMP_(ULONG) AddRef();
. J/ M! ^' f, D0 ~. k   STDMETHODIMP_(ULONG) Release();

: c8 o' E0 W% k7 x# t" ?
// implementation
3 b- `/ S6 G/ l2 W, kprivate:
  j3 x( I( e4 i* i! X0 ~        HRESULT __stdcall StateVariableChanged(IUPnPService* pService, LPCWSTR pszStateVarName, VARIANT varValue);
0 _' O, o2 ^! w        HRESULT __stdcall ServiceInstanceDied(IUPnPService* pService);
: P% E" Z9 I0 |; K* y% }' Z) W4 a$ m

2 T, N0 |2 [* wprivate:
        CUPnPImplWinServ& m_instance;
        LONG m_lRefCount;
};
' c/ ]5 [7 Y! I. B! B4 ^7 Y
7 X) b& l, i! t5 k
) r8 m( w% u; M/////////////////////////////////////////////////
$ o0 Q$ F- r6 r2 a) L, z3 f

使用时只需要使用抽象类的接口。
CUPnPImpl::SetMessageOnResult设置需要接受UPNP端口映射是否成功的窗口句柄和消息ID.
& |0 R& n% D. gCUPnPImpl::StartDiscovery将开启一个异步设备查找并进行端口映射,其参数为需要映射的内网端口.
* ^- K  y* o/ }9 Y6 [1 cCUPnPImpl::StopAsyncFind停止设备查找.
: S4 X" }( b! F7 r8 e( C- jCUPnPImpl::DeletePorts删除端口映射.