Hi,
On Thu, Jun 28, 2018 at 09:07:20PM +0800, joeyli wrote:
> Hi Chen Yu,
> 
> On Wed, Jun 20, 2018 at 05:40:32PM +0800, Chen Yu wrote:
> > Use the helper functions introduced previously to encrypt
> > the page data before they are submitted to the block device.
> > Besides, for the case of hibernation compression, the data
> > are firstly compressed and then encrypted, and vice versa
> > for the resume process.
> >
> 
> I want to suggest my solution that it direct signs/encrypts the
> memory snapshot image. This solution is already shipped with
> SLE12 a couple of years:
> 
> https://github.com/joeyli/linux-s4sign/commits/s4sign-hmac-encrypted-key-v0.2-v4.17-rc3
> 
I did not see image page encryption in above link, if I understand
correctly, your code focus on signation and encrypt the hidden data,
but not target for the whole snapshot data.
> The above patches still need to clean up. I am working on some
> other bugs, but I can clean up and send out it ASAP.
> 
> The advantage of this solution is that it produces a signed and
> encrypted image. Not just for writing to block device by kernel,
> it also can provide a signed/encrypted image to user space. User
> space can store the encrypted image to anywhere.
> 
> I am OK for your user space key generator because I didn't have
> similar solution yet. I am working on the EFI master key and also
> want to adapt hibernation to keyring. I will continue the works.
> 
The user space tool can easily add the keyring support besides
ioctl if needed.

