* Determine whether the given stack trace includes any references to a
  * to-be-patched or to-be-unpatched function.
  */
-static int klp_check_stack_func(struct klp_func *func,
-                               struct stack_trace *trace)
+static int klp_check_stack_func(struct klp_func *func, unsigned long *entries,
+                               unsigned int nr_entries)
 {
        unsigned long func_addr, func_size, address;
        struct klp_ops *ops;
        int i;
 
-       for (i = 0; i < trace->nr_entries; i++) {
-               address = trace->entries[i];
+       for (i = 0; i < nr_entries; i++) {
+               address = entries[i];
 
                if (klp_target_state == KLP_UNPATCHED) {
                         /*
 static int klp_check_stack(struct task_struct *task, char *err_buf)
 {
        static unsigned long entries[MAX_STACK_ENTRIES];
-       struct stack_trace trace;
        struct klp_object *obj;
        struct klp_func *func;
-       int ret;
+       int ret, nr_entries;
 
-       trace.skip = 0;
-       trace.nr_entries = 0;
-       trace.max_entries = MAX_STACK_ENTRIES;
-       trace.entries = entries;
-       ret = save_stack_trace_tsk_reliable(task, &trace);
+       ret = stack_trace_save_tsk_reliable(task, entries, ARRAY_SIZE(entries));
        WARN_ON_ONCE(ret == -ENOSYS);
-       if (ret) {
+       if (ret < 0) {
                snprintf(err_buf, STACK_ERR_BUF_SIZE,
                         "%s: %s:%d has an unreliable stack\n",
                         __func__, task->comm, task->pid);
                return ret;
        }
+       nr_entries = ret;
 
        klp_for_each_object(klp_transition_patch, obj) {
                if (!obj->patched)
                        continue;
                klp_for_each_func(obj, func) {
-                       ret = klp_check_stack_func(func, &trace);
+                       ret = klp_check_stack_func(func, entries, nr_entries);
                        if (ret) {
                                snprintf(err_buf, STACK_ERR_BUF_SIZE,
                                         "%s: %s:%d is sleeping on function %s\n",