#include <linux/nl80211.h>
 #include <net/cfg80211.h>
 #include "ieee80211_i.h"
+#include "driver-ops.h"
 
 static enum ieee80211_chan_mode
 __ieee80211_get_channel_mode(struct ieee80211_local *local,
                      enum ieee80211_chanctx_mode mode)
 {
        struct ieee80211_chanctx *ctx;
+       int err;
 
        lockdep_assert_held(&local->chanctx_mtx);
 
        ctx->conf.channel_type = channel_type;
        ctx->mode = mode;
 
+       err = drv_add_chanctx(local, ctx);
+       if (err) {
+               kfree(ctx);
+               return ERR_PTR(err);
+       }
+
        list_add(&ctx->list, &local->chanctx_list);
 
        return ctx;
 
        WARN_ON_ONCE(ctx->refcount != 0);
 
+       drv_remove_chanctx(local, ctx);
+
        list_del(&ctx->list);
        kfree_rcu(ctx, rcu_head);
 }
 static int ieee80211_assign_vif_chanctx(struct ieee80211_sub_if_data *sdata,
                                        struct ieee80211_chanctx *ctx)
 {
-       struct ieee80211_local *local __maybe_unused = sdata->local;
+       struct ieee80211_local *local = sdata->local;
+       int ret;
 
        lockdep_assert_held(&local->chanctx_mtx);
 
+       ret = drv_assign_vif_chanctx(local, sdata, ctx);
+       if (ret)
+               return ret;
+
        rcu_assign_pointer(sdata->vif.chanctx_conf, &ctx->conf);
        ctx->refcount++;
 
 static void ieee80211_unassign_vif_chanctx(struct ieee80211_sub_if_data *sdata,
                                           struct ieee80211_chanctx *ctx)
 {
-       struct ieee80211_local *local __maybe_unused = sdata->local;
+       struct ieee80211_local *local = sdata->local;
 
        lockdep_assert_held(&local->chanctx_mtx);
 
        ctx->refcount--;
        rcu_assign_pointer(sdata->vif.chanctx_conf, NULL);
+
+       drv_unassign_vif_chanctx(local, sdata, ctx);
 }
 
 static void __ieee80211_vif_release_channel(struct ieee80211_sub_if_data *sdata)