Change in flashrom[master]: cli_classic.c: implement set_wp_region operation

[Edward O'Callaghan (Code Review)]

Edward O'Callaghan has submitted this change. ( https://review.coreboot.org/c/flashrom/+/52531 )

Change subject: cli_classic.c: implement set_wp_region operation
......................................................................

cli_classic.c: implement set_wp_region operation

set_wp_region allows to set the wp_range based on a layout region.

Signed-off-by: Daniel Campello 
Change-Id: Ibad68a038ab38b9986b0d8b5f5eb6c73b20ef381
Reviewed-on: https://review.coreboot.org/c/flashrom/+/52531
Reviewed-by: Edward O'Callaghan 
Tested-by: build bot (Jenkins) 
---
M cli_classic.c
M layout.c
M layout.h
3 files changed, 31 insertions(+), 0 deletions(-)

Approvals:
  build bot (Jenkins): Verified
  Edward O'Callaghan: Looks good to me, approved

diff --git a/cli_classic.c b/cli_classic.c
index 58e81cd..5fd8092 100644
--- a/cli_classic.c
+++ b/cli_classic.c
@@ -65,6 +65,7 @@
           "      --wp-list                     list write protect range\n"
           "      --wp-status                   show write protect status\n"
           "      --wp-range=<start>,<len>      set write protect range\n"
+	       "      --wp-region <region>          set write protect region\n"
           "      --flash-name                  read out the detected flash name\n"
           "      --flash-size                  read out the detected flash size\n"
           "      --fmap                        read ROM layout from fmap embedded in ROM\n"
@@ -236,6 +237,7 @@
    char *pparam = NULL;
    struct layout_include_args *include_args = NULL;
    char *wp_mode_opt = NULL;
+	char *wp_region = NULL;

/*
     * Safety-guard against a user who has (mistakenly) closed
@@ -462,6 +464,10 @@
    		}
 #endif /* STANDALONE */
    		break;
+		case OPTION_WP_SET_REGION:
+			set_wp_region = 1;
+			wp_region = strdup(optarg);
+			break;
    	default:
    		cli_classic_abort_usage(NULL);
    		break;
@@ -772,6 +778,15 @@
    	ret |= wp->set_range(fill_flash, wp_start, wp_len);
    }

+	if (set_wp_region && wp_region) {
+		if (get_region_range(layout, wp_region, &wp_start, &wp_len)) {
+			ret = 1;
+			goto out_release;
+		}
+		ret |= wp->set_range(fill_flash, wp_start, wp_len);
+		free(wp_region);
+	}
+
    if (!ret && set_wp_enable) {
    	enum wp_mode wp_mode;

diff --git a/layout.c b/layout.c
index 7fd203d..4d1dd56 100644
--- a/layout.c
+++ b/layout.c
@@ -184,6 +184,20 @@
    return 0;
 }

+int get_region_range(struct flashrom_layout *const l, const char *name,
+		     unsigned int *start, unsigned int *len)
+{
+	size_t i;
+	for (i = 0; i < l->num_entries; ++i) {
+		if (!strcmp(l->entries[i].name, name)) {
+			*start = l->entries[i].start;
+			*len = l->entries[i].end - l->entries[i].start + 1;
+			return 0;
+		}
+	}
+	return 1;
+}
+
 /* process -i arguments
  * returns 0 to indicate success, >0 to indicate failure
  */
diff --git a/layout.h b/layout.h
index 327f5bf..f574097 100644
--- a/layout.h
+++ b/layout.h
@@ -65,6 +65,8 @@
 struct flashrom_flashctx;
 const struct flashrom_layout *get_layout(const struct flashrom_flashctx *const flashctx);

+int get_region_range(struct flashrom_layout *const l, const char *name,
+		     unsigned int *start, unsigned int *len);
 int process_include_args(struct flashrom_layout *l, const struct layout_include_args *const args);
 const struct romentry *layout_next_included_region(const struct flashrom_layout *, chipoff_t);
 const struct romentry *layout_next_included(const struct flashrom_layout *, const struct romentry *);