xnu-7195.101.1.tar.gz

[apple/xnu.git] / libkdd / tests / Tests.swift

diff --git a/libkdd/tests/Tests.swift b/libkdd/tests/Tests.swift
index cd7f46ea060194439d7bcb3edfc1ce9f34c3a95b..34f8d9e8eeac36a4cc4cf0b119fe20fe8f351307 100644 (file)
--- a/libkdd/tests/Tests.swift
+++ b/libkdd/tests/Tests.swift
@@ -734,6 +734,66 @@ class Tests: XCTestCase {
         XCTAssert(dict.value(forKeyPath: "kcdata_crashinfo.task_snapshots.0.crashed_threadid")  as? Int == 42)
     }
 
+    func testDispatchQueueLabel() {
+        let buffer = NSMutableData(capacity:1000)!
+
+        var item = kcdata_item()
+        let dql = "houston.we.had.a.problem"
+        var payload32 : UInt32
+
+        item.type = KCDATA_BUFFER_BEGIN_STACKSHOT
+        item.flags = 0
+        item.size = 0
+        buffer.append(&item, length: MemoryLayout<kcdata_item>.size)
+
+        item.type = UInt32(KCDATA_TYPE_CONTAINER_BEGIN)
+        item.flags = 0
+        item.size = UInt32(MemoryLayout<UInt32>.size)
+        buffer.append(&item, length: MemoryLayout<kcdata_item>.size)
+        payload32 = UInt32(STACKSHOT_KCCONTAINER_TASK)
+        buffer.append(&payload32, length:MemoryLayout<UInt32>.size)
+
+        item.type = UInt32(KCDATA_TYPE_CONTAINER_BEGIN)
+        item.flags = 0
+        item.size = UInt32(MemoryLayout<UInt32>.size)
+        buffer.append(&item, length: MemoryLayout<kcdata_item>.size)
+        payload32 = UInt32(STACKSHOT_KCCONTAINER_THREAD)
+        buffer.append(&payload32, length:MemoryLayout<UInt32>.size)
+
+        item.type = UInt32(STACKSHOT_KCTYPE_THREAD_DISPATCH_QUEUE_LABEL)
+        item.flags = 0
+        item.size = UInt32(dql.utf8.count + 1)
+        buffer.append(&item, length: MemoryLayout<kcdata_item>.size)
+        dql.utf8CString.withUnsafeBufferPointer({
+            buffer.append($0.baseAddress!, length:dql.utf8.count + 1)
+        })
+
+        item.type = UInt32(KCDATA_TYPE_CONTAINER_END)
+        item.flags = 0
+        item.size = UInt32(MemoryLayout<UInt32>.size)
+        buffer.append(&item, length: MemoryLayout<kcdata_item>.size)
+        payload32 = UInt32(STACKSHOT_KCCONTAINER_THREAD)
+        buffer.append(&payload32, length:MemoryLayout<UInt32>.size)
+
+        item.type = UInt32(KCDATA_TYPE_CONTAINER_END)
+        item.flags = 0
+        item.size = UInt32(MemoryLayout<UInt32>.size)
+        buffer.append(&item, length: MemoryLayout<kcdata_item>.size)
+        payload32 = UInt32(STACKSHOT_KCCONTAINER_TASK)
+        buffer.append(&payload32, length:MemoryLayout<UInt32>.size)
+
+
+        item.type = KCDATA_TYPE_BUFFER_END
+        item.flags = 0
+        item.size = 0
+        buffer.append(&item, length: MemoryLayout<kcdata_item>.size)
+
+        guard let dict = try? self.parseBuffer(buffer)
+            else { XCTFail(); return; }
+
+        XCTAssert(dict.value(forKeyPath: "kcdata_stackshot.task_snapshots.0.thread_snapshots.0.dispatch_queue_label")  as? String == dql)
+    }
+
     func testRepeatedContainer() {
         //repeated container of same name and key shoudl fail
 
@@ -1208,7 +1268,7 @@ class Tests: XCTestCase {
         // check that we agree with sample file
 
         guard let sampledata = self.dataWithResource(name)
-            else { XCTFail(); return }
+            else { XCTFail("failed to open bundle resource named " + name); return }
         var dict : NSDictionary?
 
         dict = try? self.parseBuffer(sampledata) as NSDictionary
@@ -1233,14 +1293,16 @@ class Tests: XCTestCase {
                               self.dataWithResource(name + ".plist")
             else {XCTFail(); return}
 
-        var dict2 = try? PropertyListSerialization.propertyList(from: plistdata as Data, options: [], format: nil)
-        if dict2 == nil {
-            dict2 = try? PropertyListSerialization.propertyList(from:decompress(plistdata) as Data, options:[], format: nil)
+        var opt_dict2 = try? PropertyListSerialization.propertyList(from: plistdata as Data, options: [], format: nil)
+        if opt_dict2 == nil {
+            opt_dict2 = try? PropertyListSerialization.propertyList(from:decompress(plistdata) as Data, options:[], format: nil)
         }
+        guard let dict2 = opt_dict2
+            else { XCTFail(); return}
 
-        XCTAssert(dict2 != nil)
+        XCTAssertEqual(dict, dict2 as! NSDictionary);
 
-        XCTAssert(dict == dict2 as? NSDictionary)
+        //XCTAssert(dict == dict2 as? NSDictionary)
 
         // check that we agree with python
 
@@ -1338,10 +1400,18 @@ class Tests: XCTestCase {
         self.testSampleStackshot("stackshot-sample-thread-groups")
     }
 
+    func testSampleThreadGroupsFlags() {
+        self.testSampleStackshot("stackshot-sample-thread-groups-flags")
+    }
+
     func testSampleCoalitions() {
         self.testSampleStackshot("stackshot-sample-coalitions")
     }
 
+    func testSampleTurnstileInfo() {
+        self.testSampleStackshot("stackshot-sample-turnstileinfo")
+    }
+
     func testStackshotSharedcacheV2() {
         self.testSampleStackshot("stackshot-sample-sharedcachev2")
     }
@@ -1366,10 +1436,38 @@ class Tests: XCTestCase {
         self.testSampleStackshot("stackshot-sample-thread-policy")
     }
 
+    func testDeltaStackshotWithThreadPolicy() {
+        self.testSampleStackshot("stackshot-sample-delta-thread-policy")
+    }
+
     func testStackshotWithInstrsCycles() {
         self.testSampleStackshot("stackshot-sample-instrs-cycles")
     }
 
+    func testStackshotWithStacktop() {
+        self.testSampleStackshot("stackshot-sample-stacktop")
+    }
+
+    func testStackshotWithASID() {
+        self.testSampleStackshot("stackshot-sample-asid")
+    }
+
+    func testStackshotWithPageTables() {
+        self.testSampleStackshot("stackshot-sample-asid-pagetable")
+    }
+
+    func testStackshotCPUTimes() {
+        self.testSampleStackshot("stackshot-sample-cpu-times")
+    }
+    
+    func testStackshotWithSharedCacheLayout() {
+        self.testSampleStackshot("stackshot-with-shared-cache-layout")
+    }
+
+    func testStackshotDispatchQueueLabel() {
+        self.testSampleStackshot("stackshot-sample-dispatch-queue-label")
+    }
+
     func testTrivial() {
     }
 }