The steps argument is only read by the next function, so it can
be passed directly as an argument rather than via opal_dev.

Normally, the steps is an array on the stack, so the pointer stops
being valid then the function that set opal_dev.steps returns.
If opal_dev.steps was not set to NULL before return it would become
a dangling pointer. When the steps are passed as argument this
becomes easier to see and more difficult to misuse.

Signed-off-by: David Kozub <z...@linux.fjfi.cvut.cz>
Reviewed-by: Scott Bauer <sba...@plzdonthack.me>
---
 block/sed-opal.c | 142 ++++++++++++++++++++---------------------------
 1 file changed, 61 insertions(+), 81 deletions(-)

diff --git a/block/sed-opal.c b/block/sed-opal.c
index 3493bb979978..21f2789a20cb 100644
--- a/block/sed-opal.c
+++ b/block/sed-opal.c
@@ -88,7 +88,6 @@ struct opal_dev {
        void *data;
        sec_send_recv *send_recv;
 
-       const struct opal_step *steps;
        struct mutex dev_lock;
        u16 comid;
        u32 hsn;
@@ -387,19 +386,19 @@ static void check_geometry(struct opal_dev *dev, const 
void *data)
        dev->lowest_lba = geo->lowest_aligned_lba;
 }
 
-static int next(struct opal_dev *dev)
+static int next(struct opal_dev *dev, const struct opal_step *steps,
+               size_t n_steps)
 {
        const struct opal_step *step;
-       int state = 0, error = 0;
+       size_t state;
+       int error = 0;
 
-       do {
-               step = &dev->steps[state];
-               if (!step->fn)
-                       break;
+       for (state = 0; !error && state < n_steps; state++) {
+               step = &steps[state];
 
                error = step->fn(dev, step->data);
                if (error) {
-                       pr_debug("Step %d (%pS) failed wit error %d: %s\n",
+                       pr_debug("Step %zu (%pS) failed with error %d: %s\n",
                                 state, step->fn, error,
                                 opal_error_to_human(error));
 
@@ -416,8 +415,7 @@ static int next(struct opal_dev *dev)
                        }
 
                }
-               state++;
-       } while (!error);
+       }
 
        return error;
 }
@@ -1941,17 +1939,13 @@ static int end_opal_session(struct opal_dev *dev, void 
*data)
 static int end_opal_session_error(struct opal_dev *dev)
 {
        const struct opal_step error_end_session[] = {
-               { end_opal_session, },
-               { NULL, }
+               { end_opal_session, }
        };
-       dev->steps = error_end_session;
-       return next(dev);
+       return next(dev, error_end_session, ARRAY_SIZE(error_end_session));
 }
 
