HelenOS sources
This source file includes following definitions.
- print_syntax
- nic_addr_format
- get_wifi_list
- get_wifi_by_index
- wifi_list
- wifi_connect
- wifi_disconnect
- wifi_scan
- main
#include <ieee80211_iface.h>
#include <inet/inetcfg.h>
#include <inet/dhcp.h>
#include <errno.h>
#include <stdio.h>
#include <stdlib.h>
#include <str.h>
#include <str_error.h>
#include <loc.h>
#define NAME "wifi_supplicant"
#define enum_name(name_arr, i) \
((i < 0) ? "NA" : name_arr[i])
static const char *ieee80211_security_type_strs[] = {
"OPEN", "WEP", "WPA", "WPA2"
};
static const char *ieee80211_security_alg_strs[] = {
"WEP40", "WEP104", "CCMP", "TKIP"
};
static const char *ieee80211_security_auth_strs[] = {
"PSK", "8021X"
};
static void print_syntax(void)
{
printf("syntax:\n");
printf("\t" NAME " [<cmd> [<args...>]]\n");
printf("\t<cmd> is:\n");
printf("\tlist - list wifi devices in <index>: <name> format\n");
printf("\tscan <index> [-n] - output scan results (force scan "
"immediately)\n");
printf("\tconnect <index> <ssid_prefix> [<password>] - connect to "
"network\n");
printf("\tdisconnect <index> - disconnect from network\n");
}
static void nic_addr_format(nic_address_t *addr, char *out, size_t out_size)
{
snprintf(out, out_size, "%02x:%02x:%02x:%02x:%02x:%02x",
addr->address[0], addr->address[1], addr->address[2],
addr->address[3], addr->address[4], addr->address[5]);
}
static errno_t get_wifi_list(service_id_t **wifis, size_t *count)
{
category_id_t wifi_cat;
errno_t rc = loc_category_get_id("ieee80211", &wifi_cat, 0);
if (rc != EOK) {
printf("Error resolving category 'ieee80211'.\n");
return rc;
}
rc = loc_category_get_svcs(wifi_cat, wifis, count);
if (rc != EOK) {
printf("Error getting list of WIFIs.\n");
return rc;
}
return EOK;
}
static async_sess_t *get_wifi_by_index(size_t i)
{
service_id_t *wifis = NULL;
size_t count;
errno_t rc = get_wifi_list(&wifis, &count);
if (rc != EOK) {
printf("Error fetching wifi list.\n");
return NULL;
}
if (i >= count) {
printf("Invalid wifi index.\n");
free(wifis);
return NULL;
}
async_sess_t *sess =
loc_service_connect(wifis[i], INTERFACE_DDF, 0);
if (sess == NULL) {
printf("Error connecting to service.\n");
free(wifis);
return NULL;
}
return sess;
}
static errno_t wifi_list(void)
{
service_id_t *wifis = NULL;
size_t count;
errno_t rc = get_wifi_list(&wifis, &count);
if (rc != EOK) {
printf("Error fetching wifi list: %s\n", str_error(rc));
return EINVAL;
}
printf("[Index]: [Service Name]\n");
for (size_t i = 0; i < count; i++) {
char *svc_name;
rc = loc_service_get_name(wifis[i], &svc_name);
if (rc != EOK) {
printf("Error getting service name: %s\n", str_error(rc));
free(wifis);
return rc;
}
printf("%zu: %s\n", i, svc_name);
free(svc_name);
}
return EOK;
}
static errno_t wifi_connect(uint32_t index, char *ssid_start, char *password)
{
assert(ssid_start);
async_sess_t *sess = get_wifi_by_index(index);
if (sess == NULL) {
printf("Specified WIFI doesn't exist or cannot connect to "
"it.\n");
return EINVAL;
}
errno_t rc = ieee80211_disconnect(sess);
if (rc != EOK) {
if (rc == EREFUSED)
printf("Device is not ready yet.\n");
else
printf("Error when disconnecting device: %s\n",
str_error(rc));
return rc;
}
rc = ieee80211_connect(sess, ssid_start, password);
if (rc != EOK) {
if (rc == EREFUSED)
printf("Device is not ready yet.\n");
else if (rc == ETIMEOUT)
printf("Timeout when authenticating to network.\n");
else if (rc == ENOENT)
printf("Given SSID not in scan results.\n");
else
printf("Error when connecting to network: %s\n",
str_error(rc));
return rc;
}
printf("Successfully connected to network!\n");
return EOK;
}
static errno_t wifi_disconnect(uint32_t index)
{
async_sess_t *sess = get_wifi_by_index(index);
if (sess == NULL) {
printf("Specified WIFI doesn't exist or cannot connect to "
"it.\n");
return EINVAL;
}
errno_t rc = ieee80211_disconnect(sess);
if (rc != EOK) {
if (rc == EREFUSED)
printf("Device is not ready yet.\n");
else if (rc == EINVAL)
printf("Not connected to any WiFi network.\n");
else
printf("Error when disconnecting from network: %s\n",
str_error(rc));
return rc;
}
printf("Successfully disconnected.\n");
return EOK;
}
static errno_t wifi_scan(uint32_t index, bool now)
{
async_sess_t *sess = get_wifi_by_index(index);
if (sess == NULL) {
printf("Specified WIFI doesn't exist or cannot connect to "
"it.\n");
return EINVAL;
}
ieee80211_scan_results_t scan_results;
errno_t rc = ieee80211_get_scan_results(sess, &scan_results, now);
if (rc != EOK) {
if (rc == EREFUSED)
printf("Device is not ready yet.\n");
else
printf("Failed to fetch scan results: %s\n",
str_error(rc));
return rc;
}
if (scan_results.length == 0)
return EOK;
printf("%16.16s %17s %4s %5s %5s %7s %7s\n",
"SSID", "MAC", "CHAN", "TYPE", "AUTH", "UNI-ALG", "GRP-ALG");
for (uint8_t i = 0; i < scan_results.length; i++) {
ieee80211_scan_result_t result = scan_results.results[i];
char mac_addr_hex[18];
nic_addr_format(&result.bssid, mac_addr_hex, 18);
printf("%16.16s %17s %4d %5s %5s %7s %7s\n",
result.ssid, mac_addr_hex,
result.channel,
enum_name(ieee80211_security_type_strs, result.security.type),
enum_name(ieee80211_security_auth_strs, result.security.auth),
enum_name(ieee80211_security_alg_strs, result.security.pair_alg),
enum_name(ieee80211_security_alg_strs, result.security.group_alg));
}
return EOK;
}
int main(int argc, char *argv[])
{
errno_t rc = inetcfg_init();
if (rc != EOK) {
printf("%s: Failed connecting to inetcfg service: %s.\n",
NAME, str_error(rc));
return 1;
}
rc = dhcp_init();
if (rc != EOK) {
printf("%s: Failed connecting to dhcp service: %s.\n",
NAME, str_error(rc));
return 1;
}
if (argc == 2) {
if (!str_cmp(argv[1], "list"))
return wifi_list();
} else if (argc > 2) {
uint32_t index;
rc = str_uint32_t(argv[2], NULL, 10, false, &index);
if (rc != EOK) {
printf("%s: Invalid argument.\n", NAME);
print_syntax();
return EINVAL;
}
if (!str_cmp(argv[1], "scan")) {
bool now = false;
if (argc > 3)
if (!str_cmp(argv[3], "-n"))
now = true;
return wifi_scan(index, now);
} else if (!str_cmp(argv[1], "connect")) {
char *pass = NULL;
if (argc > 3) {
if (argc > 4)
pass = argv[4];
return wifi_connect(index, argv[3], pass);
}
} else if (!str_cmp(argv[1], "disconnect"))
return wifi_disconnect(index);
}
print_syntax();
return EOK;
}
HelenOS homepage, sources at GitHub