/*
* Copyright (c) 2017 Apple Inc. All rights reserved.
*
* @APPLE_LICENSE_HEADER_START@
*
* 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 2.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.opensource.apple.com/apsl/ 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, QUIET ENJOYMENT OR NON-INFRINGEMENT.
* Please see the License for the specific language governing rights and
* limitations under the License.
*
* @APPLE_LICENSE_HEADER_END@
*/
#import <Foundation/Foundation.h>
#include <stdio.h>
#include <stdlib.h>
#include <getopt.h>
#include "../src/radix_tree_internal.h"
void
usage() {
printf ("usage: radix-tree [-l 0xADDRESS | 0xSTART-0xEND] FILENAME\n");
printf ("\n");
printf ("This is a debugging tool for the radix-tree sidetable used to track VM allocations\n");
printf ("under MallocStackLogging=lite.\n");
printf ("\n");
printf (" radix-tree FILE # print out radix tree as text\n");
printf (" radix-tree -l 0xf00 FILE # lookup address in radix tree\n");
printf (" radix-tree -l 0xf00-0xba FILE # lookup address range in radix tree\n");
printf ("\n");
exit(0);
}
uint64_t minsize = 4096;
int main(int argc, char **argv) {
int ch;
uint64_t start = 0, end = 0;
while ((ch = getopt(argc, argv, "l:")) != -1) {
switch (ch) {
case 'l': {
char *p = strchr(optarg, '-');
if (p) {
*p = 0;
end = strtoull(p+1, NULL, 16);
start = strtoull(optarg, NULL, 16);
} else {
start = strtoull(optarg, NULL, 16);
}
if (start usage();
}
}
break;
default:
case '?':
usage();
}
}
argc -= optind;
argv += optind;
if (argc != 1)
usage();
NSData *data;
if (0==strcmp(argv[0], "-")) {
data = [[NSFileHandle fileHandleWithStandardInput] readDataToEndOfFile];
} else {
data = [NSData dataWithContentsOfFile:[NSString stringWithUTF8String:argv[0]]];
}
if (!data) {
fprintf(stderr, "failed to read data\n");
return 1;
}
struct radix_tree *tree = (void*) [data bytes];
radix_tree_fsck(tree);
if (start != 0 && end != 0) {
uint64_t last_stackid = -1;
uint64_t last_start = 0;
for (uint64_t a = start; a <= end; a += minsize) {
uint64_t stackid = a == end ? -1 : radix_tree_lookup(tree, a);
if (last_stackid != -1) {
if (stackid != last_stackid) {
printf ("[ last_stackid = stackid;
last_start = a;
}
} else {
if (stackid != -1) {
last_stackid = stackid;
last_start = a;
}
}
}
} else if (start != 0) {
printf (" } else {
radix_tree_print(tree);
}
}