-static inline void setup_opal_dev(struct opal_dev *dev,
-                                 const struct opal_step *steps)
+static inline void setup_opal_dev(struct opal_dev *dev)
 {
-       dev->steps = steps;
        dev->tsn = 0;
        dev->hsn = 0;
        dev->prev_data = NULL;
@@ -1960,14 +1954,13 @@ static inline void setup_opal_dev(struct opal_dev *dev,
 static int check_opal_support(struct opal_dev *dev)
 {
        const struct opal_step steps[] = {
-               { opal_discovery0, },
-               { NULL, }
+               { opal_discovery0, }
        };
        int ret;
 
        mutex_lock(&dev->dev_lock);
-       setup_opal_dev(dev, steps);
-       ret = next(dev);
+       setup_opal_dev(dev);
+       ret = next(dev, steps, ARRAY_SIZE(steps));
        dev->supported = !ret;
        mutex_unlock(&dev->dev_lock);
        return ret;
@@ -2024,14 +2017,13 @@ static int opal_secure_erase_locking_range(struct 
opal_dev *dev,
                { start_auth_opal_session, opal_session },
                { get_active_key, &opal_session->opal_key.lr },
                { gen_key, },
-               { end_opal_session, },
-               { NULL, }
+               { end_opal_session, }
        };
        int ret;
 
        mutex_lock(&dev->dev_lock);
-       setup_opal_dev(dev, erase_steps);
-       ret = next(dev);
+       setup_opal_dev(dev);
+       ret = next(dev, erase_steps, ARRAY_SIZE(erase_steps));
        mutex_unlock(&dev->dev_lock);
        return ret;
 }
@@ -2043,14 +2035,13 @@ static int opal_erase_locking_range(struct opal_dev 
*dev,
                { opal_discovery0, },
                { start_auth_opal_session, opal_session },
                { erase_locking_range, opal_session },
-               { end_opal_session, },
-               { NULL, }
+               { end_opal_session, }
        };
        int ret;
 
        mutex_lock(&dev->dev_lock);
-       setup_opal_dev(dev, erase_steps);
-       ret = next(dev);
+       setup_opal_dev(dev);
+       ret = next(dev, erase_steps, ARRAY_SIZE(erase_steps));
        mutex_unlock(&dev->dev_lock);
        return ret;
 }
@@ -2067,8 +2058,7 @@ static int opal_enable_disable_shadow_mbr(struct opal_dev 
*dev,
                { end_opal_session, },
                { start_admin1LSP_opal_session, &opal_mbr->key },
                { set_mbr_enable_disable, &token },
-               { end_opal_session, },
-               { NULL, }
+               { end_opal_session, }
        };
        int ret;
 
@@ -2077,8 +2067,8 @@ static int opal_enable_disable_shadow_mbr(struct opal_dev 
*dev,
                return -EINVAL;
 
        mutex_lock(&dev->dev_lock);
-       setup_opal_dev(dev, mbr_steps);
-       ret = next(dev);
+       setup_opal_dev(dev);
+       ret = next(dev, mbr_steps, ARRAY_SIZE(mbr_steps));
        mutex_unlock(&dev->dev_lock);
        return ret;
 }
@@ -2091,8 +2081,7 @@ static int opal_mbr_status(struct opal_dev *dev, struct 
opal_mbr_data *opal_mbr)
                { opal_discovery0, },
                { start_admin1LSP_opal_session, &opal_mbr->key },
                { set_mbr_done, &token },
-               { end_opal_session, },
-               { NULL, }
+               { end_opal_session, }
        };
        int ret;
 
@@ -2101,8 +2090,8 @@ static int opal_mbr_status(struct opal_dev *dev, struct 
opal_mbr_data *opal_mbr)
                return -EINVAL;
 
        mutex_lock(&dev->dev_lock);
-       setup_opal_dev(dev, mbr_steps);
-       ret = next(dev);
+       setup_opal_dev(dev);
+       ret = next(dev, mbr_steps, ARRAY_SIZE(mbr_steps));
        mutex_unlock(&dev->dev_lock);
        return ret;
 }
@@ -2114,8 +2103,7 @@ static int opal_write_shadow_mbr(struct opal_dev *dev,
                { opal_discovery0, },
                { start_admin1LSP_opal_session, &info->key },
                { write_shadow_mbr, info },
-               { end_opal_session, },
-               { NULL, }
+               { end_opal_session, }
        };
        int ret;
 
@@ -2126,8 +2114,8 @@ static int opal_write_shadow_mbr(struct opal_dev *dev,
                return -EINVAL;
 
        mutex_lock(&dev->dev_lock);
-       setup_opal_dev(dev, mbr_steps);
-       ret = next(dev);
+       setup_opal_dev(dev);
+       ret = next(dev, mbr_steps, ARRAY_SIZE(mbr_steps));
        mutex_unlock(&dev->dev_lock);
        return ret;
 }
@@ -2144,7 +2132,7 @@ static int opal_save(struct opal_dev *dev, struct 
opal_lock_unlock *lk_unlk)
        suspend->lr = lk_unlk->session.opal_key.lr;
 
        mutex_lock(&dev->dev_lock);
