client (Win): including WSL disk image in BOINC disk usage

If using boinc-buda-runner, find its disk image size
and count as BOINC client disk usage.

Author: davidpanderson

client/client_state.cpp

@@ -258,7 +258,7 @@ void CLIENT_STATE::show_host_info() {
         }
     } else {
         msg_printf(NULL, MSG_INFO, "Usable WSL distros:");
-        for (auto& wsl : host_info.wsl_distros.distros) {
+        for (WSL_DISTRO &wsl : host_info.wsl_distros.distros) {
             msg_printf(NULL, MSG_INFO,
                 "-   %s (WSL %d)%s",
                 wsl.distro_name.c_str(),
@@ -290,6 +290,13 @@ void CLIENT_STATE::show_host_info() {
                 msg_printf(NULL, MSG_INFO, "-      BOINC WSL distro version %d",
                     wsl.boinc_buda_runner_version
                 );
+                double size;
+                int retval = dir_size_alloc(wd.base_path.c_str(), size);
+                if (!retval) {
+                    char buf[256];
+                    nbytes_to_string(size, 0, buf, sizeof(buf));
+                    msg_printf(NULL, MSG_INFO, "-      Disk usage: %s", buf);
+                }
             }
         }
     }

client/cs_prefs.cpp

@@ -86,12 +86,17 @@ int CLIENT_STATE::get_disk_usages() {
 
     client_disk_usage = 0;
     total_disk_usage = 0;
+
+    // disk usage in project directories
+    //
     for (PROJECT *p: projects) {
         p->disk_usage = 0;
         retval = dir_size_alloc(p->project_dir(), size);
         if (!retval) p->disk_usage = size;
     }
 
+    // disk usage in slot directories
+    //
     for (i=0; i<active_tasks.active_tasks.size(); i++) {
         ACTIVE_TASK* atp = active_tasks.active_tasks[i];
         get_slot_dir(atp->slot, buf, sizeof(buf));
@@ -102,11 +107,29 @@ int CLIENT_STATE::get_disk_usages() {
     for (PROJECT *p: projects) {
         total_disk_usage += p->disk_usage;
     }
+
+    // disk usage top level of BOINC data dir: XML and log files
+    //
     retval = dir_size_alloc(".", size, false);
     if (!retval) {
         client_disk_usage = size;
         total_disk_usage += size;
     }
+
+#ifdef _WIN64
+    // Win: if using boinc-buda-runner WSL distro,
+    // include its disk image size
+    //
+    WSL_DISTRO *wd = host_info.wsl_distros.find_docker();
+    if (wd) {
+        retval = dir_size_alloc(wd.base_path.c_str(), size);
+        if (!retval) {
+            client_disk_usage = size;
+            total_disk_usage += size;
+        }
+    }
+#endif
+
     return 0;
 }
 

client/hostinfo_wsl.cpp

@@ -127,6 +127,18 @@ int get_all_distros(WSL_DISTROS& distros) {
             if (!strcmp(wsl_guid, default_wsl_guid)) {
                 distro.is_default = true;
             }
+
+            // if BOINC distro, get its data dir so we can find disk usage
+            //
+            if (distro.distro_name == BOINC_WSL_DISTRO_NAME) {
+                ret = RegQueryValueEx(hSubKey, "BasePath", NULL, NULL,
+                    (LPBYTE)wsl_name, &wsl_name_len
+                );
+                if (ret == ERROR_SUCCESS) {
+                    distro.base_path = wsl_name;
+                }
+            }
+
             distros.distros.push_back(distro);
         }
         RegCloseKey(hSubKey);

lib/wslinfo.h

@@ -56,6 +56,8 @@ struct WSL_DISTRO {
     DOCKER_TYPE docker_compose_type;
     int boinc_buda_runner_version;
         // if this distro is boinc_buda_runner, the version
+    std::string base_path;
+        // the dir where the disk image (.vhdx file) is stored
 
     WSL_DISTRO(){
         clear();