Branch data Line data Source code
1 : : #include <unistd.h>
2 : : #include <sys/types.h>
3 : : #include <sys/stat.h>
4 : : #include <fcntl.h>
5 : : #include <stdlib.h>
6 : :
7 : : #include "crtools.h"
8 : : #include "image.h"
9 : : #include "files.h"
10 : : #include "files-reg.h"
11 : : #include "pipes.h"
12 : :
13 : : #include "fifo.h"
14 : :
15 : : #include "protobuf.h"
16 : : #include "protobuf/regfile.pb-c.h"
17 : : #include "protobuf/fifo.pb-c.h"
18 : :
19 : : /*
20 : : * FIFO checkpoint and restore is done in a bit unusual manner.
21 : : * We use files-reg.c engine to save fifo path and flags,
22 : : * thus regular files image will contain fifo descriptos which
23 : : * are useless for reg-files engine itself but needed for our fifo
24 : : * engine.
25 : : *
26 : : * In particual we dump fifo-entry automatically and appropriate
27 : : * reg-file entry manually, thus on restore we need to ask reg-file
28 : : * engine to restore fifo path and flags via direct call.
29 : : */
30 : :
31 : : struct fifo_info {
32 : : struct list_head list;
33 : : struct file_desc d;
34 : : FifoEntry *fe;
35 : : bool restore_data;
36 : : };
37 : :
38 : : static LIST_HEAD(fifo_head);
39 : : static struct pipe_data_dump pd_fifo = { .img_type = CR_FD_FIFO_DATA, };
40 : :
41 : 14 : static int dump_one_fifo(int lfd, u32 id, const struct fd_parms *p)
42 : : {
43 : 14 : int img = fdset_fd(glob_fdset, CR_FD_FIFO);
44 : 14 : FifoEntry e = FIFO_ENTRY__INIT;
45 : :
46 : : /*
47 : : * It's a trick here, we use regular files dumping
48 : : * code to save path to a fifo, then we reuse it
49 : : * on restore.
50 : : */
51 [ + - ]: 14 : if (dump_one_reg_file(lfd, id, p))
52 : : return -1;
53 : :
54 : 14 : pr_info("Dumping fifo %d with id %#x pipe_id %#x\n",
55 : : lfd, id, pipe_id(p));
56 : :
57 : 14 : e.id = id;
58 : 28 : e.pipe_id = pipe_id(p);
59 : :
60 [ + - ]: 14 : if (pb_write_one(img, &e, PB_FIFO))
61 : : return -1;
62 : :
63 : 14 : return dump_one_pipe_data(&pd_fifo, lfd, p);
64 : : }
65 : :
66 : : static const struct fdtype_ops fifo_ops = {
67 : : .type = FD_TYPES__FIFO,
68 : : .dump = dump_one_fifo,
69 : : };
70 : :
71 : 16 : int dump_fifo(struct fd_parms *p, int lfd, const struct cr_fdset *set)
72 : : {
73 : 16 : return do_dump_gen_file(p, lfd, &fifo_ops, set);
74 : : }
75 : :
76 : : static struct pipe_data_rst *pd_hash_fifo[PIPE_DATA_HASH_SIZE];
77 : :
78 : 14 : static int do_open_fifo(struct reg_file_info *rfi, void *arg)
79 : : {
80 : 14 : struct fifo_info *info = arg;
81 : 14 : int new_fifo, fake_fifo = -1;
82 : :
83 : : /*
84 : : * The fifos (except read-write fifos) do wait until
85 : : * another pipe-end get connected, so to be able to
86 : : * proceed the restoration procedure we open a fake
87 : : * fifo here.
88 : : */
89 : 14 : fake_fifo = open(rfi->path, O_RDWR);
90 [ - + ]: 14 : if (fake_fifo < 0) {
91 : 0 : pr_perror("Can't open fake fifo %#x [%s]", info->fe->id, rfi->path);
92 : 0 : return -1;
93 : : }
94 : :
95 : 14 : new_fifo = open(rfi->path, rfi->rfe->flags);
96 [ - + ]: 14 : if (new_fifo < 0) {
97 : 0 : pr_perror("Can't open fifo %#x [%s]", info->fe->id, rfi->path);
98 : 0 : goto out;
99 : : }
100 : :
101 [ + + ]: 14 : if (info->restore_data)
102 [ - + ]: 10 : if (restore_pipe_data(CR_FD_FIFO_DATA, fake_fifo,
103 : 10 : info->fe->pipe_id, pd_hash_fifo)) {
104 : 0 : close(new_fifo);
105 : 0 : new_fifo = -1;
106 : : }
107 : :
108 : : out:
109 : 14 : close(fake_fifo);
110 : 14 : return new_fifo;
111 : : }
112 : :
113 : 14 : static int open_fifo_fd(struct file_desc *d)
114 : : {
115 : 14 : struct fifo_info *info = container_of(d, struct fifo_info, d);
116 : :
117 : 14 : return open_path_by_id(info->fe->id, do_open_fifo, info);
118 : : }
119 : :
120 : : static struct file_desc_ops fifo_desc_ops = {
121 : : .type = FD_TYPES__FIFO,
122 : : .open = open_fifo_fd,
123 : : };
124 : :
125 : 31 : static int collect_one_fifo(void *o, ProtobufCMessage *base)
126 : : {
127 : 31 : struct fifo_info *info = o, *f;
128 : :
129 : 31 : info->fe = pb_msg(base, FifoEntry);
130 : 31 : pr_info("Collected fifo entry ID %#x PIPE ID %#x\n",
131 : : info->fe->id, info->fe->pipe_id);
132 : :
133 : 31 : file_desc_add(&info->d, info->fe->id, &fifo_desc_ops);
134 : :
135 : : /* check who will restore the fifo data */
136 [ + + ]: 36 : list_for_each_entry(f, &fifo_head, list)
137 [ + + ]: 15 : if (f->fe->pipe_id == info->fe->pipe_id)
138 : : break;
139 : :
140 [ + + ]: 31 : if (&f->list == &fifo_head) {
141 : 21 : list_add(&info->list, &fifo_head);
142 : 21 : info->restore_data = true;
143 : : } else {
144 : 10 : INIT_LIST_HEAD(&info->list);
145 : 10 : info->restore_data = false;
146 : : }
147 : :
148 : 31 : return 0;
149 : : }
150 : :
151 : 355 : int collect_fifo(void)
152 : : {
153 : : int ret;
154 : :
155 : 355 : ret = collect_image(CR_FD_FIFO, PB_FIFO,
156 : : sizeof(struct fifo_info), collect_one_fifo);
157 [ + - ]: 355 : if (!ret)
158 : 355 : ret = collect_pipe_data(CR_FD_FIFO_DATA, pd_hash_fifo);
159 : :
160 : 355 : return ret;
161 : 28 : }
|