-       setup_opal_dev(dev, NULL);
+       setup_opal_dev(dev);
        add_suspend_info(dev, suspend);
        mutex_unlock(&dev->dev_lock);
        return 0;
@@ -2157,8 +2145,7 @@ static int opal_add_user_to_lr(struct opal_dev *dev,
                { opal_discovery0, },
                { start_admin1LSP_opal_session, &lk_unlk->session.opal_key },
                { add_user_to_lr, lk_unlk },
-               { end_opal_session, },
-               { NULL, }
+               { end_opal_session, }
        };
        int ret;
 
@@ -2180,8 +2167,8 @@ static int opal_add_user_to_lr(struct opal_dev *dev,
        }
 
        mutex_lock(&dev->dev_lock);
-       setup_opal_dev(dev, steps);
-       ret = next(dev);
+       setup_opal_dev(dev);
+       ret = next(dev, steps, ARRAY_SIZE(steps));
        mutex_unlock(&dev->dev_lock);
        return ret;
 }
@@ -2191,14 +2178,13 @@ static int opal_reverttper(struct opal_dev *dev, struct 
opal_key *opal)
        const struct opal_step revert_steps[] = {
                { opal_discovery0, },
                { start_SIDASP_opal_session, opal },
-               { revert_tper, }, /* controller will terminate session */
-               { NULL, }
+               { revert_tper, } /* controller will terminate session */
        };
        int ret;
 
        mutex_lock(&dev->dev_lock);
