Retrieve list of attributes from a nested attribute

[android-x86/system-connectivity-wificond.git] / main.cpp

diff --git a/main.cpp b/main.cpp
index 5a982b4..401c0a4 100644 (file)
--- a/main.cpp
+++ b/main.cpp
@@ -15,6 +15,7 @@
  */
 
 #include <unistd.h>
+#include <sys/capability.h>
 
 #include <csignal>
 #include <memory>
@@ -25,6 +26,8 @@
 #include <binder/IServiceManager.h>
 #include <binder/ProcessState.h>
 #include <cutils/properties.h>
+#include <libminijail.h>
+#include <private/android_filesystem_config.h>
 #include <utils/String16.h>
 #include <wifi_hal/driver_tool.h>
 #include <wifi_system/hal_tool.h>
@@ -32,12 +35,17 @@
 
 #include "wificond/ipc_constants.h"
 #include "wificond/looper_backed_event_loop.h"
+#include "wificond/net/netlink_manager.h"
+#include "wificond/net/netlink_utils.h"
+#include "wificond/scanning/scan_utils.h"
 #include "wificond/server.h"
 
 using android::net::wifi::IWificond;
 using android::wifi_hal::DriverTool;
 using android::wifi_system::HalTool;
+using android::wifi_system::HostapdManager;
 using android::wifi_system::InterfaceTool;
+using android::wifi_system::SupplicantManager;
 using android::wificond::ipc_constants::kDevModePropertyKey;
 using android::wificond::ipc_constants::kDevModeServiceName;
 using android::wificond::ipc_constants::kServiceName;
@@ -98,6 +106,27 @@ void RegisterServiceOrCrash(const android::sp<android::IBinder>& service) {
            android::NO_ERROR);
 }
 
+void DoPrivilegedSetupOrCrash() {
+  // take ownership of the magic firmware change path
+  // TODO: Return CHECK: b/31225859
+  if (chown(DriverTool::kFirmwareReloadPath, AID_WIFI, AID_WIFI) != 0) {
+    LOG(INFO) << "Error changing ownership of '"
+              << DriverTool::kFirmwareReloadPath
+              << "' to wifi:wifi, (" << strerror(errno) << ")";
+  }
+}
+
+void DropPrivilegesOrCrash() {
+  minijail* j = minijail_new();
+  CHECK(minijail_change_user(j, "wifi") == 0);
+  CHECK(minijail_change_group(j, "wifi") == 0);
+  minijail_use_caps(j,
+                    CAP_TO_MASK(CAP_NET_ADMIN) |
+                    CAP_TO_MASK(CAP_NET_RAW));
+  minijail_enter(j);
+  minijail_destroy(j);
+}
+
 }  // namespace
 
 void OnBinderReadReady(int fd) {
@@ -108,6 +137,9 @@ int main(int argc, char** argv) {
   android::base::InitLogging(argv, android::base::LogdLogger(android::base::SYSTEM));
   LOG(INFO) << "wificond is starting up...";
 
+  DoPrivilegedSetupOrCrash();
+  DropPrivilegesOrCrash();
+
   unique_ptr<android::wificond::LooperBackedEventLoop> event_dispatcher(
       new android::wificond::LooperBackedEventLoop());
   ScopedSignalHandler scoped_signal_handler(event_dispatcher.get());
@@ -118,12 +150,21 @@ int main(int argc, char** argv) {
       android::wificond::EventLoop::kModeInput,
       &OnBinderReadReady)) << "Failed to watch binder FD";
 
+  android::wificond::NetlinkManager netlink_manager(event_dispatcher.get());
+  CHECK(netlink_manager.Start()) << "Failed to start netlink manager";
+  android::wificond::NetlinkUtils netlink_utils(&netlink_manager);
+  android::wificond::ScanUtils scan_utils(&netlink_manager);
 
-  android::sp<android::IBinder> server = new android::wificond::Server(
+  unique_ptr<android::wificond::Server> server(new android::wificond::Server(
       unique_ptr<HalTool>(new HalTool),
       unique_ptr<InterfaceTool>(new InterfaceTool),
-      unique_ptr<DriverTool>(new DriverTool));
-  RegisterServiceOrCrash(server);
+      unique_ptr<DriverTool>(new DriverTool),
+      unique_ptr<SupplicantManager>(new SupplicantManager()),
+      unique_ptr<HostapdManager>(new HostapdManager()),
+      &netlink_utils,
+      &scan_utils));
+  server->CleanUpSystemState();
+  RegisterServiceOrCrash(server.get());
 
   event_dispatcher->Poll();
   LOG(INFO) << "wificond is about to exit";