xref: /haiku/src/apps/networkstatus/NetworkStatusView.cpp (revision 4c8e85b316c35a9161f5a1c50ad70bc91c83a76f) 
1 /*
2  * Copyright 2006-2013, Haiku, Inc. All rights reserved.
3  * Distributed under the terms of the MIT License.
4  *
5  * Authors:
6  *		Dario Casalinuovo
7  *		Axel Dörfler, axeld@pinc-software.de
8  *		Rene Gollent, rene@gollent.com
9  *		Hugo Santos, hugosantos@gmail.com
10  */
11 
12 
13 #include "NetworkStatusView.h"
14 
15 #include <algorithm>
16 #include <set>
17 #include <vector>
18 
19 #include <arpa/inet.h>
20 #include <net/if.h>
21 #include <stdio.h>
22 #include <stdlib.h>
23 #include <strings.h>
24 #include <sys/socket.h>
25 #include <sys/sockio.h>
26 #include <unistd.h>
27 
28 #include <AboutWindow.h>
29 #include <Alert.h>
30 #include <Application.h>
31 #include <Catalog.h>
32 #include <Bitmap.h>
33 #include <Deskbar.h>
34 #include <Dragger.h>
35 #include <Drivers.h>
36 #include <IconUtils.h>
37 #include <Locale.h>
38 #include <MenuItem.h>
39 #include <MessageRunner.h>
40 #include <NetworkInterface.h>
41 #include <NetworkRoster.h>
42 #include <PopUpMenu.h>
43 #include <Resources.h>
44 #include <Roster.h>
45 #include <String.h>
46 #include <TextView.h>
47 
48 #include "NetworkStatus.h"
49 #include "NetworkStatusIcons.h"
50 #include "RadioView.h"
51 #include "WirelessNetworkMenuItem.h"
52 
53 
54 #undef B_TRANSLATION_CONTEXT
55 #define B_TRANSLATION_CONTEXT "NetworkStatusView"
56 
57 
58 static const char *kStatusDescriptions[] = {
59 	B_TRANSLATE("Unknown"),
60 	B_TRANSLATE("No link"),
61 	B_TRANSLATE("No stateful configuration"),
62 	B_TRANSLATE("Configuring"),
63 	B_TRANSLATE("Ready")
64 };
65 
66 extern "C" _EXPORT BView *instantiate_deskbar_item(float maxWidth, float maxHeight);
67 
68 
69 const uint32 kMsgShowConfiguration = 'shcf';
70 const uint32 kMsgOpenNetworkPreferences = 'onwp';
71 const uint32 kMsgJoinNetwork = 'join';
72 
73 const uint32 kMinIconWidth = 16;
74 const uint32 kMinIconHeight = 16;
75 
76 
77 //	#pragma mark - NetworkStatusView
78 
79 
80 NetworkStatusView::NetworkStatusView(BRect frame, int32 resizingMode,
81 		bool inDeskbar)
82 	: BView(frame, kDeskbarItemName, resizingMode,
83 		B_WILL_DRAW | B_TRANSPARENT_BACKGROUND | B_FRAME_EVENTS),
84 	fInDeskbar(inDeskbar)
85 {
86 	_Init();
87 
88 	if (!inDeskbar) {
89 		// we were obviously added to a standard window - let's add a dragger
90 		frame.OffsetTo(B_ORIGIN);
91 		frame.top = frame.bottom - 7;
92 		frame.left = frame.right - 7;
93 		BDragger* dragger = new BDragger(frame, this,
94 			B_FOLLOW_RIGHT | B_FOLLOW_BOTTOM);
95 		AddChild(dragger);
96 	} else
97 		_Update();
98 }
99 
100 
101 NetworkStatusView::NetworkStatusView(BMessage* archive)
102 	: BView(archive),
103 	fInDeskbar(false)
104 {
105 	app_info info;
106 	if (be_app->GetAppInfo(&info) == B_OK
107 		&& !strcasecmp(info.signature, "application/x-vnd.Be-TSKB"))
108 		fInDeskbar = true;
109 
110 	_Init();
111 }
112 
113 
114 NetworkStatusView::~NetworkStatusView()
115 {
116 }
117 
118 
119 void
120 NetworkStatusView::_Init()
121 {
122 	for (int i = 0; i < kStatusCount; i++) {
123 		fTrayIcons[i] = NULL;
124 		fNotifyIcons[i] = NULL;
125 	}
126 
127 	_UpdateBitmaps();
128 }
129 
130 
131 void
132 NetworkStatusView::_UpdateBitmaps()
133 {
134 	for (int i = 0; i < kStatusCount; i++) {
135 		delete fTrayIcons[i];
136 		delete fNotifyIcons[i];
137 		fTrayIcons[i] = NULL;
138 		fNotifyIcons[i] = NULL;
139 	}
140 
141 	image_info info;
142 	if (our_image(info) != B_OK)
143 		return;
144 
145 	BFile file(info.name, B_READ_ONLY);
146 	if (file.InitCheck() < B_OK)
147 		return;
148 
149 	BResources resources(&file);
150 #ifdef HAIKU_TARGET_PLATFORM_HAIKU
151 	if (resources.InitCheck() < B_OK)
152 		return;
153 #endif
154 
155 	for (int i = 0; i < kStatusCount; i++) {
156 		const void* data = NULL;
157 		size_t size;
158 		data = resources.LoadResource(B_VECTOR_ICON_TYPE,
159 			kNetworkStatusNoDevice + i, &size);
160 		if (data != NULL) {
161 			// Scale main tray icon
162 			BBitmap* trayIcon = new BBitmap(Bounds(), B_RGBA32);
163 			if (trayIcon->InitCheck() == B_OK
164 				&& BIconUtils::GetVectorIcon((const uint8 *)data,
165 					size, trayIcon) == B_OK) {
166 				fTrayIcons[i] = trayIcon;
167 			} else
168 				delete trayIcon;
169 
170 			// Scale notification icon
171 			BBitmap* notifyIcon = new BBitmap(BRect(0, 0, 31, 31), B_RGBA32);
172 			if (notifyIcon->InitCheck() == B_OK
173 				&& BIconUtils::GetVectorIcon((const uint8 *)data,
174 					size, notifyIcon) == B_OK) {
175 				fNotifyIcons[i] = notifyIcon;
176 			} else
177 				delete notifyIcon;
178 		}
179 	}
180 }
181 
182 
183 void
184 NetworkStatusView::_Quit()
185 {
186 	if (fInDeskbar) {
187 		BDeskbar deskbar;
188 		deskbar.RemoveItem(kDeskbarItemName);
189 	} else
190 		be_app->PostMessage(B_QUIT_REQUESTED);
191 }
192 
193 
194 NetworkStatusView*
195 NetworkStatusView::Instantiate(BMessage* archive)
196 {
197 	if (!validate_instantiation(archive, "NetworkStatusView"))
198 		return NULL;
199 
200 	return new NetworkStatusView(archive);
201 }
202 
203 
204 status_t
205 NetworkStatusView::Archive(BMessage* archive, bool deep) const
206 {
207 	status_t status = BView::Archive(archive, deep);
208 	if (status == B_OK)
209 		status = archive->AddString("add_on", kSignature);
210 	if (status == B_OK)
211 		status = archive->AddString("class", "NetworkStatusView");
212 
213 	return status;
214 }
215 
216 
217 void
218 NetworkStatusView::AttachedToWindow()
219 {
220 	BView::AttachedToWindow();
221 
222 	SetViewColor(B_TRANSPARENT_COLOR);
223 
224 	start_watching_network(
225 		B_WATCH_NETWORK_INTERFACE_CHANGES | B_WATCH_NETWORK_LINK_CHANGES, this);
226 
227 	_Update();
228 }
229 
230 
231 void
232 NetworkStatusView::DetachedFromWindow()
233 {
234 	stop_watching_network(this);
235 }
236 
237 
238 void
239 NetworkStatusView::MessageReceived(BMessage* message)
240 {
241 	switch (message->what) {
242 		case B_NETWORK_MONITOR:
243 			_Update();
244 			break;
245 
246 		case kMsgShowConfiguration:
247 			_ShowConfiguration(message);
248 			break;
249 
250 		case kMsgOpenNetworkPreferences:
251 			_OpenNetworksPreferences();
252 			break;
253 
254 		case kMsgJoinNetwork:
255 		{
256 			const char* deviceName;
257 			const char* name;
258 			BNetworkAddress address;
259 			if (message->FindString("device", &deviceName) == B_OK
260 				&& message->FindString("name", &name) == B_OK
261 				&& message->FindFlat("address", &address) == B_OK) {
262 				BNetworkDevice device(deviceName);
263 				status_t status = device.JoinNetwork(address);
264 				if (status != B_OK) {
265 					BString text
266 						= B_TRANSLATE("Could not join wireless network:\n");
267 					text << strerror(status);
268 					BAlert* alert = new BAlert(name, text.String(),
269 						B_TRANSLATE("OK"), NULL, NULL, B_WIDTH_AS_USUAL,
270 						B_STOP_ALERT);
271 					alert->SetFlags(alert->Flags() | B_CLOSE_ON_ESCAPE);
272 					alert->Go(NULL);
273 				}
274 			}
275 			break;
276 		}
277 
278 		case B_ABOUT_REQUESTED:
279 			_AboutRequested();
280 			break;
281 
282 		case B_QUIT_REQUESTED:
283 			_Quit();
284 			break;
285 
286 		default:
287 			BView::MessageReceived(message);
288 			break;
289 	}
290 }
291 
292 
293 void
294 NetworkStatusView::FrameResized(float width, float height)
295 {
296 	_UpdateBitmaps();
297 	Invalidate();
298 }
299 
300 
301 void
302 NetworkStatusView::Draw(BRect updateRect)
303 {
304 	int32 status = kStatusUnknown;
305 	for (std::map<BString, int32>::const_iterator it
306 		= fInterfaceStatuses.begin(); it != fInterfaceStatuses.end(); ++it) {
307 		if (it->second > status)
308 			status = it->second;
309 	}
310 
311 	if (fTrayIcons[status] == NULL)
312 		return;
313 
314 	SetDrawingMode(B_OP_ALPHA);
315 	DrawBitmap(fTrayIcons[status]);
316 	SetDrawingMode(B_OP_COPY);
317 }
318 
319 
320 void
321 NetworkStatusView::_ShowConfiguration(BMessage* message)
322 {
323 	const char* name;
324 	if (message->FindString("interface", &name) != B_OK)
325 		return;
326 
327 	BNetworkInterface networkInterface(name);
328 	if (!networkInterface.Exists())
329 		return;
330 
331 	BString text(B_TRANSLATE("%ifaceName information:\n"));
332 	text.ReplaceFirst("%ifaceName", name);
333 
334 	size_t boldLength = text.Length();
335 
336 	int32 numAddrs = networkInterface.CountAddresses();
337 	for (int32 i = 0; i < numAddrs; i++) {
338 		BNetworkInterfaceAddress address;
339 		networkInterface.GetAddressAt(i, address);
340 		switch (address.Address().Family()) {
341 			case AF_INET:
342 				text << "\n" << B_TRANSLATE("IPv4 address:") << " "
343 					<< address.Address().ToString()
344 					<< "\n" << B_TRANSLATE("Broadcast:") << " "
345 					<< address.Broadcast().ToString()
346 					<< "\n" << B_TRANSLATE("Netmask:") << " "
347 					<< address.Mask().ToString()
348 					<< "\n";
349 				break;
350 			case AF_INET6:
351 				text << "\n" << B_TRANSLATE("IPv6 address:") << " "
352 					<< address.Address().ToString()
353 					<< "/" << address.Mask().PrefixLength()
354 					<< "\n";
355 				break;
356 			default:
357 				break;
358 		}
359 	}
360 
361 	BAlert* alert = new BAlert(name, text.String(), B_TRANSLATE("OK"));
362 	alert->SetFlags(alert->Flags() | B_CLOSE_ON_ESCAPE);
363 	BTextView* view = alert->TextView();
364 	BFont font;
365 
366 	view->SetStylable(true);
367 	view->GetFont(&font);
368 	font.SetFace(B_BOLD_FACE);
369 	view->SetFontAndColor(0, boldLength, &font);
370 
371 	alert->Go(NULL);
372 }
373 
374 
375 void
376 NetworkStatusView::MouseDown(BPoint point)
377 {
378 	BPopUpMenu* menu = new BPopUpMenu(B_EMPTY_STRING, false, false);
379 	menu->SetAsyncAutoDestruct(true);
380 	menu->SetFont(be_plain_font);
381 	BString wifiInterface;
382 	BNetworkDevice device;
383 
384 	if (!fInterfaceStatuses.empty()) {
385 		for (std::map<BString, int32>::const_iterator it
386 				= fInterfaceStatuses.begin(); it != fInterfaceStatuses.end();
387 				++it) {
388 			const BString& name = it->first;
389 
390 			// we only show network of the first wireless device we find
391 			if (wifiInterface.IsEmpty()) {
392 				device.SetTo(name);
393 				if (device.IsWireless())
394 					wifiInterface = name;
395 			}
396 		}
397 	}
398 
399 	// Add wireless networks, if any, first so that we can sort the menu
400 
401 	if (!wifiInterface.IsEmpty()) {
402 		std::set<BNetworkAddress> associated;
403 		BNetworkAddress address;
404 		wireless_network network;
405 		uint32 cookie = 0;
406 		while (device.GetNextAssociatedNetwork(cookie, address) == B_OK)
407 			associated.insert(address);
408 
409 		int32 wifiCount = 0;
410 		cookie = 0;
411 		while (device.GetNextNetwork(cookie, network) == B_OK) {
412 			BMessage* message = new BMessage(kMsgJoinNetwork);
413 			message->AddString("device", wifiInterface);
414 			message->AddString("name", network.name);
415 			message->AddFlat("address", &network.address);
416 
417 			BMenuItem* item = new WirelessNetworkMenuItem(network, message);
418 			menu->AddItem(item);
419 			wifiCount++;
420 			if (associated.find(network.address) != associated.end())
421 				item->SetMarked(true);
422 		}
423 
424 		if (wifiCount == 0) {
425 			BMenuItem* item = new BMenuItem(
426 				B_TRANSLATE("<no wireless networks found>"), NULL);
427 			item->SetEnabled(false);
428 			menu->AddItem(item);
429 		} else
430 			menu->SortItems(WirelessNetworkMenuItem::CompareSignalStrength);
431 
432 		menu->AddSeparatorItem();
433 	}
434 
435 	// add action menu items
436 
437 	menu->AddItem(new BMenuItem(B_TRANSLATE(
438 		"Open network preferences" B_UTF8_ELLIPSIS),
439 		new BMessage(kMsgOpenNetworkPreferences)));
440 
441 	if (fInDeskbar) {
442 		menu->AddItem(new BMenuItem(B_TRANSLATE("Quit"),
443 			new BMessage(B_QUIT_REQUESTED)));
444 	}
445 
446 	// Add wired interfaces to top of menu
447 	if (!fInterfaceStatuses.empty()) {
448 		int32 wiredCount = 0;
449 		for (std::map<BString, int32>::const_iterator it
450 				= fInterfaceStatuses.begin(); it != fInterfaceStatuses.end();
451 				++it) {
452 			const BString& name = it->first;
453 
454 			BString label = name;
455 			label += ": ";
456 			label += kStatusDescriptions[
457 				_DetermineInterfaceStatus(name.String())];
458 
459 			BMessage* info = new BMessage(kMsgShowConfiguration);
460 			info->AddString("interface", name.String());
461 			menu->AddItem(new BMenuItem(label.String(), info), wiredCount);
462 			wiredCount++;
463 		}
464 
465 		// add separator item between wired and wireless networks
466 		// (or between wired networks and actions if no wireless found)
467 		if (wiredCount > 0)
468 			menu->AddItem(new BSeparatorItem(), wiredCount);
469 	}
470 
471 	menu->SetTargetForItems(this);
472 
473 	ConvertToScreen(&point);
474 	menu->Go(point, true, true, true);
475 }
476 
477 
478 void
479 NetworkStatusView::_AboutRequested()
480 {
481 	BAboutWindow* window = new BAboutWindow(
482 		B_TRANSLATE_SYSTEM_NAME("NetworkStatus"), kSignature);
483 
484 	const char* authors[] = {
485 		"Axel Dörfler",
486 		"Hugo Santos",
487 		NULL
488 	};
489 
490 	window->AddCopyright(2007, "Haiku, Inc.");
491 	window->AddAuthors(authors);
492 
493 	window->Show();
494 }
495 
496 
497 int32
498 NetworkStatusView::_DetermineInterfaceStatus(
499 	const BNetworkInterface& interface)
500 {
501 	uint32 flags = interface.Flags();
502 
503 	if ((flags & IFF_LINK) == 0)
504 		return kStatusNoLink;
505 	if ((flags & (IFF_UP | IFF_LINK | IFF_CONFIGURING)) == IFF_LINK)
506 		return kStatusLinkNoConfig;
507 	if ((flags & IFF_CONFIGURING) == IFF_CONFIGURING)
508 		return kStatusConnecting;
509 	if ((flags & (IFF_UP | IFF_LINK)) == (IFF_UP | IFF_LINK))
510 		return kStatusReady;
511 
512 	return kStatusUnknown;
513 }
514 
515 
516 void
517 NetworkStatusView::_Update(bool force)
518 {
519 	BNetworkRoster& roster = BNetworkRoster::Default();
520 	BNetworkInterface interface;
521 	uint32 cookie = 0;
522 
523 	while (roster.GetNextInterface(&cookie, interface) == B_OK) {
524 		if ((interface.Flags() & IFF_LOOPBACK) == 0) {
525 			int32 oldStatus = kStatusUnknown;
526 			if (fInterfaceStatuses.find(interface.Name())
527 				!= fInterfaceStatuses.end()) {
528 				oldStatus = fInterfaceStatuses[interface.Name()];
529 			}
530 			int32 status = _DetermineInterfaceStatus(interface);
531 			if (oldStatus != status) {
532 				BNotification notification(B_INFORMATION_NOTIFICATION);
533 				notification.SetGroup(B_TRANSLATE("Network Status"));
534 				notification.SetTitle(interface.Name());
535 				notification.SetMessageID(interface.Name());
536 				notification.SetIcon(fNotifyIcons[status]);
537 				if (status == kStatusConnecting
538 					|| (status == kStatusReady
539 						&& oldStatus == kStatusConnecting)
540 					|| (status == kStatusNoLink
541 						&& oldStatus == kStatusReady)
542 					|| (status == kStatusNoLink
543 						&& oldStatus == kStatusConnecting)) {
544 					// A significant state change, raise notification.
545 					notification.SetContent(kStatusDescriptions[status]);
546 					notification.Send();
547 				}
548 				Invalidate();
549 			}
550 			fInterfaceStatuses[interface.Name()] = status;
551 		}
552 	}
553 }
554 
555 
556 void
557 NetworkStatusView::_OpenNetworksPreferences()
558 {
559 	status_t status = be_roster->Launch("application/x-vnd.Haiku-Network");
560 	if (status != B_OK && status != B_ALREADY_RUNNING) {
561 		BString errorMessage(B_TRANSLATE("Launching the network preflet "
562 			"failed.\n\nError: "));
563 		errorMessage << strerror(status);
564 		BAlert* alert = new BAlert("launch error", errorMessage.String(),
565 			B_TRANSLATE("OK"));
566 		alert->SetFlags(alert->Flags() | B_CLOSE_ON_ESCAPE);
567 
568 		// asynchronous alert in order to not block replicant host application
569 		alert->Go(NULL);
570 	}
571 }
572 
573 
574 //	#pragma mark -
575 
576 
577 extern "C" _EXPORT BView *
578 instantiate_deskbar_item(float maxWidth, float maxHeight)
579 {
580 	return new NetworkStatusView(BRect(0, 0, maxHeight - 1, maxHeight - 1),
581 		B_FOLLOW_LEFT | B_FOLLOW_TOP, true);
582 }
583