hc
2024-12-19 9370bb92b2d16684ee45cf24e879c93c509162da
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
121
122
123
124
125
126
127
128
// SPDX-License-Identifier: GPL-2.0
/*
 * Copyright IBM Corp. 2016
 * Author(s): Martin Schwidefsky <schwidefsky@de.ibm.com>
 */
 
#include <linux/kernel.h>
#include <linux/syscalls.h>
#include <linux/signal.h>
#include <linux/mm.h>
#include <linux/slab.h>
#include <asm/guarded_storage.h>
#include "entry.h"
 
void guarded_storage_release(struct task_struct *tsk)
{
   kfree(tsk->thread.gs_cb);
   kfree(tsk->thread.gs_bc_cb);
}
 
static int gs_enable(void)
{
   struct gs_cb *gs_cb;
 
   if (!current->thread.gs_cb) {
       gs_cb = kzalloc(sizeof(*gs_cb), GFP_KERNEL);
       if (!gs_cb)
           return -ENOMEM;
       gs_cb->gsd = 25;
       preempt_disable();
       __ctl_set_bit(2, 4);
       load_gs_cb(gs_cb);
       current->thread.gs_cb = gs_cb;
       preempt_enable();
   }
   return 0;
}
 
static int gs_disable(void)
{
   if (current->thread.gs_cb) {
       preempt_disable();
       kfree(current->thread.gs_cb);
       current->thread.gs_cb = NULL;
       __ctl_clear_bit(2, 4);
       preempt_enable();
   }
   return 0;
}
 
static int gs_set_bc_cb(struct gs_cb __user *u_gs_cb)
{
   struct gs_cb *gs_cb;
 
   gs_cb = current->thread.gs_bc_cb;
   if (!gs_cb) {
       gs_cb = kzalloc(sizeof(*gs_cb), GFP_KERNEL);
       if (!gs_cb)
           return -ENOMEM;
       current->thread.gs_bc_cb = gs_cb;
   }
   if (copy_from_user(gs_cb, u_gs_cb, sizeof(*gs_cb)))
       return -EFAULT;
   return 0;
}
 
static int gs_clear_bc_cb(void)
{
   struct gs_cb *gs_cb;
 
   gs_cb = current->thread.gs_bc_cb;
   current->thread.gs_bc_cb = NULL;
   kfree(gs_cb);
   return 0;
}
 
void gs_load_bc_cb(struct pt_regs *regs)
{
   struct gs_cb *gs_cb;
 
   preempt_disable();
   clear_thread_flag(TIF_GUARDED_STORAGE);
   gs_cb = current->thread.gs_bc_cb;
   if (gs_cb) {
       kfree(current->thread.gs_cb);
       current->thread.gs_bc_cb = NULL;
       __ctl_set_bit(2, 4);
       load_gs_cb(gs_cb);
       current->thread.gs_cb = gs_cb;
   }
   preempt_enable();
}
 
static int gs_broadcast(void)
{
   struct task_struct *sibling;
 
   read_lock(&tasklist_lock);
   for_each_thread(current, sibling) {
       if (!sibling->thread.gs_bc_cb)
           continue;
       if (test_and_set_tsk_thread_flag(sibling, TIF_GUARDED_STORAGE))
           kick_process(sibling);
   }
   read_unlock(&tasklist_lock);
   return 0;
}
 
SYSCALL_DEFINE2(s390_guarded_storage, int, command,
       struct gs_cb __user *, gs_cb)
{
   if (!MACHINE_HAS_GS)
       return -EOPNOTSUPP;
   switch (command) {
   case GS_ENABLE:
       return gs_enable();
   case GS_DISABLE:
       return gs_disable();
   case GS_SET_BC_CB:
       return gs_set_bc_cb(gs_cb);
   case GS_CLEAR_BC_CB:
       return gs_clear_bc_cb();
   case GS_BROADCAST:
       return gs_broadcast();
   default:
       return -EINVAL;
   }
}