lkml.org 
[lkml]   [2014]   [Mar]   [28]   [last100]   RSS Feed
Views: [wrap][no wrap]   [headers]  [forward] 
 
Messages in this thread
Patch in this message
/
From
Subject[PATCH 12/28] ktap: add runtime/kp_obj.[c|h]
Date
runtime/kp_obj.c include some common ktap object operation,
like object dump, object length.
It also include generic memory allocate code.

Exposed functions:
1). kp_malloc/kp_zalloc/kp_free

2). kp_obj_dump/kp_obj_show

3). kp_obj_rawequal

4). kp_obj_len

5). kp_obj_new
allocate new object, all object is linked in G(ks)->allgc.

6). kp_obj_kstack2str
convert kernel stack to string.

7). kp_obj_freeall
free all object, called in kp_vm_exit.

Signed-off-by: Jovi Zhangwei <jovi.zhangwei@gmail.com>
---
tools/ktap/runtime/kp_obj.c | 281 ++++++++++++++++++++++++++++++++++++++++++++
tools/ktap/runtime/kp_obj.h | 19 +++
2 files changed, 300 insertions(+)
create mode 100644 tools/ktap/runtime/kp_obj.c
create mode 100644 tools/ktap/runtime/kp_obj.h

diff --git a/tools/ktap/runtime/kp_obj.c b/tools/ktap/runtime/kp_obj.c
new file mode 100644
index 0000000..a51ab74
--- /dev/null
+++ b/tools/ktap/runtime/kp_obj.c
@@ -0,0 +1,281 @@
+/*
+ * kp_obj.c - ktap object generic operation
+ *
+ * This file is part of ktap by Jovi Zhangwei.
+ *
+ * Copyright (C) 2012-2014 Jovi Zhangwei <jovi.zhangwei@gmail.com>.
+ *
+ * Adapted from luajit and lua interpreter.
+ * Copyright (C) 2005-2014 Mike Pall.
+ * Copyright (C) 1994-2008 Lua.org, PUC-Rio.
+ *
+ * ktap is free software; you can redistribute it and/or modify it
+ * under the terms and conditions of the GNU General Public License,
+ * version 2, as published by the Free Software Foundation.
+ *
+ * ktap is distributed in the hope it will be useful, but WITHOUT
+ * ANY WARRANTY; without even the implied warranty of MERCHANTABILITY or
+ * FITNESS FOR A PARTICULAR PURPOSE. See the GNU General Public License for
+ * more details.
+ *
+ * You should have received a copy of the GNU General Public License along with
+ * this program; if not, write to the Free Software Foundation, Inc.,
+ * 51 Franklin St - Fifth Floor, Boston, MA 02110-1301 USA.
+ */
+
+#include <linux/stacktrace.h>
+#include <linux/module.h>
+#include <linux/kallsyms.h>
+#include <linux/slab.h>
+#include "../include/ktap_types.h"
+#include "kp_obj.h"
+#include "kp_str.h"
+#include "kp_tab.h"
+#include "ktap.h"
+#include "kp_vm.h"
+#include "kp_transport.h"
+
+/* Error message strings. */
+const char *kp_err_allmsg =
+#define ERRDEF(name, msg) msg "\0"
+#include "../include/ktap_errmsg.h"
+;
+
+/* memory allocation flag */
+#define KTAP_ALLOC_FLAGS ((GFP_KERNEL | __GFP_NORETRY | __GFP_NOWARN) \
+ & ~__GFP_WAIT)
+
+void *kp_malloc(ktap_state_t *ks, int size)
+{
+ void *addr;
+
+ addr = kmalloc(size, KTAP_ALLOC_FLAGS);
+ if (unlikely(!addr)) {
+ kp_error(ks, "kmalloc failed\n");
+ }
+ return addr;
+}
+
+void *kp_zalloc(ktap_state_t *ks, int size)
+{
+ void *addr;
+
+ addr = kzalloc(size, KTAP_ALLOC_FLAGS);
+ if (unlikely(!addr))
+ kp_error(ks, "kzalloc failed\n");
+ return addr;
+}
+
+void kp_free(ktap_state_t *ks, void *addr)
+{
+ kfree(addr);
+}
+
+
+void kp_obj_dump(ktap_state_t *ks, const ktap_val_t *v)
+{
+ switch (itype(v)) {
+ case KTAP_TNIL:
+ kp_puts(ks, "NIL");
+ break;
+ case KTAP_TTRUE:
+ kp_printf(ks, "true");
+ break;
+ case KTAP_TFALSE:
+ kp_printf(ks, "false");
+ break;
+ case KTAP_TNUM:
+ kp_printf(ks, "NUM %ld", nvalue(v));
+ break;
+ case KTAP_TLIGHTUD:
+ kp_printf(ks, "LIGHTUD 0x%lx", (unsigned long)pvalue(v));
+ break;
+ case KTAP_TFUNC:
+ kp_printf(ks, "FUNCTION 0x%lx", (unsigned long)fvalue(v));
+ break;
+ case KTAP_TSTR:
+ kp_printf(ks, "STR #%s", svalue(v));
+ break;
+ case KTAP_TTAB:
+ kp_printf(ks, "TABLE 0x%lx", (unsigned long)hvalue(v));
+ break;
+ default:
+ kp_printf(ks, "GCVALUE 0x%lx", (unsigned long)gcvalue(v));
+ break;
+ }
+}
+
+void kp_obj_show(ktap_state_t *ks, const ktap_val_t *v)
+{
+ switch (itype(v)) {
+ case KTAP_TNIL:
+ kp_puts(ks, "nil");
+ break;
+ case KTAP_TTRUE:
+ kp_puts(ks, "true");
+ break;
+ case KTAP_TFALSE:
+ kp_puts(ks, "false");
+ break;
+ case KTAP_TNUM:
+ kp_printf(ks, "%ld", nvalue(v));
+ break;
+ case KTAP_TLIGHTUD:
+ kp_printf(ks, "lightud 0x%lx", (unsigned long)pvalue(v));
+ break;
+ case KTAP_TCFUNC:
+ kp_printf(ks, "cfunction 0x%lx", (unsigned long)fvalue(v));
+ break;
+ case KTAP_TFUNC:
+ kp_printf(ks, "function 0x%lx", (unsigned long)gcvalue(v));
+ break;
+ case KTAP_TSTR:
+ kp_puts(ks, svalue(v));
+ break;
+ case KTAP_TTAB:
+ kp_printf(ks, "table 0x%lx", (unsigned long)hvalue(v));
+ break;
+ case KTAP_TEVENTSTR:
+ /* check event context */
+ if (!ks->current_event) {
+ kp_error(ks,
+ "cannot stringify event str in invalid context\n");
+ return;
+ }
+
+ kp_transport_event_write(ks, ks->current_event);
+ break;
+ case KTAP_TKSTACK:
+ kp_transport_print_kstack(ks, v->val.stack.depth,
+ v->val.stack.skip);
+ break;
+ default:
+ kp_error(ks, "print unknown value type: %d\n", itype(v));
+ break;
+ }
+}
+
+
+/*
+ * equality of ktap values.
+ */
+int kp_obj_rawequal(const ktap_val_t *t1, const ktap_val_t *t2)
+{
+ switch (itype(t1)) {
+ case KTAP_TNIL:
+ case KTAP_TTRUE:
+ case KTAP_TFALSE:
+ return 1;
+ case KTAP_TNUM:
+ return nvalue(t1) == nvalue(t2);
+ case KTAP_TLIGHTUD:
+ return pvalue(t1) == pvalue(t2);
+ case KTAP_TFUNC:
+ return fvalue(t1) == fvalue(t2);
+ case KTAP_TSTR:
+ return rawtsvalue(t1) == rawtsvalue(t2);
+ case KTAP_TTAB:
+ return hvalue(t1) == hvalue(t2);
+ default:
+ return gcvalue(t1) == gcvalue(t2);
+ }
+
+ return 0;
+}
+
+/*
+ * ktap will not use lua's length operator for table,
+ * also # is not for length operator any more in ktap.
+ */
+int kp_obj_len(ktap_state_t *ks, const ktap_val_t *v)
+{
+ switch(itype(v)) {
+ case KTAP_TTAB:
+ return kp_tab_len(ks, hvalue(v));
+ case KTAP_TSTR:
+ return rawtsvalue(v)->len;
+ default:
+ kp_printf(ks, "cannot get length of type %d\n", v->type);
+ return -1;
+ }
+ return 0;
+}
+
+/* need to protect allgc field? */
+ktap_obj_t *kp_obj_new(ktap_state_t *ks, size_t size)
+{
+ ktap_obj_t *o, **list;
+
+ if (ks != G(ks)->mainthread) {
+ kp_error(ks, "kp_obj_new only can be called in mainthread\n");
+ return NULL;
+ }
+
+ o = kp_malloc(ks, size);
+ if (unlikely(!o))
+ return NULL;
+
+ list = &G(ks)->allgc;
+ gch(o)->nextgc = *list;
+ *list = o;
+
+ return o;
+}
+
+
+/* this function may be time consuming, move out from table set/get? */
+ktap_str_t *kp_obj_kstack2str(ktap_state_t *ks, uint16_t depth, uint16_t skip)
+{
+ struct stack_trace trace;
+ unsigned long *bt;
+ char *btstr, *p;
+ int i;
+
+ bt = kp_this_cpu_print_buffer(ks); /* use print percpu buffer */
+ trace.nr_entries = 0;
+ trace.skip = skip;
+ trace.max_entries = depth;
+ trace.entries = (unsigned long *)(bt + 1);
+ save_stack_trace(&trace);
+
+ /* convert backtrace to string */
+ p = btstr = kp_this_cpu_temp_buffer(ks);
+ for (i = 0; i < trace.nr_entries; i++) {
+ unsigned long addr = trace.entries[i];
+
+ if (addr == ULONG_MAX)
+ break;
+
+ p += sprint_symbol(p, addr);
+ *p++ = '\n';
+ }
+
+ return kp_str_new(ks, btstr, p - btstr);
+}
+
+static void free_gclist(ktap_state_t *ks, ktap_obj_t *o)
+{
+ while (o) {
+ ktap_obj_t *next;
+
+ next = gch(o)->nextgc;
+ switch (gch(o)->gct) {
+ case ~KTAP_TTAB:
+ kp_tab_free(ks, (ktap_tab_t *)o);
+ break;
+ case ~KTAP_TUPVAL:
+ kp_freeupval(ks, (ktap_upval_t *)o);
+ break;
+ default:
+ kp_free(ks, o);
+ }
+ o = next;
+ }
+}
+
+void kp_obj_freeall(ktap_state_t *ks)
+{
+ free_gclist(ks, G(ks)->allgc);
+ G(ks)->allgc = NULL;
+}
+
diff --git a/tools/ktap/runtime/kp_obj.h b/tools/ktap/runtime/kp_obj.h
new file mode 100644
index 0000000..b24aab0
--- /dev/null
+++ b/tools/ktap/runtime/kp_obj.h
@@ -0,0 +1,19 @@
+#ifndef __KTAP_OBJ_H__
+#define __KTAP_OBJ_H__
+
+void *kp_malloc(ktap_state_t *ks, int size);
+void *kp_zalloc(ktap_state_t *ks, int size);
+void kp_free(ktap_state_t *ks, void *addr);
+
+void kp_obj_dump(ktap_state_t *ks, const ktap_val_t *v);
+void kp_obj_show(ktap_state_t *ks, const ktap_val_t *v);
+int kp_obj_len(ktap_state_t *ks, const ktap_val_t *rb);
+ktap_obj_t *kp_obj_new(ktap_state_t *ks, size_t size);
+int kp_obj_rawequal(const ktap_val_t *t1, const ktap_val_t *t2);
+ktap_str_t *kp_obj_kstack2str(ktap_state_t *ks, uint16_t depth, uint16_t skip);
+void kp_obj_freeall(ktap_state_t *ks);
+
+#define kp_obj_equal(o1, o2) \
+ (((o1)->type == (o2)->type) && kp_obj_rawequal(o1, o2))
+
+#endif /* __KTAP_OBJ_H__ */
--
1.8.1.4


\
 
 \ /
  Last update: 2014-03-28 16:01    [W:0.171 / U:3.544 seconds]
©2003-2020 Jasper Spaans|hosted at Digital Ocean and TransIP|Read the blog|Advertise on this site