]>
Commit | Line | Data |
---|---|---|
1 | /////////////////////////////////////////////////////////////////////////////// | |
2 | // Name: src/common/socketiohandler.cpp | |
3 | // Purpose: implementation of wxFDIOHandler for wxSocket | |
4 | // Author: Angel Vidal, Lukasz Michalski | |
5 | // Created: 08.24.06 | |
6 | // RCS-ID: $Id$ | |
7 | // Copyright: (c) 2006 Angel vidal | |
8 | // (c) 2007 Vadim Zeitlin <vadim@wxwidgets.org> | |
9 | // License: wxWindows licence | |
10 | /////////////////////////////////////////////////////////////////////////////// | |
11 | ||
12 | // ============================================================================ | |
13 | // declarations | |
14 | // ============================================================================ | |
15 | ||
16 | // ---------------------------------------------------------------------------- | |
17 | // headers | |
18 | // ---------------------------------------------------------------------------- | |
19 | ||
20 | // for compilers that support precompilation, includes "wx.h". | |
21 | #include "wx/wxprec.h" | |
22 | ||
23 | #ifdef __BORLANDC__ | |
24 | #pragma hdrstop | |
25 | #endif | |
26 | ||
27 | #if wxUSE_SOCKETS && wxUSE_SELECT_DISPATCHER | |
28 | ||
29 | #include "wx/apptrait.h" | |
30 | #include "wx/private/socket.h" | |
31 | #include "wx/link.h" | |
32 | ||
33 | // ============================================================================ | |
34 | // implementation | |
35 | // ============================================================================ | |
36 | ||
37 | // ---------------------------------------------------------------------------- | |
38 | // wxSocketFDIOManager: socket manager using wxFDIODispatcher | |
39 | // ---------------------------------------------------------------------------- | |
40 | ||
41 | class wxSocketFDIOManager : public wxSocketFDBasedManager | |
42 | { | |
43 | public: | |
44 | virtual void Install_Callback(wxSocketImpl *socket, wxSocketNotify event); | |
45 | virtual void Uninstall_Callback(wxSocketImpl *socket, wxSocketNotify event); | |
46 | }; | |
47 | ||
48 | void wxSocketFDIOManager::Install_Callback(wxSocketImpl *socket_, | |
49 | wxSocketNotify event) | |
50 | { | |
51 | wxSocketImplUnix * const socket = static_cast<wxSocketImplUnix *>(socket_); | |
52 | ||
53 | const int fd = socket->m_fd; | |
54 | ||
55 | if ( fd == -1 ) | |
56 | return; | |
57 | ||
58 | const SocketDir d = GetDirForEvent(socket, event); | |
59 | ||
60 | wxFDIODispatcher * const dispatcher = wxFDIODispatcher::Get(); | |
61 | if ( !dispatcher ) | |
62 | return; | |
63 | ||
64 | FD(socket, d) = fd; | |
65 | ||
66 | // register it when it's used for the first time, update it if it had been | |
67 | // previously registered | |
68 | const bool alreadyRegistered = socket->HasAnyEnabledCallbacks(); | |
69 | ||
70 | socket->EnableCallback(d == FD_INPUT ? wxFDIO_INPUT : wxFDIO_OUTPUT); | |
71 | ||
72 | if ( alreadyRegistered ) | |
73 | dispatcher->ModifyFD(fd, socket, socket->GetEnabledCallbacks()); | |
74 | else | |
75 | dispatcher->RegisterFD(fd, socket, socket->GetEnabledCallbacks()); | |
76 | } | |
77 | ||
78 | void wxSocketFDIOManager::Uninstall_Callback(wxSocketImpl *socket_, | |
79 | wxSocketNotify event) | |
80 | { | |
81 | wxSocketImplUnix * const socket = static_cast<wxSocketImplUnix *>(socket_); | |
82 | ||
83 | const SocketDir d = GetDirForEvent(socket, event); | |
84 | ||
85 | const int fd = FD(socket, d); | |
86 | if ( fd == -1 ) | |
87 | return; | |
88 | ||
89 | FD(socket, d) = -1; | |
90 | ||
91 | const wxFDIODispatcherEntryFlags | |
92 | flag = d == FD_INPUT ? wxFDIO_INPUT : wxFDIO_OUTPUT; | |
93 | ||
94 | wxFDIODispatcher * const dispatcher = wxFDIODispatcher::Get(); | |
95 | if ( !dispatcher ) | |
96 | return; | |
97 | ||
98 | socket->DisableCallback(flag); | |
99 | ||
100 | if ( !socket->HasAnyEnabledCallbacks() ) | |
101 | dispatcher->UnregisterFD(fd); | |
102 | else | |
103 | dispatcher->ModifyFD(fd, socket, socket->GetEnabledCallbacks()); | |
104 | } | |
105 | ||
106 | // set the wxBase variable to point to our wxSocketManager implementation | |
107 | // | |
108 | // see comments in wx/apptrait.h for the explanation of why do we do it | |
109 | // like this | |
110 | static struct ManagerSetter | |
111 | { | |
112 | ManagerSetter() | |
113 | { | |
114 | static wxSocketFDIOManager s_manager; | |
115 | wxAppTraits::SetDefaultSocketManager(&s_manager); | |
116 | } | |
117 | } gs_managerSetter; | |
118 | ||
119 | ||
120 | // see the relative linker macro in socket.cpp | |
121 | wxFORCE_LINK_THIS_MODULE( socketiohandler ); | |
122 | ||
123 | #endif // wxUSE_SOCKETS |