Re: [PATCH v2 04/11] pstore/blk: blkoops: support console recorder

[WeiXiong Liao <liaoweixiong@allwinnertech.com>]

hi Kees Cook,

On 2020/3/19 AM 2:16, Kees Cook wrote:
> On Fri, Feb 07, 2020 at 08:25:48PM +0800, WeiXiong Liao wrote:
>> Support recorder for console. To enable console recorder, just make
>> console_size be greater than 0 and a multiple of 4096.
>>
>> Signed-off-by: WeiXiong Liao <liaoweixiong@allwinnertech.com>
>> ---
>>  fs/pstore/Kconfig          |  12 ++++++
>>  fs/pstore/blkoops.c        |  11 +++++
>>  fs/pstore/blkzone.c        | 101 ++++++++++++++++++++++++++++++++++-----------
>>  include/linux/blkoops.h    |   6 ++-
>>  include/linux/pstore_blk.h |   8 +++-
>>  5 files changed, 112 insertions(+), 26 deletions(-)
>>
>> diff --git a/fs/pstore/Kconfig b/fs/pstore/Kconfig
>> index bbf1fdb5eaa7..5f0a42823028 100644
>> --- a/fs/pstore/Kconfig
>> +++ b/fs/pstore/Kconfig
>> @@ -198,6 +198,18 @@ config PSTORE_BLKOOPS_PMSG_SIZE
>>  	  NOTE that, both kconfig and module parameters can configure blkoops,
>>  	  but module parameters have priority over kconfig.
>>  
>> +config PSTORE_BLKOOPS_CONSOLE_SIZE
>> +	int "console size in kbytes for blkoops"
>> +	depends on PSTORE_BLKOOPS
>> +	depends on PSTORE_CONSOLE
>> +	default 64
> 
> Same tricks here as for the PMSG.
> 

Same reply as for the PMSG.

