|
@@ -20,15 +20,9 @@
|
|
|
|
|
|
#define NO_FURTHER_WRITE_ACTION -1
|
|
|
|
|
|
-struct capsule_info {
|
|
|
- bool header_obtained;
|
|
|
- int reset_type;
|
|
|
- long index;
|
|
|
- size_t count;
|
|
|
- size_t total_size;
|
|
|
- struct page **pages;
|
|
|
- size_t page_bytes_remain;
|
|
|
-};
|
|
|
+#ifndef phys_to_page
|
|
|
+#define phys_to_page(x) pfn_to_page((x) >> PAGE_SHIFT)
|
|
|
+#endif
|
|
|
|
|
|
/**
|
|
|
* efi_free_all_buff_pages - free all previous allocated buffer pages
|
|
@@ -41,64 +35,69 @@ struct capsule_info {
|
|
|
static void efi_free_all_buff_pages(struct capsule_info *cap_info)
|
|
|
{
|
|
|
while (cap_info->index > 0)
|
|
|
- __free_page(cap_info->pages[--cap_info->index]);
|
|
|
+ __free_page(phys_to_page(cap_info->pages[--cap_info->index]));
|
|
|
|
|
|
cap_info->index = NO_FURTHER_WRITE_ACTION;
|
|
|
}
|
|
|
|
|
|
-/**
|
|
|
- * efi_capsule_setup_info - obtain the efi capsule header in the binary and
|
|
|
- * setup capsule_info structure
|
|
|
- * @cap_info: pointer to current instance of capsule_info structure
|
|
|
- * @kbuff: a mapped first page buffer pointer
|
|
|
- * @hdr_bytes: the total received number of bytes for efi header
|
|
|
- **/
|
|
|
-static ssize_t efi_capsule_setup_info(struct capsule_info *cap_info,
|
|
|
- void *kbuff, size_t hdr_bytes)
|
|
|
+int __efi_capsule_setup_info(struct capsule_info *cap_info)
|
|
|
{
|
|
|
- efi_capsule_header_t *cap_hdr;
|
|
|
size_t pages_needed;
|
|
|
int ret;
|
|
|
void *temp_page;
|
|
|
|
|
|
- /* Only process data block that is larger than efi header size */
|
|
|
- if (hdr_bytes < sizeof(efi_capsule_header_t))
|
|
|
- return 0;
|
|
|
-
|
|
|
- /* Reset back to the correct offset of header */
|
|
|
- cap_hdr = kbuff - cap_info->count;
|
|
|
- pages_needed = ALIGN(cap_hdr->imagesize, PAGE_SIZE) >> PAGE_SHIFT;
|
|
|
+ pages_needed = ALIGN(cap_info->total_size, PAGE_SIZE) / PAGE_SIZE;
|
|
|
|
|
|
if (pages_needed == 0) {
|
|
|
- pr_err("%s: pages count invalid\n", __func__);
|
|
|
+ pr_err("invalid capsule size");
|
|
|
return -EINVAL;
|
|
|
}
|
|
|
|
|
|
/* Check if the capsule binary supported */
|
|
|
- ret = efi_capsule_supported(cap_hdr->guid, cap_hdr->flags,
|
|
|
- cap_hdr->imagesize,
|
|
|
+ ret = efi_capsule_supported(cap_info->header.guid,
|
|
|
+ cap_info->header.flags,
|
|
|
+ cap_info->header.imagesize,
|
|
|
&cap_info->reset_type);
|
|
|
if (ret) {
|
|
|
- pr_err("%s: efi_capsule_supported() failed\n",
|
|
|
- __func__);
|
|
|
+ pr_err("capsule not supported\n");
|
|
|
return ret;
|
|
|
}
|
|
|
|
|
|
- cap_info->total_size = cap_hdr->imagesize;
|
|
|
temp_page = krealloc(cap_info->pages,
|
|
|
pages_needed * sizeof(void *),
|
|
|
GFP_KERNEL | __GFP_ZERO);
|
|
|
- if (!temp_page) {
|
|
|
- pr_debug("%s: krealloc() failed\n", __func__);
|
|
|
+ if (!temp_page)
|
|
|
return -ENOMEM;
|
|
|
- }
|
|
|
|
|
|
cap_info->pages = temp_page;
|
|
|
- cap_info->header_obtained = true;
|
|
|
|
|
|
return 0;
|
|
|
}
|
|
|
|
|
|
+/**
|
|
|
+ * efi_capsule_setup_info - obtain the efi capsule header in the binary and
|
|
|
+ * setup capsule_info structure
|
|
|
+ * @cap_info: pointer to current instance of capsule_info structure
|
|
|
+ * @kbuff: a mapped first page buffer pointer
|
|
|
+ * @hdr_bytes: the total received number of bytes for efi header
|
|
|
+ *
|
|
|
+ * Platforms with non-standard capsule update mechanisms can override
|
|
|
+ * this __weak function so they can perform any required capsule
|
|
|
+ * image munging. See quark_quirk_function() for an example.
|
|
|
+ **/
|
|
|
+int __weak efi_capsule_setup_info(struct capsule_info *cap_info, void *kbuff,
|
|
|
+ size_t hdr_bytes)
|
|
|
+{
|
|
|
+ /* Only process data block that is larger than efi header size */
|
|
|
+ if (hdr_bytes < sizeof(efi_capsule_header_t))
|
|
|
+ return 0;
|
|
|
+
|
|
|
+ memcpy(&cap_info->header, kbuff, sizeof(cap_info->header));
|
|
|
+ cap_info->total_size = cap_info->header.imagesize;
|
|
|
+
|
|
|
+ return __efi_capsule_setup_info(cap_info);
|
|
|
+}
|
|
|
+
|
|
|
/**
|
|
|
* efi_capsule_submit_update - invoke the efi_capsule_update API once binary
|
|
|
* upload done
|
|
@@ -107,26 +106,17 @@ static ssize_t efi_capsule_setup_info(struct capsule_info *cap_info,
|
|
|
static ssize_t efi_capsule_submit_update(struct capsule_info *cap_info)
|
|
|
{
|
|
|
int ret;
|
|
|
- void *cap_hdr_temp;
|
|
|
-
|
|
|
- cap_hdr_temp = vmap(cap_info->pages, cap_info->index,
|
|
|
- VM_MAP, PAGE_KERNEL);
|
|
|
- if (!cap_hdr_temp) {
|
|
|
- pr_debug("%s: vmap() failed\n", __func__);
|
|
|
- return -EFAULT;
|
|
|
- }
|
|
|
|
|
|
- ret = efi_capsule_update(cap_hdr_temp, cap_info->pages);
|
|
|
- vunmap(cap_hdr_temp);
|
|
|
+ ret = efi_capsule_update(&cap_info->header, cap_info->pages);
|
|
|
if (ret) {
|
|
|
- pr_err("%s: efi_capsule_update() failed\n", __func__);
|
|
|
+ pr_err("capsule update failed\n");
|
|
|
return ret;
|
|
|
}
|
|
|
|
|
|
/* Indicate capsule binary uploading is done */
|
|
|
cap_info->index = NO_FURTHER_WRITE_ACTION;
|
|
|
- pr_info("%s: Successfully upload capsule file with reboot type '%s'\n",
|
|
|
- __func__, !cap_info->reset_type ? "RESET_COLD" :
|
|
|
+ pr_info("Successfully upload capsule file with reboot type '%s'\n",
|
|
|
+ !cap_info->reset_type ? "RESET_COLD" :
|
|
|
cap_info->reset_type == 1 ? "RESET_WARM" :
|
|
|
"RESET_SHUTDOWN");
|
|
|
return 0;
|
|
@@ -171,37 +161,30 @@ static ssize_t efi_capsule_write(struct file *file, const char __user *buff,
|
|
|
if (!cap_info->page_bytes_remain) {
|
|
|
page = alloc_page(GFP_KERNEL);
|
|
|
if (!page) {
|
|
|
- pr_debug("%s: alloc_page() failed\n", __func__);
|
|
|
ret = -ENOMEM;
|
|
|
goto failed;
|
|
|
}
|
|
|
|
|
|
- cap_info->pages[cap_info->index++] = page;
|
|
|
+ cap_info->pages[cap_info->index++] = page_to_phys(page);
|
|
|
cap_info->page_bytes_remain = PAGE_SIZE;
|
|
|
+ } else {
|
|
|
+ page = phys_to_page(cap_info->pages[cap_info->index - 1]);
|
|
|
}
|
|
|
|
|
|
- page = cap_info->pages[cap_info->index - 1];
|
|
|
-
|
|
|
kbuff = kmap(page);
|
|
|
- if (!kbuff) {
|
|
|
- pr_debug("%s: kmap() failed\n", __func__);
|
|
|
- ret = -EFAULT;
|
|
|
- goto failed;
|
|
|
- }
|
|
|
kbuff += PAGE_SIZE - cap_info->page_bytes_remain;
|
|
|
|
|
|
/* Copy capsule binary data from user space to kernel space buffer */
|
|
|
write_byte = min_t(size_t, count, cap_info->page_bytes_remain);
|
|
|
if (copy_from_user(kbuff, buff, write_byte)) {
|
|
|
- pr_debug("%s: copy_from_user() failed\n", __func__);
|
|
|
ret = -EFAULT;
|
|
|
goto fail_unmap;
|
|
|
}
|
|
|
cap_info->page_bytes_remain -= write_byte;
|
|
|
|
|
|
/* Setup capsule binary info structure */
|
|
|
- if (!cap_info->header_obtained) {
|
|
|
- ret = efi_capsule_setup_info(cap_info, kbuff,
|
|
|
+ if (cap_info->header.headersize == 0) {
|
|
|
+ ret = efi_capsule_setup_info(cap_info, kbuff - cap_info->count,
|
|
|
cap_info->count + write_byte);
|
|
|
if (ret)
|
|
|
goto fail_unmap;
|
|
@@ -211,11 +194,10 @@ static ssize_t efi_capsule_write(struct file *file, const char __user *buff,
|
|
|
kunmap(page);
|
|
|
|
|
|
/* Submit the full binary to efi_capsule_update() API */
|
|
|
- if (cap_info->header_obtained &&
|
|
|
+ if (cap_info->header.headersize > 0 &&
|
|
|
cap_info->count >= cap_info->total_size) {
|
|
|
if (cap_info->count > cap_info->total_size) {
|
|
|
- pr_err("%s: upload size exceeded header defined size\n",
|
|
|
- __func__);
|
|
|
+ pr_err("capsule upload size exceeded header defined size\n");
|
|
|
ret = -EINVAL;
|
|
|
goto failed;
|
|
|
}
|
|
@@ -249,7 +231,7 @@ static int efi_capsule_flush(struct file *file, fl_owner_t id)
|
|
|
struct capsule_info *cap_info = file->private_data;
|
|
|
|
|
|
if (cap_info->index > 0) {
|
|
|
- pr_err("%s: capsule upload not complete\n", __func__);
|
|
|
+ pr_err("capsule upload not complete\n");
|
|
|
efi_free_all_buff_pages(cap_info);
|
|
|
ret = -ECANCELED;
|
|
|
}
|
|
@@ -328,8 +310,7 @@ static int __init efi_capsule_loader_init(void)
|
|
|
|
|
|
ret = misc_register(&efi_capsule_misc);
|
|
|
if (ret)
|
|
|
- pr_err("%s: Failed to register misc char file note\n",
|
|
|
- __func__);
|
|
|
+ pr_err("Unable to register capsule loader device\n");
|
|
|
|
|
|
return ret;
|
|
|
}
|