Best,
Yu
> Thanks a lot!
> Joey Lee
>  
> > Suggested-by: Rafael J. Wysocki <rafael.j.wyso...@intel.com>
> > Cc: Rafael J. Wysocki <rafael.j.wyso...@intel.com>
> > Cc: Pavel Machek <pa...@ucw.cz>
> > Cc: Len Brown <len.br...@intel.com>
> > Cc: Borislav Petkov <b...@alien8.de>
> > Cc: "Lee, Chun-Yi" <j...@suse.com>
> > Cc: linux...@vger.kernel.org
> > Cc: linux-kernel@vger.kernel.org
> > Signed-off-by: Chen Yu <yu.c.c...@intel.com>
> > ---
> >  kernel/power/power.h |   1 +
> >  kernel/power/swap.c  | 215 
> > ++++++++++++++++++++++++++++++++++++++++++++++++---
> >  2 files changed, 205 insertions(+), 11 deletions(-)
> > 
> > diff --git a/kernel/power/power.h b/kernel/power/power.h
> > index 660aac3..637695c 100644
> > --- a/kernel/power/power.h
> > +++ b/kernel/power/power.h
> > @@ -207,6 +207,7 @@ extern int swsusp_swap_in_use(void);
> >  #define SF_PLATFORM_MODE   1
> >  #define SF_NOCOMPRESS_MODE 2
> >  #define SF_CRC32_MODE              4
> > +#define SF_ENCRYPT_MODE            8
> >  
> >  /* kernel/power/hibernate.c */
> >  extern int swsusp_check(void);
> > diff --git a/kernel/power/swap.c b/kernel/power/swap.c
> > index c2bcf97..2b6b3d0 100644
> > --- a/kernel/power/swap.c
> > +++ b/kernel/power/swap.c
> > @@ -102,14 +102,16 @@ struct swap_map_handle {
> >     unsigned int k;
> >     unsigned long reqd_free_pages;
> >     u32 crc32;
> > +   bool crypto;
> >  };
> >  
> >  struct swsusp_header {
> >     char reserved[PAGE_SIZE - 20 - sizeof(sector_t) - sizeof(int) -
> > -                 sizeof(u32)];
> > +                   sizeof(u32) - HIBERNATE_SALT_BYTES];
> >     u32     crc32;
> >     sector_t image;
> >     unsigned int flags;     /* Flags to pass to the "boot" kernel */
> > +   char salt[HIBERNATE_SALT_BYTES];
> >     char    orig_sig[10];
> >     char    sig[10];
> >  } __packed;
> > @@ -127,6 +129,53 @@ struct swsusp_extent {
> >     unsigned long end;
> >  };
> >  
> > +/* For encryption/decryption. */
> > +static struct hibernation_crypto *hibernation_crypto_ops;
> > +
> > +void set_hibernation_ops(struct hibernation_crypto *ops)
> > +{
> > +   hibernation_crypto_ops = ops;
> > +}
> > +EXPORT_SYMBOL_GPL(set_hibernation_ops);
> > +
> > +static int crypto_data(const char *inbuf,
> > +                       int inlen,
> > +                       char *outbuf,
> > +                       int outlen,
> > +                       bool encrypt,
> > +                       int page_idx)
> > +{
> > +   if (hibernation_crypto_ops &&
> > +       hibernation_crypto_ops->crypto_data)
> > +           return hibernation_crypto_ops->crypto_data(inbuf,
> > +                   inlen, outbuf, outlen, encrypt, page_idx);
> > +   else
> > +           return -EINVAL;
> > +}
> > +
> > +static void crypto_save(void *outbuf)
> > +{
> > +   if (hibernation_crypto_ops &&
> > +       hibernation_crypto_ops->save)
> > +           hibernation_crypto_ops->save(outbuf);
> > +}
> > +
> > +static void crypto_restore(void *inbuf)
> > +{
> > +   if (hibernation_crypto_ops &&
> > +       hibernation_crypto_ops->restore)
> > +           hibernation_crypto_ops->restore(inbuf);
> > +}
> > +
> > +static int crypto_init(bool suspend)
> > +{
> > +   if (hibernation_crypto_ops &&
> > +       hibernation_crypto_ops->init)
> > +           return hibernation_crypto_ops->init(suspend);
> > +   else
> > +           return -EINVAL;
> > +}
> > +
> >  static struct rb_root swsusp_extents = RB_ROOT;
> >  
> >  static int swsusp_extents_insert(unsigned long swap_offset)
> > @@ -318,6 +367,10 @@ static int mark_swapfiles(struct swap_map_handle 
> > *handle, unsigned int flags)
> >             swsusp_header->flags = flags;
> >             if (flags & SF_CRC32_MODE)
> >                     swsusp_header->crc32 = handle->crc32;
> > +           if (handle->crypto) {
> > +                   swsusp_header->flags |= SF_ENCRYPT_MODE;
> > +                   crypto_save((void *)swsusp_header->salt);
> > +           }
> >             error = hib_submit_io(REQ_OP_WRITE, REQ_SYNC,
> >                                   swsusp_resume_block, swsusp_header, NULL);
> >     } else {
> > @@ -535,11 +588,12 @@ static int save_image(struct swap_map_handle *handle,
> >  {
> >     unsigned int m;
> >     int ret;
> > -   int nr_pages;
> > +   int nr_pages, crypto_page_idx;
> >     int err2;
> >     struct hib_bio_batch hb;
> >     ktime_t start;
> >     ktime_t stop;
> > +   void *tmp = NULL, *crypt_buf = NULL;
> >  
> >     hib_init_batch(&hb);
> >  
> > @@ -549,12 +603,33 @@ static int save_image(struct swap_map_handle *handle,
> >     if (!m)
> >             m = 1;
> >     nr_pages = 0;
> > +   crypto_page_idx = 0;
> > +   if (handle->crypto) {
> > +           crypt_buf = (void *)get_zeroed_page(GFP_KERNEL);
> > +           if (!crypt_buf)
> > +                   return -ENOMEM;
> > +   }
> > +
> >     start = ktime_get();
> >     while (1) {
> >             ret = snapshot_read_next(snapshot);
> >             if (ret <= 0)
> >                     break;
> > -           ret = swap_write_page(handle, data_of(*snapshot), &hb);
> > +           tmp = data_of(*snapshot);
> > +           if (handle->crypto) {
> > +                   /* Encryption before submit_io.*/
> > +                   ret = crypto_data(data_of(*snapshot),
> > +                                     PAGE_SIZE,
> > +                                     crypt_buf,
> > +                                     PAGE_SIZE,
> > +                                     true,
> > +                                     crypto_page_idx);
> > +                   if (ret)
> > +                           goto out;
> > +                   crypto_page_idx++;
> > +                   tmp = crypt_buf;
> > +           }
> > +           ret = swap_write_page(handle, tmp, &hb);
> >             if (ret)
> >                     break;
> >             if (!(nr_pages % m))
> > @@ -569,6 +644,9 @@ static int save_image(struct swap_map_handle *handle,
> >     if (!ret)
> >             pr_info("Image saving done\n");
> >     swsusp_show_speed(start, stop, nr_to_write, "Wrote");
> > + out:
> > +   if (crypt_buf)
> > +           free_page((unsigned long)crypt_buf);
> >     return ret;
> >  }
> >  
> > @@ -671,7 +749,7 @@ static int save_image_lzo(struct swap_map_handle 
> > *handle,
> >  {
> >     unsigned int m;
> >     int ret = 0;
> > -   int nr_pages;
> > +   int nr_pages, crypto_page_idx;
> >     int err2;
> >     struct hib_bio_batch hb;
> >     ktime_t start;
> > @@ -767,6 +845,7 @@ static int save_image_lzo(struct swap_map_handle 
> > *handle,
> >     if (!m)
> >             m = 1;
> >     nr_pages = 0;
> > +   crypto_page_idx = 0;
> >     start = ktime_get();
> >     for (;;) {
> >             for (thr = 0; thr < nr_threads; thr++) {
> > @@ -835,7 +914,25 @@ static int save_image_lzo(struct swap_map_handle 
> > *handle,
> >                     for (off = 0;
> >                          off < LZO_HEADER + data[thr].cmp_len;
> >                          off += PAGE_SIZE) {
> > -                           memcpy(page, data[thr].cmp + off, PAGE_SIZE);
> > +                           if (handle->crypto) {
> > +                                   /*
> > +                                    * Encrypt the compressed data
> > +                                    * before we write them to the
> > +                                    * block device.
> > +                                    */
> > +                                   ret = crypto_data(data[thr].cmp + off,
> > +                                                     PAGE_SIZE,
> > +                                                     page,
> > +                                                     PAGE_SIZE,
> > +                                                     true,
> > +                                                     crypto_page_idx);
> > +                                   if (ret)
> > +                                           goto out_finish;
> > +                                   crypto_page_idx++;
> > +                           } else {
> > +                                   memcpy(page, data[thr].cmp + off,
> > +                                           PAGE_SIZE);
> > +                           }
> >  
> >                             ret = swap_write_page(handle, page, &hb);
> >                             if (ret)
> > @@ -909,6 +1006,7 @@ int swsusp_write(unsigned int flags)
> >     int error;
> >  
> >     pages = snapshot_get_image_size();
> > +   memset(&handle, 0, sizeof(struct swap_map_handle));
> >     error = get_swap_writer(&handle);
> >     if (error) {
> >             pr_err("Cannot get swap writer\n");
> > @@ -922,6 +1020,9 @@ int swsusp_write(unsigned int flags)
> >             }
> >     }
> >     memset(&snapshot, 0, sizeof(struct snapshot_handle));
> > +   if (!crypto_init(true))
> > +           /* The image needs to be encrypted. */
> > +           handle.crypto = true;
> >     error = snapshot_read_next(&snapshot);
> >     if (error < PAGE_SIZE) {
> >             if (error >= 0)
> > @@ -1059,7 +1160,8 @@ static int load_image(struct swap_map_handle *handle,
> >     ktime_t stop;
> >     struct hib_bio_batch hb;
> >     int err2;
> > -   unsigned nr_pages;
> > +   unsigned nr_pages, crypto_page_idx;
> > +   void *crypt_buf = NULL;
> >  
> >     hib_init_batch(&hb);
> >  
> > @@ -1069,18 +1171,42 @@ static int load_image(struct swap_map_handle 
> > *handle,
> >     if (!m)
> >             m = 1;
> >     nr_pages = 0;
> > +   crypto_page_idx = 0;
> > +   if (handle->crypto) {
> > +           crypt_buf = (void *)get_zeroed_page(GFP_KERNEL);
> > +           if (!crypt_buf)
> > +                   return -ENOMEM;
> > +   }
> >     start = ktime_get();
> >     for ( ; ; ) {
> >             ret = snapshot_write_next(snapshot);
> >             if (ret <= 0)
> >                     break;
> > -           ret = swap_read_page(handle, data_of(*snapshot), &hb);
> > +           if (handle->crypto)
> > +                   ret = swap_read_page(handle, crypt_buf, &hb);
> > +           else
> > +                   ret = swap_read_page(handle, data_of(*snapshot), &hb);
> >             if (ret)
> >                     break;
> >             if (snapshot->sync_read)
> >                     ret = hib_wait_io(&hb);
> >             if (ret)
> >                     break;
> > +           if (handle->crypto) {
> > +                   /*
> > +                    * Need a decryption for the
> > +                    * data read from the block
> > +                    * device.
> > +                    */
> > +                   ret = crypto_data(crypt_buf, PAGE_SIZE,
> > +                                     data_of(*snapshot),
> > +                                     PAGE_SIZE,
> > +                                     false,
> > +                                     crypto_page_idx);
> > +                   if (ret)
> > +                           break;
> > +                   crypto_page_idx++;
> > +           }
> >             if (!(nr_pages % m))
> >                     pr_info("Image loading progress: %3d%%\n",
> >                             nr_pages / m * 10);
> > @@ -1097,6 +1223,8 @@ static int load_image(struct swap_map_handle *handle,
> >                     ret = -ENODATA;
> >     }
> >     swsusp_show_speed(start, stop, nr_to_read, "Read");
> > +   if (crypt_buf)
> > +           free_page((unsigned long)crypt_buf);
> >     return ret;
> >  }
> >  
> > @@ -1164,7 +1292,7 @@ static int load_image_lzo(struct swap_map_handle 
> > *handle,
> >     struct hib_bio_batch hb;
> >     ktime_t start;
> >     ktime_t stop;
> > -   unsigned nr_pages;
> > +   unsigned nr_pages, crypto_page_idx;
> >     size_t off;
> >     unsigned i, thr, run_threads, nr_threads;
> >     unsigned ring = 0, pg = 0, ring_size = 0,
> > @@ -1173,6 +1301,7 @@ static int load_image_lzo(struct swap_map_handle 
> > *handle,
> >     unsigned char **page = NULL;
> >     struct dec_data *data = NULL;
> >     struct crc_data *crc = NULL;
> > +   void *first_page = NULL;
> >  
> >     hib_init_batch(&hb);
> >  
> > @@ -1278,6 +1407,18 @@ static int load_image_lzo(struct swap_map_handle 
> > *handle,
> >     }
> >     want = ring_size = i;
> >  
> > +   /*
> > +    * The first page of data[thr] contains the length of
> > +    * compressed data, this page should not mess up the
> > +    * read buffer, so we allocate a separate page for it.
> > +    */
> > +   if (handle->crypto) {
> > +           first_page = (void *)get_zeroed_page(GFP_KERNEL);
> > +           if (!first_page) {
> > +                   ret = -ENOMEM;
> > +                   goto out_clean;
> > +           }
> > +   }
> >     pr_info("Using %u thread(s) for decompression\n", nr_threads);
> >     pr_info("Loading and decompressing image data (%u pages)...\n",
> >             nr_to_read);
> > @@ -1285,6 +1426,7 @@ static int load_image_lzo(struct swap_map_handle 
> > *handle,
> >     if (!m)
> >             m = 1;
> >     nr_pages = 0;
> > +   crypto_page_idx = 0;
> >     start = ktime_get();
> >  
> >     ret = snapshot_write_next(snapshot);
> > @@ -1336,7 +1478,24 @@ static int load_image_lzo(struct swap_map_handle 
> > *handle,
> >             }
> >  
> >             for (thr = 0; have && thr < nr_threads; thr++) {
> > -                   data[thr].cmp_len = *(size_t *)page[pg];
> > +                   if (handle->crypto) {
> > +                           /*
> > +                            * Need to decrypt the first page
> > +                            * of each data[thr], which contains
> > +                            * the compressed data length.
> > +                            */
> > +                           ret = crypto_data(page[pg],
> > +                                             PAGE_SIZE,
> > +                                             first_page,
> > +                                             PAGE_SIZE,
> > +                                             false,
> > +                                             crypto_page_idx);
> > +                           if (ret)
> > +                                   goto out_finish;
> > +                           data[thr].cmp_len = *(size_t *)first_page;
> > +                   } else {
> > +                           data[thr].cmp_len = *(size_t *)page[pg];
> > +                   }
> >                     if (unlikely(!data[thr].cmp_len ||
> >                                  data[thr].cmp_len >
> >                                  lzo1x_worst_compress(LZO_UNC_SIZE))) {
> > @@ -1358,8 +1517,26 @@ static int load_image_lzo(struct swap_map_handle 
> > *handle,
> >                     for (off = 0;
> >                          off < LZO_HEADER + data[thr].cmp_len;
> >                          off += PAGE_SIZE) {
> > -                           memcpy(data[thr].cmp + off,
> > -                                  page[pg], PAGE_SIZE);
> > +                           if (handle->crypto) {
> > +                                   /*
> > +                                    * Decrypt the compressed data
> > +                                    * and leverage the decompression
> > +                                    * threads to get it done.
> > +                                    */
> > +                                   ret = crypto_data(page[pg],
> > +                                                     PAGE_SIZE,
> > +                                                     data[thr].cmp + off,
> > +                                                     PAGE_SIZE,
> > +                                                     false,
> > +                                                     crypto_page_idx);
> > +                                   if (ret)
> > +                                           goto out_finish;
> > +                                   crypto_page_idx++;
> > +                           } else {
> > +                                   memcpy(data[thr].cmp + off,
> > +                                           page[pg], PAGE_SIZE);
> > +
> > +                           }
> >                             have--;
> >                             want++;
> >                             if (++pg >= ring_size)
> > @@ -1452,6 +1629,8 @@ static int load_image_lzo(struct swap_map_handle 
> > *handle,
> >  out_clean:
> >     for (i = 0; i < ring_size; i++)
> >             free_page((unsigned long)page[i]);
> > +   if (first_page)
> > +           free_page((unsigned long)first_page);
> >     if (crc) {
> >             if (crc->thr)
> >                     kthread_stop(crc->thr);
> > @@ -1482,6 +1661,7 @@ int swsusp_read(unsigned int *flags_p)
> >     struct swsusp_info *header;
> >  
> >     memset(&snapshot, 0, sizeof(struct snapshot_handle));
> > +   memset(&handle, 0, sizeof(struct swap_map_handle));
> >     error = snapshot_write_next(&snapshot);
> >     if (error < PAGE_SIZE)
> >             return error < 0 ? error : -EFAULT;
> > @@ -1489,6 +1669,16 @@ int swsusp_read(unsigned int *flags_p)
> >     error = get_swap_reader(&handle, flags_p);
> >     if (error)
> >             goto end;
> > +   if (*flags_p & SF_ENCRYPT_MODE) {
> > +           error = crypto_init(false);
> > +           if (!error) {
> > +                   /* The image has been encrypted. */
> > +                   handle.crypto = true;
> > +           } else {
> > +                   pr_err("Failed to init cipher during resume.\n");
> > +                   goto end;
> > +           }
> > +   }
> >     if (!error)
> >             error = swap_read_page(&handle, header, NULL);
> >     if (!error) {
> > @@ -1526,6 +1716,9 @@ int swsusp_check(void)
> >  
> >             if (!memcmp(HIBERNATE_SIG, swsusp_header->sig, 10)) {
> >                     memcpy(swsusp_header->sig, swsusp_header->orig_sig, 10);
> > +                   /* Read salt passed from previous kernel. */
> > +                   if (swsusp_header->flags & SF_ENCRYPT_MODE)
> > +                           crypto_restore((void *)&swsusp_header->salt);
> >                     /* Reset swap signature now */
> >                     error = hib_submit_io(REQ_OP_WRITE, REQ_SYNC,
> >                                             swsusp_resume_block,
> > -- 
> > 2.7.4
> > 

Reply via email to