arozos/disk.go

package main

/*
	ArOZ Online Disk Service Endpoint Handler

	This is a module to provide access to the disk services
*/

import (
	"net/http"

	"imuslab.com/arozos/mod/disk/diskcapacity"
	"imuslab.com/arozos/mod/disk/diskmg"
	diskspace "imuslab.com/arozos/mod/disk/diskspace"
	"imuslab.com/arozos/mod/disk/raid"
	smart "imuslab.com/arozos/mod/disk/smart"
	sortfile "imuslab.com/arozos/mod/disk/sortfile"
	prout "imuslab.com/arozos/mod/prouter"
	"imuslab.com/arozos/mod/utils"
)

func RAIDServiceInit() {
	/*
		RAID Management

		Handle physical disk RAID for more NAS OS like experience
	*/

	if *allow_hardware_management {
		rm, err := raid.NewRaidManager(raid.Options{})
		if err == nil {
			raidManager = rm

		} else {
			//Unable to start RAID manager. Skip it.
			systemWideLogger.PrintAndLog("RAID", "Unable to start RAID manager", err)
		}

		/* Flush mdadm RAID */
		if raidManager != nil {
			if !*skip_mdadm_reload {
				err := raidManager.FlushReload()
				if err != nil {
					systemWideLogger.PrintAndLog("RAID", "mdadm reload failed: "+err.Error(), err)
				}
			}
		}
	}
}

