1 files changed, 144 insertions, 0 deletions
diff --git a/system_cmds/kpgo.tproj/kpgo.c b/system_cmds/kpgo.tproj/kpgo.c
new file mode 100644
index 0000000..dacfc4c
--- /dev/null
+++ b/system_cmds/kpgo.tproj/kpgo.c
@@ -0,0 +1,144 @@
+/*
+ * Copyright (c) 2014-2016 Apple Computer, Inc. All rights reserved.
+ *
+ * @APPLE_LICENSE_HEADER_START@
+ *
+ * "Portions Copyright (c) 1999 Apple Computer, Inc.  All Rights
+ * Reserved.  This file contains Original Code and/or Modifications of
+ * Original Code as defined in and that are subject to the Apple Public
+ * Source License Version 1.0 (the 'License').  You may not use this file
+ * except in compliance with the License.  Please obtain a copy of the
+ * License at http://www.apple.com/publicsource and read it before using
+ * this file.
+ *
+ * The Original Code and all software distributed under the License are
+ * distributed on an 'AS IS' basis, WITHOUT WARRANTY OF ANY KIND, EITHER
+ * EXPRESS OR IMPLIED, AND APPLE HEREBY DISCLAIMS ALL SUCH WARRANTIES,
+ * INCLUDING WITHOUT LIMITATION, ANY WARRANTIES OF MERCHANTABILITY,
+ * FITNESS FOR A PARTICULAR PURPOSE OR NON-INFRINGEMENT.  Please see the
+ * License for the specific language governing rights and limitations
+ * under the License."
+ *
+ * @APPLE_LICENSE_HEADER_END@
+ */
+
+
+#include <sys/pgo.h>
+
+#include <stdio.h>
+#include <errno.h>
+#include <string.h>
+#include <stdlib.h>
+#include <unistd.h>
+
+static void usage(char **argv)
+{
+	fprintf (stderr, "usage: %s [-H] [-m] [-w] [uuid] >datafile\n", argv[0]);
+        fprintf (stderr, "    uuid : the UUID of a kext\n");
+        fprintf (stderr, "    -H   : grab data for the HIB segment\n");
+        fprintf (stderr, "    -w   : wait for the kext to be unloaded\n");
+        fprintf (stderr, "    -m   : request metadata\n");
+        fprintf (stderr, "    -R   : reset all counters\n");
+	exit(1);
+}
+
+int main(int argc, char **argv)
+{
+	int flags = 0;
+        int data_flags = 0;
+        uuid_t *uuidp = NULL;
+        uuid_t uuid;
+        int c;
+
+        while ((c = getopt(argc, argv, "hHwmR")) != EOF) {
+            switch(c) {
+            case 'R':
+                flags |= PGO_RESET_ALL;
+                break;
+            case 'H':
+                flags |= PGO_HIB;
+                break;
+            case 'm':
+                flags |= PGO_METADATA;
+                break;
+            case 'w':
+                data_flags |= PGO_WAIT_FOR_UNLOAD;
+                break;
+            case '?':
+            case 'h':
+            default:
+                usage(argv);
+                break;
+            }
+        }
+
+        if (optind < argc)
+        {
+            if (optind == argc - 1 &&
+                0 == uuid_parse(argv[optind], uuid))
+            {
+                uuidp = &uuid;
+            } else {
+                usage(argv);
+            }
+        }
+
+        if (flags & PGO_RESET_ALL) {
+            if (flags != PGO_RESET_ALL || uuidp) {
+                usage(argv);
+            }
+            ssize_t r = grab_pgo_data(NULL, PGO_RESET_ALL, NULL, 0);
+            if (r < 0)
+            {
+                perror("grab_pgo_data");
+                return 1;
+            }
+            return 0;
+        }
+
+	ssize_t size = grab_pgo_data(uuidp, flags, NULL, 0);
+
+	if (size < 0)
+	{
+		perror("grab_pgo_data");
+		return 1;
+	}
+
+
+	fprintf (stderr, "size = %ld\n", (long) size);
+
+	unsigned char *buffer = valloc(size);
+	if (!buffer)
+	{
+		perror("valloc");
+		return 1;
+	}
+
+	ssize_t r = grab_pgo_data(uuidp, flags | data_flags, buffer, size);
+
+
+	if (r < 0)
+	{
+		perror("grab_pgo_data");
+		return 1;
+	}
+
+        if (isatty(STDOUT_FILENO)) {
+            fprintf (stderr, "%s: refusing to write binary data to a tty!\n", argv[0]);
+            return 1;
+        }
+
+        while (size > 0) {
+            errno = 0;
+            r = write(STDOUT_FILENO, buffer, size);
+            if (r > 0) {
+                buffer += r;
+                size -= r;
+            } else {
+                perror ("write");
+                return 1;
+            }
+        }
+
+	return 0;
+}