getInitfsExtraFiles: add scripts from hooks-extra (MR 27)

Put scripts from /etc/postmarketos-mkinitfs/hooks-extra into the extra
initramfs instead of the regular one, similar to how it is possible with
files listed in /etc/postmarketos-mkinitfs/files-extra.

This way we will be able to launch hooks not only very early in the
initramfs as it's currently the case. But also later on after the
initramfs-extra was extracted, and more files are available. ondev2 will
make use of this feature.

main.go

@@ -165,7 +165,7 @@ func getHookFiles(filesdir string) (files []string, err error) {
 		s := bufio.NewScanner(f)
 		for s.Scan() {
 			if filelist, err := getFiles([]string{s.Text()}, true); err != nil {
-				return nil, fmt.Errorf("getHookFiles: unable to find file %q required by %q", s.Text(), path)
+				return nil, fmt.Errorf("getHookFiles: unable to add file %q required by %q: %w", s.Text(), path, err)
 			} else {
 				files = append(files, filelist...)
 			}
@@ -218,20 +218,30 @@ func getBinaryDeps(file string) (files []string, err error) {
 		return files, err
 	}
 
-	libdirs := []string{"/usr/lib", "/lib"}
+	// we don't recursively search these paths for performance reasons
+	libdirGlobs := []string{
+		"/usr/lib",
+		"/lib",
+		"/usr/lib/expect*",
+	}
+
 	for _, lib := range libs {
 		found := false
-		for _, libdir := range libdirs {
-			path := filepath.Join(libdir, lib)
-			if _, err := os.Stat(path); err == nil {
-				binaryDepFiles, err := getBinaryDeps(path)
-				if err != nil {
-					return files, err
+	findDepLoop:
+		for _, libdirGlob := range libdirGlobs {
+			libdirs, _ := filepath.Glob(libdirGlob)
+			for _, libdir := range libdirs {
+				path := filepath.Join(libdir, lib)
+				if _, err := os.Stat(path); err == nil {
+					binaryDepFiles, err := getBinaryDeps(path)
+					if err != nil {
+						return files, err
+					}
+					files = append(files, binaryDepFiles...)
+					files = append(files, path)
+					found = true
+					break findDepLoop
 				}
-				files = append(files, binaryDepFiles...)
-				files = append(files, path)
-				found = true
-				break
 			}
 		}
 		if !found {
@@ -275,7 +285,7 @@ func getFile(file string, required bool) (files []string, err error) {
 	fileInfo, err := os.Stat(file)
 	if err != nil {
 		if required {
-			return files, errors.New("getFile: File does not exist :" + file)
+			return files, fmt.Errorf("getFile: failed to stat file %q: %w", file, err)
 		}
 		return files, nil
 	}
@@ -426,11 +436,16 @@ func getFdeFiles(devinfo deviceinfo.DeviceInfo) (files []string, err error) {
 	return
 }
 
-func getHookScripts() (files []string) {
-	scripts, _ := filepath.Glob("/etc/postmarketos-mkinitfs/hooks/*.sh")
-	files = append(files, scripts...)
-
-	return
+func getHookScripts(scriptsdir string) (files []string, err error) {
+	fileInfo, err := os.ReadDir(scriptsdir)
+	if err != nil {
+		return nil, fmt.Errorf("getHookScripts: unable to read hook script dir: %w", err)
+	}
+	for _, file := range fileInfo {
+		path := filepath.Join(scriptsdir, file.Name())
+		files = append(files, path)
+	}
+	return files, nil
 }
 
 func getInitfsExtraFiles(devinfo deviceinfo.DeviceInfo) (files []string, err error) {
@@ -466,6 +481,15 @@ func getInitfsExtraFiles(devinfo deviceinfo.DeviceInfo) (files []string, err err
 		}
 	}
 
+	if exists("/etc/postmarketos-mkinitfs/hooks-extra") {
+		log.Println("- Including extra hook scripts")
+		if hookScripts, err := getHookScripts("/etc/postmarketos-mkinitfs/hooks-extra"); err != nil {
+			return nil, err
+		} else {
+			files = append(files, hookScripts...)
+		}
+	}
+
 	if exists("/usr/bin/osk-sdl") {
 		log.Println("- Including FDE support")
 		if fdeFiles, err := getFdeFiles(devinfo); err != nil {
@@ -506,9 +530,14 @@ func getInitfsFiles(devinfo deviceinfo.DeviceInfo) (files []string, err error) {
 		}
 	}
 
-	log.Println("- Including hook scripts")
-	hookScripts := getHookScripts()
-	files = append(files, hookScripts...)
+	if exists("/etc/postmarketos-mkinitfs/hooks") {
+		log.Println("- Including hook scripts")
+		if hookScripts, err := getHookScripts("/etc/postmarketos-mkinitfs/hooks"); err != nil {
+			return nil, err
+		} else {
+			files = append(files, hookScripts...)
+		}
+	}
 
 	log.Println("- Including required binaries")
 	if filelist, err := getFiles(requiredFiles, true); err != nil {