>> +	help
>> +	  This just sets size of console (console_size) for pstore/blk. The
>> +	  size is in KB and must be a multiple of 4.
>> +
>> +	  NOTE that, both kconfig and module parameters can configure blkoops,
>> +	  but module parameters have priority over kconfig.
>> +
>>  config PSTORE_BLKOOPS_BLKDEV
>>  	string "block device for blkoops"
>>  	depends on PSTORE_BLKOOPS
>> diff --git a/fs/pstore/blkoops.c b/fs/pstore/blkoops.c
>> index 02e6e4c1f965..05990bc3b168 100644
>> --- a/fs/pstore/blkoops.c
>> +++ b/fs/pstore/blkoops.c
>> @@ -20,6 +20,10 @@
>>  module_param(pmsg_size, long, 0400);
>>  MODULE_PARM_DESC(pmsg_size, "pmsg size in kbytes");
>>  
>> +static long console_size = -1;
>> +module_param(console_size, long, 0400);
>> +MODULE_PARM_DESC(console_size, "console size in kbytes");
>> +
>>  static int dump_oops = -1;
>>  module_param(dump_oops, int, 0400);
>>  MODULE_PARM_DESC(total_size, "whether dump oops");
>> @@ -70,6 +74,12 @@
>>  #define DEFAULT_PMSG_SIZE 0
>>  #endif
>>  
>> +#ifdef CONFIG_PSTORE_BLKOOPS_CONSOLE_SIZE
>> +#define DEFAULT_CONSOLE_SIZE CONFIG_PSTORE_BLKOOPS_CONSOLE_SIZE
>> +#else
>> +#define DEFAULT_CONSOLE_SIZE 0
>> +#endif
>> +
>>  #ifdef CONFIG_PSTORE_BLKOOPS_DUMP_OOPS
>>  #define DEFAULT_DUMP_OOPS CONFIG_PSTORE_BLKOOPS_DUMP_OOPS
>>  #else
>> @@ -124,6 +134,7 @@ int blkoops_register_device(struct blkoops_device *bo_dev)
>>  
>>  	verify_size(dmesg_size, DEFAULT_DMESG_SIZE, 4096);
>>  	verify_size(pmsg_size, DEFAULT_PMSG_SIZE, 4096);
>> +	verify_size(console_size, DEFAULT_CONSOLE_SIZE, 4096);
>>  #undef verify_size
>>  	dump_oops = !!(dump_oops < 0 ? DEFAULT_DUMP_OOPS : dump_oops);
>>  
>> diff --git a/fs/pstore/blkzone.c b/fs/pstore/blkzone.c
>> index a3464252d52e..9a7e9b06ccf7 100644
>> --- a/fs/pstore/blkzone.c
>> +++ b/fs/pstore/blkzone.c
>> @@ -88,9 +88,11 @@ struct blkz_zone {
>>  struct blkz_context {
>>  	struct blkz_zone **dbzs;	/* dmesg block zones */
>>  	struct blkz_zone *pbz;		/* Pmsg block zone */
>> +	struct blkz_zone *cbz;		/* console block zone */
>>  	unsigned int dmesg_max_cnt;
>>  	unsigned int dmesg_read_cnt;
>>  	unsigned int pmsg_read_cnt;
>> +	unsigned int console_read_cnt;
>>  	unsigned int dmesg_write_cnt;
>>  	/*
>>  	 * the counter should be recovered when recover.
>> @@ -111,6 +113,9 @@ struct blkz_context {
>>  };
>>  static struct blkz_context blkz_cxt;
>>  
>> +static void blkz_flush_all_dirty_zones(struct work_struct *);
>> +static DECLARE_WORK(blkz_cleaner, blkz_flush_all_dirty_zones);
>> +
>>  enum blkz_flush_mode {
>>  	FLUSH_NONE = 0,
>>  	FLUSH_PART,
>> @@ -200,6 +205,9 @@ static int blkz_zone_write(struct blkz_zone *zone,
>>  	return 0;
>>  set_dirty:
>>  	atomic_set(&zone->dirty, true);
>> +	/* flush dirty zones nicely */
>> +	if (wcnt == -EBUSY && !is_on_panic())
>> +		schedule_work(&blkz_cleaner);
>>  	return -EBUSY;
>>  }
>>  
>> @@ -266,6 +274,15 @@ static int blkz_move_zone(struct blkz_zone *old, struct blkz_zone *new)
>>  	return 0;
>>  }
>>  
>> +static void blkz_flush_all_dirty_zones(struct work_struct *work)
>> +{
>> +	struct blkz_context *cxt = &blkz_cxt;
>> +
>> +	blkz_flush_dirty_zone(cxt->pbz);
>> +	blkz_flush_dirty_zone(cxt->cbz);
>> +	blkz_flush_dirty_zones(cxt->dbzs, cxt->dmesg_max_cnt);
>> +}
>> +
>>  static int blkz_recover_dmesg_data(struct blkz_context *cxt)
>>  {
>>  	struct blkz_info *info = cxt->bzinfo;
>> @@ -419,15 +436,13 @@ static int blkz_recover_dmesg(struct blkz_context *cxt)
>>  	return ret;
>>  }
>>  
>> -static int blkz_recover_pmsg(struct blkz_context *cxt)
>> +static int blkz_recover_zone(struct blkz_context *cxt, struct blkz_zone *zone)
>>  {
>>  	struct blkz_info *info = cxt->bzinfo;
>>  	struct blkz_buffer *oldbuf;
>> -	struct blkz_zone *zone = NULL;
>>  	int ret = 0;
>>  	ssize_t rcnt, len;
>>  
>> -	zone = cxt->pbz;
>>  	if (!zone || zone->oldbuf)
>>  		return 0;
>>  
>> @@ -493,7 +508,11 @@ static inline int blkz_recovery(struct blkz_context *cxt)
>>  	if (ret)
>>  		goto recover_fail;
>>  
>> -	ret = blkz_recover_pmsg(cxt);
>> +	ret = blkz_recover_zone(cxt, cxt->pbz);
>> +	if (ret)
>> +		goto recover_fail;
>> +
>> +	ret = blkz_recover_zone(cxt, cxt->cbz);
>>  	if (ret)
>>  		goto recover_fail;
>>  
>> @@ -512,6 +531,7 @@ static int blkz_pstore_open(struct pstore_info *psi)
>>  
>>  	cxt->dmesg_read_cnt = 0;
>>  	cxt->pmsg_read_cnt = 0;
>> +	cxt->console_read_cnt = 0;
>>  	return 0;
>>  }
>>  
>> @@ -539,7 +559,7 @@ static inline int blkz_dmesg_erase(struct blkz_context *cxt,
>>  	return blkz_zone_write(zone, FLUSH_META, NULL, 0, 0);
>>  }
>>  
>> -static inline int blkz_pmsg_erase(struct blkz_context *cxt,
>> +static inline int blkz_record_erase(struct blkz_context *cxt,
>>  		struct blkz_zone *zone)
>>  {
>>  	if (unlikely(!blkz_old_ok(zone)))
>> @@ -566,9 +586,10 @@ static int blkz_pstore_erase(struct pstore_record *record)
>>  	case PSTORE_TYPE_DMESG:
>>  		return blkz_dmesg_erase(cxt, cxt->dbzs[record->id]);
>>  	case PSTORE_TYPE_PMSG:
>> -		return blkz_pmsg_erase(cxt, cxt->pbz);
>> -	default:
>> -		return -EINVAL;
>> +		return blkz_record_erase(cxt, cxt->pbz);
>> +	case PSTORE_TYPE_CONSOLE:
>> +		return blkz_record_erase(cxt, cxt->cbz);
>> +	default: return -EINVAL;
>>  	}
>>  }
>>  
>> @@ -653,17 +674,15 @@ static int notrace blkz_dmesg_write(struct blkz_context *cxt,
>>  	return 0;
>>  }
>>  
>> -static int notrace blkz_pmsg_write(struct blkz_context *cxt,
>> -		struct pstore_record *record)
>> +static int notrace blkz_record_write(struct blkz_context *cxt,
>> +		struct blkz_zone *zone, struct pstore_record *record)
> 
> How about generalizing this earlier in the patch series instead of
> mutating it here?
> 

