void kernel(){
setup_idt();
clean_screen();
print_mem_info();
mm_init();
init_8259A();
*(int*)0xf0000000 = 0; //cause pf interrupt
//asm volatile("int $0xff");
while(1);
}
void input_eval_loop() {
char** recHistory = NULL; int curRecHistEntry = 0;
while (1) {
dPrintf(">");
dConsoleRedraw();
int res = dGetLine(expr,INPUTBUFLEN);
dPuts(expr);
update_cmd_history(expr);
dConsoleRedraw();
if(strcmp(expr, "testmode") == 0) {
//TestMode(1);
} else if(strcmp(expr, "meminfo") == 0) {
print_mem_info();
} else if(strcmp(expr, "memgc") == 0) {
gc();
} else {
execution_in_progress = 1;
PM_SetCPUFreq(152);
run(expr);
PM_SetCPUFreq(16);
execution_in_progress = 0;
}
}
}
void input_eval_loop(int isRecording) {
char** recHistory = NULL; int curRecHistEntry = 0;
if(isRecording) recHistory = (char**)alloca(200); // space for 200 pointers to history entries
while (1) {
DefineStatusMessage((char*)"", 1, 0, 0);
strcpy(expr, (char*)"");
printf("\x1e");
dConsoleRedraw();
int res = gets(expr,INPUTBUFLEN);
if(res == 2) {
dConsolePut("\n");
select_script_and_run();
continue;
}
if(res == 4) {
dConsolePut("\n");
select_strip_script();
continue;
}
if(res == 3) {
dConsolePut("\n");
char buf[100] = "";
sprintf(buf, "prizmUIkeyHandler(%d,%d)", custom_key_to_handle, custom_key_to_handle_modifier);
strcpy(expr, (char*)buf);
execution_in_progress = 1;
run(buf);
execution_in_progress = 0;
check_do_graph();
if(run_startup_script_again) { run_startup_script_again = 0; run_startup_script(); }
continue;
}
puts(expr);
update_cmd_history(expr);
dConsoleRedraw();
if(strcmp(expr, "testmode") == 0) {
TestMode(1);
} else if(strcmp(expr, "meminfo") == 0) {
print_mem_info();
} else if(strcmp(expr, "memgc") == 0) {
gc();
} else if(strcmp(expr, "record") == 0) {
if(!isRecording) script_recorder();
else {
// create and save a script. this must be done here, because we used alloca
// the "clean" way would be using malloc&free, but on the Prizm the heap is already being heavily used by the Eigenmath core.
if(curRecHistEntry == 0) {
printf("Nothing to record.\n");
return;
}
printf("Recording stopped.\n");
printf("Type a name for the script, or\n");
printf("leave empty to discard.\n:");
char inputname[MAX_FILENAME_SIZE+1] = "";
gets(inputname,MAX_FILENAME_SIZE-50);
puts(inputname);
if(!strlen(inputname)) {
// user aborted
printf("Recording discarded.\n");
return;
}
if (aborttimer > 0) {
Timer_Stop(aborttimer);
Timer_Deinstall(aborttimer);
}
char filename[MAX_FILENAME_SIZE+1] = "";
sprintf(filename, "\\\\fls0\\%s.txt", inputname);
unsigned short pFile[MAX_FILENAME_SIZE+1];
Bfile_StrToName_ncpy(pFile, (unsigned char*)filename, strlen(filename)+1);
// calculate size
int size = 0;
int maxHistory = curRecHistEntry - 1; //because we ++'ed at the end of last addition
for(int i=0; i <= maxHistory; i++) {
size = size + strlen(recHistory[i]) + 1; // 1 byte for \n. we will use unix line termination
}
int BCEres = Bfile_CreateEntry_OS(pFile, CREATEMODE_FILE, &size);
if(BCEres >= 0) // Did it create?
{
BCEres = Bfile_OpenFile_OS(pFile, READWRITE, 0); // Get handle
for(int i=0; i <= maxHistory; i++) {
char* buf = (char*)alloca(strlen(recHistory[i])+5);
strcpy(buf, recHistory[i]);
strcat(buf, (char*)"\n");
Bfile_WriteFile_OS(BCEres, buf, strlen(recHistory[i])+1);
}
Bfile_CloseFile_OS(BCEres);
printf("Script created.\n");
} else {
printf("An error occurred when creating the script for recording.\n");
}
aborttimer = Timer_Install(0, check_execution_abort, 100);
if (aborttimer > 0) Timer_Start(aborttimer);
return;
}
} else {
execution_in_progress = 1;
has_drawn_graph = 0;
run(expr);
// run_startup_script cannot run from inside eval_clear because then it would be a run() inside a run()
if(run_startup_script_again) { run_startup_script_again = 0; run_startup_script(); }
execution_in_progress = 0;
// if recording, add input to record
if(isRecording && curRecHistEntry <= 200) {
recHistory[curRecHistEntry] = (char*)alloca(strlen(expr)+2); // 2 bytes for security
strcpy(recHistory[curRecHistEntry], expr);
curRecHistEntry++;
}
check_do_graph();
}
}
}
int main(int argc, char *argv[]) {
pm_kernel_t *ker;
pm_process_t *proc;
pid_t *pids;
struct proc_info **procs;
size_t num_procs;
unsigned long total_pss;
unsigned long total_uss;
char cmdline[256]; // this must be within the range of int
int error;
/// M: get swap usage @{
int show_swap;
/// @}
#define WS_OFF 0
#define WS_ONLY 1
#define WS_RESET 2
int ws;
int arg;
size_t i, j;
signal(SIGPIPE, SIG_IGN);
compfn = &sort_by_pss;
order = -1;
ws = WS_OFF;
/// M: get swap usage @{
show_swap = 0;
/// @}
for (arg = 1; arg < argc; arg++) {
if (!strcmp(argv[arg], "-v")) { compfn = &sort_by_vss; continue; }
if (!strcmp(argv[arg], "-r")) { compfn = &sort_by_rss; continue; }
if (!strcmp(argv[arg], "-p")) { compfn = &sort_by_pss; continue; }
if (!strcmp(argv[arg], "-u")) { compfn = &sort_by_uss; continue; }
if (!strcmp(argv[arg], "-w")) { ws = WS_ONLY; continue; }
if (!strcmp(argv[arg], "-W")) { ws = WS_RESET; continue; }
if (!strcmp(argv[arg], "-R")) { order *= -1; continue; }
/// M: get swap usage @{
if (!strcmp(argv[arg], "-s")) { show_swap = 1; continue; }
/// @}
if (!strcmp(argv[arg], "-h")) { usage(argv[0]); exit(0); }
fprintf(stderr, "Invalid argument \"%s\".\n", argv[arg]);
usage(argv[0]);
exit(EXIT_FAILURE);
}
error = pm_kernel_create(&ker);
if (error) {
fprintf(stderr, "Error creating kernel interface -- "
"does this kernel have pagemap?\n");
exit(EXIT_FAILURE);
}
error = pm_kernel_pids(ker, &pids, &num_procs);
if (error) {
fprintf(stderr, "Error listing processes.\n");
exit(EXIT_FAILURE);
}
procs = calloc(num_procs, sizeof(struct proc_info*));
if (procs == NULL) {
fprintf(stderr, "calloc: %s", strerror(errno));
exit(EXIT_FAILURE);
}
for (i = 0; i < num_procs; i++) {
procs[i] = malloc(sizeof(struct proc_info));
if (procs[i] == NULL) {
fprintf(stderr, "malloc: %s\n", strerror(errno));
exit(EXIT_FAILURE);
}
procs[i]->pid = pids[i];
pm_memusage_zero(&procs[i]->usage);
error = pm_process_create(ker, pids[i], &proc);
if (error) {
fprintf(stderr, "warning: could not create process interface for %d\n", pids[i]);
continue;
}
switch (ws) {
case WS_OFF:
error = pm_process_usage(proc, &procs[i]->usage);
break;
case WS_ONLY:
error = pm_process_workingset(proc, &procs[i]->usage, 0);
break;
case WS_RESET:
error = pm_process_workingset(proc, NULL, 1);
break;
}
if (error) {
fprintf(stderr, "warning: could not read usage for %d\n", pids[i]);
}
pm_process_destroy(proc);
}
free(pids);
if (ws == WS_RESET) exit(0);
j = 0;
for (i = 0; i < num_procs; i++) {
if (procs[i]->usage.vss) {
procs[j++] = procs[i];
} else {
free(procs[i]);
}
}
num_procs = j;
qsort(procs, num_procs, sizeof(procs[0]), compfn);
if (ws)
printf("%5s %7s %7s %7s %s\n", "PID", "WRss", "WPss", "WUss", "cmdline");
else {
/// M: get swap usage @{
if (show_swap) {
printf("%5s %7s %7s %7s %7s %7s %7s %s\n", "PID", "Vss", "Rss", "Pss", "Uss", "Swap", "PSwap", "cmdline");
} else {
printf("%5s %7s %7s %7s %7s %s\n", "PID", "Vss", "Rss", "Pss", "Uss", "cmdline");
}
/// @}
}
total_pss = 0;
total_uss = 0;
for (i = 0; i < num_procs; i++) {
if (getprocname(procs[i]->pid, cmdline, (int)sizeof(cmdline)) < 0) {
/*
* Something is probably seriously wrong if writing to the stack
* failed.
*/
free(procs[i]);
continue;
}
total_pss += procs[i]->usage.pss;
total_uss += procs[i]->usage.uss;
if (ws)
printf("%5d %6dK %6dK %6dK %s\n",
procs[i]->pid,
procs[i]->usage.rss / 1024,
procs[i]->usage.pss / 1024,
procs[i]->usage.uss / 1024,
cmdline
);
else {
/// M: get swap usage @{
if (show_swap) {
printf("%5d %6dK %6dK %6dK %6dK %6dK %6dK %s\n",
procs[i]->pid,
procs[i]->usage.vss / 1024,
procs[i]->usage.rss / 1024,
procs[i]->usage.pss / 1024,
procs[i]->usage.uss / 1024,
procs[i]->usage.swap / 1024,
procs[i]->usage.pswap / 1024,
cmdline
);
} else {
printf("%5d %6dK %6dK %6dK %6dK %s\n",
procs[i]->pid,
procs[i]->usage.vss / 1024,
procs[i]->usage.rss / 1024,
procs[i]->usage.pss / 1024,
procs[i]->usage.uss / 1024,
cmdline
);
}
/// @}
}
free(procs[i]);
}
free(procs);
if (ws) {
printf("%5s %7s %7s %7s %s\n",
"", "", "------", "------", "------");
printf("%5s %7s %6ldK %6ldK %s\n",
"", "", total_pss / 1024, total_uss / 1024, "TOTAL");
} else {
printf("%5s %7s %7s %7s %7s %s\n",
"", "", "", "------", "------", "------");
printf("%5s %7s %7s %6ldK %6ldK %s\n",
"", "", "", total_pss / 1024, total_uss / 1024, "TOTAL");
}
printf("\n");
print_mem_info();
return 0;
}
int main(int argc, char *argv[]) {
pm_kernel_t *ker;
pm_process_t *proc;
pid_t *pids;
struct proc_info **procs;
size_t num_procs;
uint64_t total_pss;
uint64_t total_uss;
uint64_t total_swap;
char cmdline[256]; // this must be within the range of int
int error;
bool has_swap = false;
uint64_t required_flags = 0;
uint64_t flags_mask = 0;
#define WS_OFF 0
#define WS_ONLY 1
#define WS_RESET 2
int ws;
int arg;
size_t i, j;
signal(SIGPIPE, SIG_IGN);
compfn = &sort_by_pss;
order = -1;
ws = WS_OFF;
for (arg = 1; arg < argc; arg++) {
if (!strcmp(argv[arg], "-v")) { compfn = &sort_by_vss; continue; }
if (!strcmp(argv[arg], "-r")) { compfn = &sort_by_rss; continue; }
if (!strcmp(argv[arg], "-p")) { compfn = &sort_by_pss; continue; }
if (!strcmp(argv[arg], "-u")) { compfn = &sort_by_uss; continue; }
if (!strcmp(argv[arg], "-s")) { compfn = &sort_by_swap; continue; }
if (!strcmp(argv[arg], "-c")) { required_flags = 0; flags_mask = PM_PAGE_SWAPBACKED; continue; }
if (!strcmp(argv[arg], "-C")) { required_flags = flags_mask = PM_PAGE_SWAPBACKED; continue; }
if (!strcmp(argv[arg], "-k")) { required_flags = flags_mask = PM_PAGE_KSM; continue; }
if (!strcmp(argv[arg], "-w")) { ws = WS_ONLY; continue; }
if (!strcmp(argv[arg], "-W")) { ws = WS_RESET; continue; }
if (!strcmp(argv[arg], "-R")) { order *= -1; continue; }
if (!strcmp(argv[arg], "-h")) { usage(argv[0]); exit(0); }
fprintf(stderr, "Invalid argument \"%s\".\n", argv[arg]);
usage(argv[0]);
exit(EXIT_FAILURE);
}
error = pm_kernel_create(&ker);
if (error) {
fprintf(stderr, "Error creating kernel interface -- "
"does this kernel have pagemap?\n");
exit(EXIT_FAILURE);
}
error = pm_kernel_pids(ker, &pids, &num_procs);
if (error) {
fprintf(stderr, "Error listing processes.\n");
exit(EXIT_FAILURE);
}
procs = calloc(num_procs, sizeof(struct proc_info*));
if (procs == NULL) {
fprintf(stderr, "calloc: %s", strerror(errno));
exit(EXIT_FAILURE);
}
for (i = 0; i < num_procs; i++) {
procs[i] = malloc(sizeof(struct proc_info));
if (procs[i] == NULL) {
fprintf(stderr, "malloc: %s\n", strerror(errno));
exit(EXIT_FAILURE);
}
procs[i]->pid = pids[i];
pm_memusage_zero(&procs[i]->usage);
error = pm_process_create(ker, pids[i], &proc);
if (error) {
fprintf(stderr, "warning: could not create process interface for %d\n", pids[i]);
continue;
}
switch (ws) {
case WS_OFF:
error = pm_process_usage_flags(proc, &procs[i]->usage, flags_mask,
required_flags);
break;
case WS_ONLY:
error = pm_process_workingset(proc, &procs[i]->usage, 0);
break;
case WS_RESET:
error = pm_process_workingset(proc, NULL, 1);
break;
}
if (error) {
fprintf(stderr, "warning: could not read usage for %d\n", pids[i]);
}
if (ws != WS_RESET && procs[i]->usage.swap) {
has_swap = true;
}
pm_process_destroy(proc);
}
free(pids);
if (ws == WS_RESET) exit(0);
j = 0;
for (i = 0; i < num_procs; i++) {
if (procs[i]->usage.vss) {
procs[j++] = procs[i];
} else {
free(procs[i]);
}
}
num_procs = j;
qsort(procs, num_procs, sizeof(procs[0]), compfn);
printf("%5s ", "PID");
if (ws) {
printf("%s %7s %7s ", "WRss", "WPss", "WUss");
if (has_swap) {
printf("%7s ", "WSwap");
}
} else {
printf("%8s %7s %7s %7s ", "Vss", "Rss", "Pss", "Uss");
if (has_swap) {
printf("%7s ", "Swap");
}
}
printf("%s\n", "cmdline");
total_pss = 0;
total_uss = 0;
total_swap = 0;
for (i = 0; i < num_procs; i++) {
if (getprocname(procs[i]->pid, cmdline, (int)sizeof(cmdline)) < 0) {
/*
* Something is probably seriously wrong if writing to the stack
* failed.
*/
free(procs[i]);
continue;
}
total_pss += procs[i]->usage.pss;
total_uss += procs[i]->usage.uss;
total_swap += procs[i]->usage.swap;
printf("%5d ", procs[i]->pid);
if (ws) {
printf("%6zuK %6zuK %6zuK ",
procs[i]->usage.rss / 1024,
procs[i]->usage.pss / 1024,
procs[i]->usage.uss / 1024
);
} else {
printf("%7zuK %6zuK %6zuK %6zuK ",
procs[i]->usage.vss / 1024,
procs[i]->usage.rss / 1024,
procs[i]->usage.pss / 1024,
procs[i]->usage.uss / 1024
);
}
if (has_swap) {
printf("%6zuK ", procs[i]->usage.swap / 1024);
}
printf("%s\n", cmdline);
free(procs[i]);
}
free(procs);
/* Print the separator line */
printf("%5s ", "");
if (ws) {
printf("%7s %7s %7s ", "", "------", "------");
} else {
printf("%8s %7s %7s %7s ", "", "", "------", "------");
}
if (has_swap) {
printf("%7s ", "------");
}
printf("%s\n", "------");
/* Print the total line */
printf("%5s ", "");
if (ws) {
printf("%7s %6" PRIu64 "K %" PRIu64 "K ",
"", total_pss / 1024, total_uss / 1024);
} else {
printf("%8s %7s %6" PRIu64 "K %6" PRIu64 "K ",
"", "", total_pss / 1024, total_uss / 1024);
}
if (has_swap) {
printf("%6" PRIu64 "K ", total_swap / 1024);
}
printf("TOTAL\n");
printf("\n");
print_mem_info();
return 0;
}