-       setup_opal_dev(dev, revert_steps);
-       ret = next(dev);
+       setup_opal_dev(dev);
+       ret = next(dev, revert_steps, ARRAY_SIZE(revert_steps));
        mutex_unlock(&dev->dev_lock);
 
        /*
@@ -2218,19 +2204,20 @@ static int __opal_lock_unlock(struct opal_dev *dev,
                { opal_discovery0, },
                { start_auth_opal_session, &lk_unlk->session },
                { lock_unlock_locking_range, lk_unlk },
-               { end_opal_session, },
-               { NULL, }
+               { end_opal_session, }
        };
        const struct opal_step unlock_sum_steps[] = {
                { opal_discovery0, },
                { start_auth_opal_session, &lk_unlk->session },
                { lock_unlock_locking_range_sum, lk_unlk },
-               { end_opal_session, },
-               { NULL, }
+               { end_opal_session, }
        };
 
-       dev->steps = lk_unlk->session.sum ? unlock_sum_steps : unlock_steps;
-       return next(dev);
+       if (lk_unlk->session.sum)
+               return next(dev, unlock_sum_steps,
+                           ARRAY_SIZE(unlock_sum_steps));
+       else
+               return next(dev, unlock_steps, ARRAY_SIZE(unlock_steps));
 }
 
 static int __opal_set_mbr_done(struct opal_dev *dev, struct opal_key *key)
@@ -2240,12 +2227,10 @@ static int __opal_set_mbr_done(struct opal_dev *dev, 
struct opal_key *key)
                { opal_discovery0, },
                { start_admin1LSP_opal_session, key },
                { set_mbr_done, &mbr_done_tf },
-               { end_opal_session, },
-               { NULL, }
+               { end_opal_session, }
        };
 
-       dev->steps = mbrdone_step;
-       return next(dev);
+       return next(dev, mbrdone_step, ARRAY_SIZE(mbrdone_step));
 }
 
 static int opal_lock_unlock(struct opal_dev *dev,
@@ -2272,8 +2257,7 @@ static int opal_take_ownership(struct opal_dev *dev, 
struct opal_key *opal)
                { end_opal_session, },
                { start_SIDASP_opal_session, opal },
                { set_sid_cpin_pin, opal },
-               { end_opal_session, },
-               { NULL, }
+               { end_opal_session, }
        };
        int ret;
 
@@ -2281,8 +2265,8 @@ static int opal_take_ownership(struct opal_dev *dev, 
struct opal_key *opal)
                return -ENODEV;
 
        mutex_lock(&dev->dev_lock);
-       setup_opal_dev(dev, owner_steps);
-       ret = next(dev);
+       setup_opal_dev(dev);
+       ret = next(dev, owner_steps, ARRAY_SIZE(owner_steps));
        mutex_unlock(&dev->dev_lock);
        return ret;
 }
@@ -2295,8 +2279,7 @@ static int opal_activate_lsp(struct opal_dev *dev,
                { start_SIDASP_opal_session, &opal_lr_act->key },
                { get_lsp_lifecycle, },
                { activate_lsp, opal_lr_act },
-               { end_opal_session, },
-               { NULL, }
+               { end_opal_session, }
        };
        int ret;
 
@@ -2304,8 +2287,8 @@ static int opal_activate_lsp(struct opal_dev *dev,
                return -EINVAL;
 
        mutex_lock(&dev->dev_lock);
-       setup_opal_dev(dev, active_steps);
-       ret = next(dev);
+       setup_opal_dev(dev);
+       ret = next(dev, active_steps, ARRAY_SIZE(active_steps));
        mutex_unlock(&dev->dev_lock);
        return ret;
 }
@@ -2317,14 +2300,13 @@ static int opal_setup_locking_range(struct opal_dev 
*dev,
                { opal_discovery0, },
                { start_auth_opal_session, &opal_lrs->session },
                { setup_locking_range, opal_lrs },
-               { end_opal_session, },
-               { NULL, }
+               { end_opal_session, }
        };
        int ret;
 
        mutex_lock(&dev->dev_lock);
-       setup_opal_dev(dev, lr_steps);
-       ret = next(dev);
+       setup_opal_dev(dev);
+       ret = next(dev, lr_steps, ARRAY_SIZE(lr_steps));
        mutex_unlock(&dev->dev_lock);
        return ret;
 }
@@ -2335,8 +2317,7 @@ static int opal_set_new_pw(struct opal_dev *dev, struct 
opal_new_pw *opal_pw)
                { opal_discovery0, },
                { start_auth_opal_session, &opal_pw->session },
                { set_new_pw, &opal_pw->new_user_pw },
-               { end_opal_session, },
-               { NULL }
+               { end_opal_session, }
        };
        int ret;
 
@@ -2347,8 +2328,8 @@ static int opal_set_new_pw(struct opal_dev *dev, struct 
opal_new_pw *opal_pw)
                return -EINVAL;
 
        mutex_lock(&dev->dev_lock);
-       setup_opal_dev(dev, pw_steps);
-       ret = next(dev);
+       setup_opal_dev(dev);
+       ret = next(dev, pw_steps, ARRAY_SIZE(pw_steps));
        mutex_unlock(&dev->dev_lock);
        return ret;
 }
@@ -2360,8 +2341,7 @@ static int opal_activate_user(struct opal_dev *dev,
                { opal_discovery0, },
                { start_admin1LSP_opal_session, &opal_session->opal_key },
                { internal_activate_user, opal_session },
-               { end_opal_session, },
-               { NULL, }
+               { end_opal_session, }
        };
        int ret;
 
@@ -2373,8 +2353,8 @@ static int opal_activate_user(struct opal_dev *dev,
        }
 
        mutex_lock(&dev->dev_lock);
-       setup_opal_dev(dev, act_steps);
-       ret = next(dev);
+       setup_opal_dev(dev);
+       ret = next(dev, act_steps, ARRAY_SIZE(act_steps));
        mutex_unlock(&dev->dev_lock);
        return ret;
 }
@@ -2391,7 +2371,7 @@ bool opal_unlock_from_suspend(struct opal_dev *dev)
                return false;
 
        mutex_lock(&dev->dev_lock);
-       setup_opal_dev(dev, NULL);
+       setup_opal_dev(dev);
 
        list_for_each_entry(suspend, &dev->unlk_lst, node) {
                dev->tsn = 0;
-- 
2.20.1

Reply via email to