OK.

>>  {
>> -	struct blkz_zone *zone;
>>  	size_t start, rem;
>>  	int cnt = record->size;
>>  	bool is_full_data = false;
>>  	char *buf = record->buf;
>>  
>> -	zone = cxt->pbz;
>> -	if (!zone)
>> +	if (!zone || !record)
>>  		return -ENOSPC;
>>  
>>  	if (atomic_read(&zone->buffer->datalen) >= zone->buffer_size)
>> @@ -710,11 +729,20 @@ static int notrace blkz_pstore_write(struct pstore_record *record)
>>  			record->reason == KMSG_DUMP_PANIC)
>>  		atomic_set(&cxt->on_panic, 1);
>>  
>> +	/*
>> +	 * if on panic, do not write except dmesg records
>> +	 * Fix case that panic_write prints log which wakes up console recorder.
>> +	 */
>> +	if (is_on_panic() && record->type != PSTORE_TYPE_DMESG)
>> +		return -EBUSY;
>> +
>>  	switch (record->type) {
>>  	case PSTORE_TYPE_DMESG:
>>  		return blkz_dmesg_write(cxt, record);
>> +	case PSTORE_TYPE_CONSOLE:
>> +		return blkz_record_write(cxt, cxt->cbz, record);
>>  	case PSTORE_TYPE_PMSG:
>> -		return blkz_pmsg_write(cxt, record);
>> +		return blkz_record_write(cxt, cxt->pbz, record);
>>  	default:
>>  		return -EINVAL;
>>  	}
>> @@ -738,6 +766,13 @@ static struct blkz_zone *blkz_read_next_zone(struct blkz_context *cxt)
>>  			return zone;
>>  	}
>>  
>> +	if (cxt->console_read_cnt == 0) {
>> +		cxt->console_read_cnt++;
>> +		zone = cxt->cbz;
>> +		if (blkz_old_ok(zone))
>> +			return zone;
>> +	}
>> +
>>  	return NULL;
>>  }
>>  
>> @@ -799,7 +834,7 @@ static ssize_t blkz_dmesg_read(struct blkz_zone *zone,
>>  	return size + hlen;
>>  }
>>  
>> -static ssize_t blkz_pmsg_read(struct blkz_zone *zone,
>> +static ssize_t blkz_record_read(struct blkz_zone *zone,
>>  		struct pstore_record *record)
>>  {
>>  	size_t size, start;
>> @@ -825,7 +860,7 @@ static ssize_t blkz_pmsg_read(struct blkz_zone *zone,
>>  static ssize_t blkz_pstore_read(struct pstore_record *record)
>>  {
>>  	struct blkz_context *cxt = record->psi->data;
>> -	ssize_t (*blkz_read)(struct blkz_zone *zone,
>> +	ssize_t (*readop)(struct blkz_zone *zone,
>>  			struct pstore_record *record);
>>  	struct blkz_zone *zone;
>>  	ssize_t ret;
>> @@ -843,17 +878,19 @@ static ssize_t blkz_pstore_read(struct pstore_record *record)
>>  	record->type = zone->type;
>>  	switch (record->type) {
>>  	case PSTORE_TYPE_DMESG:
>> -		blkz_read = blkz_dmesg_read;
>> +		readop = blkz_dmesg_read;
>>  		record->id = cxt->dmesg_read_cnt - 1;
>>  		break;
>> +	case PSTORE_TYPE_CONSOLE:
>> +		/* fallthrough */
> 
> Since this case has no body, you can leave off the "fallthrough". (But
> if you want to mark it anyway, please use "fallthrough;" instead of a
> comment.)
> 

OK. I will fix it anywhere.

>>  	case PSTORE_TYPE_PMSG:
>> -		blkz_read = blkz_pmsg_read;
>> +		readop = blkz_record_read;
>>  		break;
>>  	default:
>>  		goto next_zone;
>>  	}
>>  
>> -	ret = blkz_read(zone, record);
>> +	ret = readop(zone, record);
>>  	if (ret == READ_NEXT_ZONE)
>>  		goto next_zone;
>>  	return ret;
>> @@ -1001,15 +1038,25 @@ static int blkz_cut_zones(struct blkz_context *cxt)
>>  		goto fail_out;
>>  	}
>>  
>> +	off_size += info->console_size;
>> +	cxt->cbz = blkz_init_zone(PSTORE_TYPE_CONSOLE, &off,
>> +			info->console_size);
>> +	if (IS_ERR(cxt->cbz)) {
>> +		err = PTR_ERR(cxt->cbz);
>> +		goto free_pmsg;
>> +	}
>> +
>>  	cxt->dbzs = blkz_init_zones(PSTORE_TYPE_DMESG, &off,
>>  			info->total_size - off_size,
>>  			info->dmesg_size, &cxt->dmesg_max_cnt);
>>  	if (IS_ERR(cxt->dbzs)) {
>>  		err = PTR_ERR(cxt->dbzs);
>> -		goto free_pmsg;
>> +		goto free_console;
>>  	}
>>  
>>  	return 0;
>> +free_console:
>> +	blkz_free_zone(&cxt->cbz);
>>  free_pmsg:
>>  	blkz_free_zone(&cxt->pbz);
>>  fail_out:
>> @@ -1027,7 +1074,7 @@ int blkz_register(struct blkz_info *info)
>>  		return -EINVAL;
>>  	}
>>  
>> -	if (!info->dmesg_size && !info->pmsg_size) {
>> +	if (!info->dmesg_size && !info->pmsg_size && !info->console_size) {
>>  		pr_warn("at least one of the records be non-zero\n");
>>  		return -EINVAL;
>>  	}
>> @@ -1055,6 +1102,7 @@ int blkz_register(struct blkz_info *info)
>>  	check_size(total_size, 4096);
>>  	check_size(dmesg_size, SECTOR_SIZE);
>>  	check_size(pmsg_size, SECTOR_SIZE);
>> +	check_size(console_size, SECTOR_SIZE);
>>  
>>  #undef check_size
>>  
>> @@ -1087,6 +1135,7 @@ int blkz_register(struct blkz_info *info)
>>  	pr_debug("\ttotal size : %ld Bytes\n", info->total_size);
>>  	pr_debug("\tdmesg size : %ld Bytes\n", info->dmesg_size);
>>  	pr_debug("\tpmsg size : %ld Bytes\n", info->pmsg_size);
>> +	pr_debug("\tconsole size : %ld Bytes\n", info->console_size);
>>  
>>  	err = blkz_cut_zones(cxt);
>>  	if (err) {
>> @@ -1108,11 +1157,15 @@ int blkz_register(struct blkz_info *info)
>>  		cxt->pstore.flags |= PSTORE_FLAGS_DMESG;
>>  	if (info->pmsg_size)
>>  		cxt->pstore.flags |= PSTORE_FLAGS_PMSG;
>> +	if (info->console_size)
>> +		cxt->pstore.flags |= PSTORE_FLAGS_CONSOLE;
>>  
>> -	pr_info("Registered %s as blkzone backend for %s%s%s\n", info->name,
>> +	pr_info("Registered %s as blkzone backend for %s%s%s%s\n",
>> +			info->name,
>>  			cxt->dbzs && cxt->bzinfo->dump_oops ? "Oops " : "",
>>  			cxt->dbzs && cxt->bzinfo->panic_write ? "Panic " : "",
>> -			cxt->pbz ? "Pmsg" : "");
>> +			cxt->pbz ? "Pmsg " : "",
>> +			cxt->cbz ? "Console" : "");
>>  
>>  	err = pstore_register(&cxt->pstore);
>>  	if (err) {
>> @@ -1139,6 +1192,8 @@ void blkz_unregister(struct blkz_info *info)
>>  {
>>  	struct blkz_context *cxt = &blkz_cxt;
>>  
>> +	flush_work(&blkz_cleaner);
>> +
>>  	pstore_unregister(&cxt->pstore);
>>  	kfree(cxt->pstore.buf);
>>  	cxt->pstore.bufsize = 0;
>> diff --git a/include/linux/blkoops.h b/include/linux/blkoops.h
>> index fe63739309aa..8f40f225545d 100644
>> --- a/include/linux/blkoops.h
>> +++ b/include/linux/blkoops.h
>> @@ -23,8 +23,10 @@
>>   *	Both of the @size and @offset parameters on this interface are
>>   *	the relative size of the space provided, not the whole disk/flash.
>>   *
>> - *	On success, the number of bytes read should be returned.
>> - *	On error, negative number should be returned.
>> + *	On success, the number of bytes read/write should be returned.
>> + *	On error, negative number should be returned. The following returning
>> + *	number means more:
>> + *	  -EBUSY: pstore/blk should try again later.
>>   * @panic_write:
>>   *	The write operation only used for panic.
>>   *
>> diff --git a/include/linux/pstore_blk.h b/include/linux/pstore_blk.h
>> index af06be25bd01..546375e04419 100644
>> --- a/include/linux/pstore_blk.h
>> +++ b/include/linux/pstore_blk.h
>> @@ -22,6 +22,9 @@
>>   * @pmsg_size:
>>   *	The size of zone for pmsg. Zero means disabled, othewise, it must be
>>   *	multiple of SECTOR_SIZE(512).
>> + * @console_size:
>> + *	The size of zone for console. Zero means disabled, othewise, it must
>> + *	be multiple of SECTOR_SIZE(512).
>>   * @dump_oops:
>>   *	Dump oops and panic log or only panic.
>>   * @read, @write:
>> @@ -33,7 +36,9 @@
>>   *	the relative size of the space provided, not the whole disk/flash.
>>   *
>>   *	On success, the number of bytes read/write should be returned.
>> - *	On error, negative number should be returned.
>> + *	On error, negative number should be returned. The following returning
>> + *	number means more:
>> + *	  -EBUSY: pstore/blk should try again later.
>>   * @panic_write:
>>   *	The write operation only used for panic. It's optional if you do not
>>   *	care panic record. If panic occur but blkzone do not recover yet, the
>> @@ -54,6 +59,7 @@ struct blkz_info {
>>  	unsigned long total_size;
>>  	unsigned long dmesg_size;
>>  	unsigned long pmsg_size;
>> +	unsigned long console_size;
>>  	int dump_oops;
>>  	blkz_read_op read;
>>  	blkz_write_op write;
>> -- 
>> 1.9.1
>>
> 

-- 
WeiXiong Liao

______________________________________________________
Linux MTD discussion mailing list
http://lists.infradead.org/mailman/listinfo/linux-mtd/