func DiskServiceInit() {
	//Register Disk Utilities under System Setting
	//Disk info are only viewable by administrator
	router := prout.NewModuleRouter(prout.RouterOption{
		ModuleName:  "System Setting",
		AdminOnly:   false,
		UserHandler: userHandler,
		DeniedHandler: func(w http.ResponseWriter, r *http.Request) {
			utils.SendErrorResponse(w, "Permission Denied")
		},
	})

	//Anyone logged in can load router
	authRouter := prout.NewModuleRouter(prout.RouterOption{
		AdminOnly:   false,
		UserHandler: userHandler,
		DeniedHandler: func(w http.ResponseWriter, r *http.Request) {
			utils.SendErrorResponse(w, "Permission Denied")
		},
	})

	//Disk Space Display endpoint
	router.HandleFunc("/system/disk/space/list", diskspace.HandleDiskSpaceList)

	//Handle Virtual Disk Properties display endpoints
	dc := diskcapacity.NewCapacityResolver(userHandler)
	router.HandleFunc("/system/disk/space/resolve", dc.HandleCapacityResolving)
	authRouter.HandleFunc("/system/disk/space/tmp", dc.HandleTmpCapacityResolving)

	//New Large File Scanner
	lfs := sortfile.NewLargeFileScanner(userHandler)
	router.HandleFunc("/system/disk/space/largeFiles", lfs.HandleLargeFileList)

	//Register settings
	registerSetting(settingModule{
		Name:         "Space Finder",
		Desc:         "Reclaim Storage Space on Disks",
		IconPath:     "SystemAO/disk/space/img/small_icon.png",
		Group:        "Disk",
		StartDir:     "SystemAO/disk/space/finder.html",
		RequireAdmin: false,
	})

	if *allow_hardware_management {
		//Displaying remaining space on disk, only enabled when allow hardware is true
		registerSetting(settingModule{
			Name:         "Disk Space",
			Desc:         "System Storage Space on Disks",
			IconPath:     "SystemAO/disk/space/img/small_icon.png",
			Group:        "Disk",
			StartDir:     "SystemAO/disk/space/diskspace.html",
			RequireAdmin: false,
		})
	}

	//Register Disk SMART services
	if sudo_mode {
		//Create a new admin router
		adminRouter := prout.NewModuleRouter(prout.RouterOption{
			ModuleName:  "System Setting",
			AdminOnly:   true,
			UserHandler: userHandler,
			DeniedHandler: func(w http.ResponseWriter, r *http.Request) {
				utils.SendErrorResponse(w, "Permission Denied")
			},
		})

		/*
			SMART Listener
			Handle disk SMART and disk information

			See disk/SMART for more information
		*/
		if *allow_hardware_management {
			smartListener, err := smart.NewSmartListener()
			if err != nil {
				//Listener creation failed
				systemWideLogger.PrintAndLog("Disk", "Failed to create SMART listener: "+err.Error(), err)
			} else {
				//Listener created. Register endpoints
				registerSetting(settingModule{
					Name:         "Disk SMART",
					Desc:         "HardDisk Health Checking",
					IconPath:     "SystemAO/disk/smart/img/small_icon.png",
					Group:        "Disk",
					StartDir:     "SystemAO/disk/smart/smart.html",
					RequireAdmin: true,
				})

				authRouter.HandleFunc("/system/disk/smart/getSMART", smartListener.GetSMART)
			}

			/*
				RAID Manager endpoints
			*/
			if raidManager != nil {
				//Register endpoints and settings for this host
				registerSetting(settingModule{
					Name:         "RAID",
					Desc:         "Providing basic mdadm features",
					IconPath:     "SystemAO/disk/raid/img/small_icon.png",
					Group:        "Disk",
					StartDir:     "SystemAO/disk/raid/index.html",
					RequireAdmin: true,
				})

				/* RAID storage pool function */
				adminRouter.HandleFunc("/system/disk/raid/list", raidManager.HandleListRaidDevices)
				adminRouter.HandleFunc("/system/disk/raid/new", raidManager.HandleCreateRAIDDevice)
				adminRouter.HandleFunc("/system/disk/raid/remove", raidManager.HandleRemoveRaideDevice)
				adminRouter.HandleFunc("/system/disk/raid/format", raidManager.HandleFormatRaidDevice)
				adminRouter.HandleFunc("/system/disk/raid/detail", raidManager.HandleLoadArrayDetail)
				adminRouter.HandleFunc("/system/disk/raid/devinfo", raidManager.HandlListChildrenDeviceInfo)
				adminRouter.HandleFunc("/system/disk/raid/removeMemeber", func(w http.ResponseWriter, r *http.Request) {
					/*
						Remove a member disk from RAID array
						Require double check for credentials
					*/
					userinfo, err := userHandler.GetUserInfoFromRequest(w, r)
					if err != nil {
						w.WriteHeader(http.StatusUnauthorized)
						w.Write([]byte("401 Unauthorized"))
						return
					}

					if !userinfo.IsAdmin() {
						utils.SendErrorResponse(w, "Permission Denied")
						return
					}

					//Double check password for this user
					password, err := utils.PostPara(r, "password")
					if err != nil {
						utils.SendErrorResponse(w, "Password Incorrect")
						return
					}

					passwordCorrect, rejectionReason := authAgent.ValidateUsernameAndPasswordWithReason(userinfo.Username, password)
					if !passwordCorrect {
						utils.SendErrorResponse(w, rejectionReason)
						return
					}

					//OK! Handle the remaining operations
					raidManager.HandleRemoveDiskFromRAIDVol(w, r)
				})

				/* Device Management functions */
				adminRouter.HandleFunc("/system/disk/devices/list", raidManager.HandleListUsableDevices)
				adminRouter.HandleFunc("/system/disk/devices/model", raidManager.HandleResolveDiskModelLabel)

				/* Advance functions*/
				adminRouter.HandleFunc("/system/disk/raid/assemble", raidManager.HandleRaidDevicesAssemble)
				adminRouter.HandleFunc("/system/disk/raid/reload", raidManager.HandleMdadmFlushReload)
			}
		}

		/*
			Disk Manager Initialization
			See disk/diskmg.go for more details

			For setting register, see setting.advance.go
		*/

		if *allow_hardware_management {
			authRouter.HandleFunc("/system/disk/diskmg/view", diskmg.HandleView)
			adminRouter.HandleFunc("/system/disk/diskmg/platform", diskmg.HandlePlatform)
			adminRouter.HandleFunc("/system/disk/diskmg/mount", func(w http.ResponseWriter, r *http.Request) {
				//Mount option require passing in all filesystem handlers
				allFsh := GetAllLoadedFsh()
				diskmg.HandleMount(w, r, allFsh)
			})
			adminRouter.HandleFunc("/system/disk/diskmg/format", func(w http.ResponseWriter, r *http.Request) {
				//Check if request are made in POST mode
				if r.Method != http.MethodPost {
					w.WriteHeader(http.StatusMethodNotAllowed)
					w.Write([]byte("405 - Method Not Allowed"))
					return
				}

				//Check if ArozOS is running in sudo mode
				if !sudo_mode {
					w.WriteHeader(http.StatusUnauthorized)
					w.Write([]byte("401 - Unauthorized (Is ArozOS running in sudo mode?)"))
					return
				}

				//Format option require passing in all filesystem handlers
				allFsh := GetAllLoadedFsh()
				diskmg.HandleFormat(w, r, allFsh)
			})
			adminRouter.HandleFunc("/system/disk/diskmg/mpt", diskmg.HandleListMountPoints)
		}

	}

}