pstore: Replace arguments for write_buf_user() API

Removes argument list in favor of pstore record, though the user buffer
remains passed separately since it must carry the __user annotation.

Signed-off-by: Kees Cook <keescook@chromium.org>
This commit is contained in:
Kees Cook 2017-03-05 00:56:38 -08:00
parent b10b471145
commit fdd0311863
4 changed files with 29 additions and 50 deletions

View File

@ -639,47 +639,36 @@ static int pstore_write_compat(struct pstore_record *record)
return record->psi->write_buf(record); return record->psi->write_buf(record);
} }
static int pstore_write_buf_user_compat(enum pstore_type_id type, static int pstore_write_buf_user_compat(struct pstore_record *record,
enum kmsg_dump_reason reason, const char __user *buf)
u64 *id, unsigned int part,
const char __user *buf,
bool compressed, size_t size,
struct pstore_info *psi)
{ {
unsigned long flags = 0; unsigned long flags = 0;
size_t i, bufsize = size; size_t i, bufsize, total_size = record->size;
long ret = 0; long ret = 0;
if (unlikely(!access_ok(VERIFY_READ, buf, size))) if (unlikely(!access_ok(VERIFY_READ, buf, total_size)))
return -EFAULT; return -EFAULT;
bufsize = total_size;
if (bufsize > psinfo->bufsize) if (bufsize > psinfo->bufsize)
bufsize = psinfo->bufsize; bufsize = psinfo->bufsize;
record->buf = psinfo->buf;
spin_lock_irqsave(&psinfo->buf_lock, flags); spin_lock_irqsave(&psinfo->buf_lock, flags);
for (i = 0; i < size; ) { for (i = 0; i < total_size; ) {
struct pstore_record record = { size_t c = min(total_size - i, bufsize);
.type = type,
.reason = reason,
.id = id,
.part = part,
.buf = psinfo->buf,
.compressed = compressed,
.psi = psi,
};
size_t c = min(size - i, bufsize);
ret = __copy_from_user(psinfo->buf, buf + i, c); ret = __copy_from_user(record->buf, buf + i, c);
if (unlikely(ret != 0)) { if (unlikely(ret != 0)) {
ret = -EFAULT; ret = -EFAULT;
break; break;
} }
record.size = c; record->size = c;
ret = psi->write_buf(&record); ret = record->psi->write_buf(record);
if (unlikely(ret < 0)) if (unlikely(ret < 0))
break; break;
i += c; i += c;
} }
spin_unlock_irqrestore(&psinfo->buf_lock, flags); spin_unlock_irqrestore(&psinfo->buf_lock, flags);
return unlikely(ret < 0) ? ret : size; return unlikely(ret < 0) ? ret : total_size;
} }
/* /*

View File

@ -23,7 +23,11 @@ static DEFINE_MUTEX(pmsg_lock);
static ssize_t write_pmsg(struct file *file, const char __user *buf, static ssize_t write_pmsg(struct file *file, const char __user *buf,
size_t count, loff_t *ppos) size_t count, loff_t *ppos)
{ {
u64 id; struct pstore_record record = {
.type = PSTORE_TYPE_PMSG,
.size = count,
.psi = psinfo,
};
int ret; int ret;
if (!count) if (!count)
@ -34,8 +38,7 @@ static ssize_t write_pmsg(struct file *file, const char __user *buf,
return -EFAULT; return -EFAULT;
mutex_lock(&pmsg_lock); mutex_lock(&pmsg_lock);
ret = psinfo->write_buf_user(PSTORE_TYPE_PMSG, 0, &id, 0, buf, 0, count, ret = psinfo->write_buf_user(&record, buf);
psinfo);
mutex_unlock(&pmsg_lock); mutex_unlock(&pmsg_lock);
return ret ? ret : count; return ret ? ret : count;
} }

View File

@ -451,19 +451,15 @@ static int notrace ramoops_pstore_write_buf(struct pstore_record *record)
return 0; return 0;
} }
static int notrace ramoops_pstore_write_buf_user(enum pstore_type_id type, static int notrace ramoops_pstore_write_buf_user(struct pstore_record *record,
enum kmsg_dump_reason reason, const char __user *buf)
u64 *id, unsigned int part,
const char __user *buf,
bool compressed, size_t size,
struct pstore_info *psi)
{ {
if (type == PSTORE_TYPE_PMSG) { if (record->type == PSTORE_TYPE_PMSG) {
struct ramoops_context *cxt = psi->data; struct ramoops_context *cxt = record->psi->data;
if (!cxt->mprz) if (!cxt->mprz)
return -ENOMEM; return -ENOMEM;
return persistent_ram_write_user(cxt->mprz, buf, size); return persistent_ram_write_user(cxt->mprz, buf, record->size);
} }
return -EINVAL; return -EINVAL;

View File

@ -152,18 +152,11 @@ struct pstore_record {
* *
* @write_buf_user: * @write_buf_user:
* Perform a frontend write to a backend record, using a specified * Perform a frontend write to a backend record, using a specified
* buffer that is coming directly from userspace. * buffer that is coming directly from userspace, instead of the
* @record @buf.
* *
* @type: in: pstore record type to write * @record: pointer to record metadata.
* @reason: * @buf: pointer to userspace contents to write to backend
* in: pstore write reason
* @id: out: unique identifier for the record
* @part: in: position in a multipart write
* @buf: in: pointer to userspace contents to write to backend record
* @compressed:
* in: if the record is compressed
* @size: in: size of the write
* @psi: in: pointer to the struct pstore_info for the backend
* *
* Returns 0 on success, and non-zero on error. * Returns 0 on success, and non-zero on error.
* *
@ -196,10 +189,8 @@ struct pstore_info {
ssize_t (*read)(struct pstore_record *record); ssize_t (*read)(struct pstore_record *record);
int (*write)(struct pstore_record *record); int (*write)(struct pstore_record *record);
int (*write_buf)(struct pstore_record *record); int (*write_buf)(struct pstore_record *record);
int (*write_buf_user)(enum pstore_type_id type, int (*write_buf_user)(struct pstore_record *record,
enum kmsg_dump_reason reason, u64 *id, const char __user *buf);
unsigned int part, const char __user *buf,
bool compressed, size_t size, struct pstore_info *psi);
int (*erase)(struct pstore_record *record); int (*erase)(struct pstore_record *record);
}; };