From 5f0e1a86ab5a30d15881ba9798a9057d78470131 Mon Sep 17 00:00:00 2001
From: Danny Lin <danny@kdrag0n.dev>
Date: Tue, 23 Nov 2021 20:12:49 -0800
Subject: [PATCH] selinux: Add code to read and serialize labels recursively

---
 src/selinux/labels.ts | 36 ++++++++++++++++++++++++++++++++++++
 1 file changed, 36 insertions(+)
 create mode 100644 src/selinux/labels.ts

diff --git a/src/selinux/labels.ts b/src/selinux/labels.ts
new file mode 100644
index 0000000..c399ce3
--- /dev/null
+++ b/src/selinux/labels.ts
@@ -0,0 +1,36 @@
+import _ from 'lodash'
+
+import { parseLines } from '../util/parse'
+import { run } from '../util/process'
+
+export type SelinuxFileLabels = Map<string, string>
+
+export async function enumerateSelinuxLabels(root: string) {
+  // Recursive, abs paths, don't follow symlinks
+  let attrs = await run(`getfattr --absolute-names --recursive --physical -n security.selinux ${root}`)
+
+  let labels: SelinuxFileLabels = new Map<string, string>()
+  let lastPath = ''
+  for (let line of parseLines(attrs, false)) {
+    let match = line.match(/^# file: (.+)$/)
+    if (match != undefined) {
+      lastPath = match[1]
+      continue
+    }
+
+    match = line.match(/^security.selinux="(.+)"$/)
+    if (match != undefined) {
+      let label = match[1]
+      labels.set(lastPath, label)
+      continue
+    }
+  }
+
+  return labels
+}
+
+export function generateFileContexts(labels: SelinuxFileLabels) {
+  return Array.from(labels.entries())
+    .map(([path, context]) => `${_.escapeRegExp(path)} ${context}`)
+    .join('\n') + '\n'
+}