static void smc_conn_abort(struct smc_sock *smc, int local_first)
 {
+       struct smc_connection *conn = &smc->conn;
+       struct smc_link_group *lgr = conn->lgr;
+
+       smc_conn_free(conn);
        if (local_first)
-               smc_lgr_cleanup_early(&smc->conn);
-       else
-               smc_conn_free(&smc->conn);
+               smc_lgr_cleanup_early(lgr);
 }
 
 /* check if there is a rdma device available for this connection. */
 
 
        if (!conn->lgr->is_smcd) {
                rc = smcr_lgr_conn_assign_link(conn, first);
-               if (rc)
+               if (rc) {
+                       conn->lgr = NULL;
                        return rc;
+               }
        }
        /* find a new alert_token_local value not yet used by some connection
         * in this link group
        return skb->len;
 }
 
-void smc_lgr_cleanup_early(struct smc_connection *conn)
+void smc_lgr_cleanup_early(struct smc_link_group *lgr)
 {
-       struct smc_link_group *lgr = conn->lgr;
        spinlock_t *lgr_lock;
 
        if (!lgr)
                return;
 
-       smc_conn_free(conn);
        smc_lgr_list_head(lgr, &lgr_lock);
        spin_lock_bh(lgr_lock);
        /* do not use this link group for new connections */
                write_lock_bh(&lgr->conns_lock);
                rc = smc_lgr_register_conn(conn, true);
                write_unlock_bh(&lgr->conns_lock);
-               if (rc)
+               if (rc) {
+                       smc_lgr_cleanup_early(lgr);
                        goto out;
+               }
        }
        conn->local_tx_ctrl.common.type = SMC_CDC_MSG_TYPE;
        conn->local_tx_ctrl.len = SMC_WR_TX_SIZE;
 
 struct smc_sock;
 struct smc_clc_msg_accept_confirm;
 
-void smc_lgr_cleanup_early(struct smc_connection *conn);
+void smc_lgr_cleanup_early(struct smc_link_group *lgr);
 void smc_lgr_terminate_sched(struct smc_link_group *lgr);
 void smcr_port_add(struct smc_ib_device *smcibdev, u8 ibport);
 void smcr_port_err(struct smc_ib_device *smcibdev, u8 ibport);