00001
00002
00003
00004
00005
00006
00007
00008
00009
00010
00011
00012
00013
00014
00015
00016
00017
00018
00019
00020
00021
00022
00023
00024
00025
00026
00027
00028
00029
00030
00031
00032
00033
00034
00035
00036
00037
00038
00039 #include "NetworkChangeDetection.h"
00040
00041 namespace ariba {
00042 namespace communication {
00043
00044 use_logging_cpp(NetworkChangeDetection);
00045 SystemEventType NetworkChangeDetectionEventType("NetworkChangeDetectionEventType");
00046
00047 NetworkChangeDetection::NetworkChangeDetection()
00048 : running( false ), monitoringThread( NULL ), routingSocket( -1 ){
00049 startMonitoring();
00050 }
00051
00052 NetworkChangeDetection::~NetworkChangeDetection(){
00053 stopMonitoring();
00054 }
00055
00056 void NetworkChangeDetection::registerNotification(NetworkChangeInterface* callback){
00057
00058 RegistrationList::iterator i = find(
00059 registrations.begin(), registrations.end(), callback );
00060
00061 if( i == registrations.end() )
00062 registrations.push_back( callback );
00063 }
00064
00065 void NetworkChangeDetection::unregisterNotification(NetworkChangeInterface* callback){
00066
00067 RegistrationList::iterator i = find(
00068 registrations.begin(), registrations.end(), callback );
00069
00070 if( i != registrations.end() )
00071 registrations.erase( i );
00072 }
00073
00074 void NetworkChangeDetection::startMonitoring(){
00075
00076 logging_debug( "starting network change monitoring ..." );
00077
00078 running = true;
00079 monitoringThread = new boost::thread(
00080 boost::bind(&NetworkChangeDetection::monitoringThreadFunc, this) );
00081 }
00082
00083 void NetworkChangeDetection::stopMonitoring(){
00084
00085 logging_debug( "stopping network change monitoring ..." );
00086
00087
00088 running = false;
00089
00090 shutdown( routingSocket, SHUT_RDWR );
00091 close( routingSocket );
00092
00093
00094 monitoringThread->join();
00095
00096 delete monitoringThread;
00097 monitoringThread = NULL;
00098 }
00099
00100 void NetworkChangeDetection::monitoringThreadFunc(NetworkChangeDetection* obj){
00101
00102
00103
00104
00105
00106
00107
00108
00109
00110
00111
00112
00113
00114
00115
00116
00117
00118
00119
00120
00121
00122 obj->routingSocket = socket( PF_NETLINK, SOCK_DGRAM, NETLINK_ROUTE );
00123 if( obj->routingSocket < 0 ){
00124 logging_error("could not connect to routing socket: " +
00125 string(strerror(errno)));
00126 return;
00127 }
00128
00129
00130
00131
00132
00133
00134
00135
00136 struct timeval tv;
00137 tv.tv_sec = 1;
00138 tv.tv_usec = 0;
00139
00140 setsockopt( obj->routingSocket, SOL_SOCKET, SO_RCVTIMEO, &tv, sizeof(tv) );
00141
00142
00143
00144
00145
00146
00147
00148 struct sockaddr_nl addr;
00149 memset( &addr, 0, sizeof(addr) );
00150 addr.nl_family = AF_NETLINK;
00151 addr.nl_groups = RTMGRP_IPV4_IFADDR | RTMGRP_IPV6_IFADDR | RTMGRP_LINK;
00152 addr.nl_pid = getpid();
00153
00154 int ret = bind( obj->routingSocket, (struct sockaddr*)&addr, sizeof(addr) );
00155 if( ret < 0 ){
00156 close( obj->routingSocket );
00157 logging_error( "could not bind routing socket: " + string(strerror(errno)) );
00158 return;
00159 }
00160
00161
00162
00163
00164
00165 logging_debug( "network change monitoring started" );
00166
00167 while( obj->running ){
00168
00169 char buffer[1024];
00170 struct nlmsghdr* header = (struct nlmsghdr*)buffer;
00171
00172 int bytesRead = recv( obj->routingSocket, &buffer, sizeof(buffer), 0 );
00173
00174 if( bytesRead < 0 ){
00175
00176
00177 if( errno == EWOULDBLOCK ) continue;
00178
00179
00180 if( errno == EBADF ) return;
00181
00182
00183 if( errno == EAGAIN || errno == EINTR ) return;
00184
00185
00186 logging_error( "could not read from routing socket: " +
00187 string(strerror(errno)) );
00188 break;
00189 }
00190
00191 for( ; bytesRead > 0; header = NLMSG_NEXT(header, bytesRead)) {
00192 if ( !NLMSG_OK(header, bytesRead) ||
00193 (size_t) bytesRead < sizeof(struct nlmsghdr) ||
00194 (size_t) bytesRead < header->nlmsg_len) {
00195 continue;
00196 }
00197
00198
00199
00200
00201
00202
00203
00204
00205
00206
00207
00208 NetworkChangeInterface::NetworkChangeInfo changeInfo;
00209
00210 switch( header->nlmsg_type ){
00211
00212
00213
00214
00215
00216
00217 case RTM_NEWLINK:
00218 case RTM_DELLINK:
00219
00220 changeInfo = obj->extractInterfaceEvent( header );
00221 break;
00222
00223
00224
00225
00226
00227
00228 case RTM_NEWADDR:
00229 case RTM_DELADDR:
00230
00231 changeInfo = obj->extractAddressEvent( header );
00232 break;
00233
00234
00235
00236
00237
00238
00239 default: break;
00240
00241 }
00242
00243 logging_info( "network change detected: [" << changeInfo << "]" );
00244
00245
00246
00247
00248
00249 SystemQueue::instance().scheduleEvent(
00250 SystemEvent( obj, NetworkChangeDetectionEventType,
00251 new NetworkChangeInterface::NetworkChangeInfo(changeInfo)));
00252
00253 }
00254 }
00255
00256 logging_debug( "network change monitoring stopped" );
00257 }
00258
00259 NetworkChangeInterface::NetworkChangeInfo NetworkChangeDetection::extractInterfaceEvent(struct nlmsghdr* header){
00260 NetworkChangeInterface::NetworkChangeInfo changeInfo;
00261
00262
00263
00264
00265
00266
00267
00268 if( header->nlmsg_type == RTM_NEWLINK ){
00269 logging_debug("network change message RTM_NEWLINK");
00270 }else if( header->nlmsg_type == RTM_DELLINK ){
00271 logging_debug("network change message RTM_DELLINK");
00272 }
00273
00274 if( header->nlmsg_len < NLMSG_LENGTH(sizeof(struct ifinfomsg)) ){
00275 logging_error( "netlink packet soo small for ifinfomsg" );
00276 return changeInfo;
00277 }
00278
00279
00280
00281
00282
00283
00284 struct ifinfomsg* detailIface = (struct ifinfomsg*)NLMSG_DATA(header);
00285 changeInfo.interface = networkInformation.getInterface( detailIface->ifi_index );
00286
00287
00288
00289
00290
00291 if( changeInfo.interface.isUp && changeInfo.interface.isRunning )
00292 changeInfo.type = NetworkChangeInterface::EventTypeInterfaceUp;
00293 else
00294 changeInfo.type = NetworkChangeInterface::EventTypeInterfaceDown;
00295
00296
00297 return changeInfo;
00298 }
00299
00300 NetworkChangeInterface::NetworkChangeInfo NetworkChangeDetection::extractAddressEvent(struct nlmsghdr* header){
00301 NetworkChangeInterface::NetworkChangeInfo changeInfo;
00302
00303
00304
00305
00306
00307
00308 if( header->nlmsg_type == RTM_NEWADDR ){
00309 logging_debug("network change message RTM_NEWADDR");
00310 changeInfo.type = NetworkChangeInterface::EventTypeAddressNew;
00311 }else if( header->nlmsg_type == RTM_DELADDR ){
00312 logging_debug("network change message RTM_DELADDR");
00313 changeInfo.type = NetworkChangeInterface::EventTypeAddressDelete;
00314 }
00315
00316 struct ifaddrmsg* detailAddr = (struct ifaddrmsg*)NLMSG_DATA(header);
00317 changeInfo.interface = networkInformation.getInterface( detailAddr->ifa_index );
00318
00319 return changeInfo;
00320 }
00321
00322 void NetworkChangeDetection::handleSystemEvent(const SystemEvent& event){
00323
00324 NetworkChangeInterface::NetworkChangeInfo* changeInfo = event.getData<NetworkChangeInterface::NetworkChangeInfo>();
00325 const NetworkChangeDetection* obj = (const NetworkChangeDetection*)event.getListener();
00326
00327 RegistrationList::const_iterator i = obj->registrations.begin();
00328 RegistrationList::const_iterator iend = obj->registrations.end();
00329
00330 for( ; i != iend; i++ )
00331 (*i)->onNetworkChange(*changeInfo);
00332
00333 delete changeInfo;
00334 }
00335
00336 }}