fs/ksmbd/mgmt/tree_connect.c

   1 // SPDX-License-Identifier: GPL-2.0-or-later
   2 /*
   3  *   Copyright (C) 2018 Samsung Electronics Co., Ltd.
   4  */
   5
   6 #include <linux/list.h>
   7 #include <linux/slab.h>
   8 #include <linux/xarray.h>
   9
  10 #include "../transport_ipc.h"
  11 #include "../connection.h"
  12
  13 #include "tree_connect.h"
  14 #include "user_config.h"
  15 #include "share_config.h"
  16 #include "user_session.h"
  17
  18 struct ksmbd_tree_conn_status
  19 ksmbd_tree_conn_connect(struct ksmbd_conn *conn, struct ksmbd_session *sess,
  20                         char *share_name)
  21 {
  22         struct ksmbd_tree_conn_status status = {-ENOENT, NULL};
  23         struct ksmbd_tree_connect_response *resp = NULL;
  24         struct ksmbd_share_config *sc;
  25         struct ksmbd_tree_connect *tree_conn = NULL;
  26         struct sockaddr *peer_addr;
  27         int ret;
  28
  29         sc = ksmbd_share_config_get(share_name);
  30         if (!sc)
  31                 return status;
  32
  33         tree_conn = kzalloc(sizeof(struct ksmbd_tree_connect), GFP_KERNEL);
  34         if (!tree_conn) {
  35                 status.ret = -ENOMEM;
  36                 goto out_error;
  37         }
  38
  39         tree_conn->id = ksmbd_acquire_tree_conn_id(sess);
  40         if (tree_conn->id < 0) {
  41                 status.ret = -EINVAL;
  42                 goto out_error;
  43         }
  44
  45         peer_addr = KSMBD_TCP_PEER_SOCKADDR(conn);
  46         resp = ksmbd_ipc_tree_connect_request(sess,
  47                                               sc,
  48                                               tree_conn,
  49                                               peer_addr);
  50         if (!resp) {
  51                 status.ret = -EINVAL;
  52                 goto out_error;
  53         }
  54
  55         status.ret = resp->status;
  56         if (status.ret != KSMBD_TREE_CONN_STATUS_OK)
  57                 goto out_error;
  58
  59         tree_conn->flags = resp->connection_flags;
  60         if (test_tree_conn_flag(tree_conn, KSMBD_TREE_CONN_FLAG_UPDATE)) {
  61                 struct ksmbd_share_config *new_sc;
  62
  63                 ksmbd_share_config_del(sc);
  64                 new_sc = ksmbd_share_config_get(share_name);
  65                 if (!new_sc) {
  66                         pr_err("Failed to update stale share config\n");
  67                         status.ret = -ESTALE;
  68                         goto out_error;
  69                 }
  70                 ksmbd_share_config_put(sc);
  71                 sc = new_sc;
  72         }
  73
  74         tree_conn->user = sess->user;
  75         tree_conn->share_conf = sc;
  76         status.tree_conn = tree_conn;
  77
  78         ret = xa_err(xa_store(&sess->tree_conns, tree_conn->id, tree_conn,
  79                               GFP_KERNEL));
  80         if (ret) {
  81                 status.ret = -ENOMEM;
  82                 goto out_error;
  83         }
  84         kvfree(resp);
  85         return status;
  86
  87 out_error:
  88         if (tree_conn)
  89                 ksmbd_release_tree_conn_id(sess, tree_conn->id);
  90         ksmbd_share_config_put(sc);
  91         kfree(tree_conn);
  92         kvfree(resp);
  93         return status;
  94 }
  95
  96 int ksmbd_tree_conn_disconnect(struct ksmbd_session *sess,
  97                                struct ksmbd_tree_connect *tree_conn)
  98 {
  99         int ret;
 100
 101         ret = ksmbd_ipc_tree_disconnect_request(sess->id, tree_conn->id);
 102         ksmbd_release_tree_conn_id(sess, tree_conn->id);
 103         xa_erase(&sess->tree_conns, tree_conn->id);
 104         ksmbd_share_config_put(tree_conn->share_conf);
 105         kfree(tree_conn);
 106         return ret;
 107 }
 108
 109 struct ksmbd_tree_connect *ksmbd_tree_conn_lookup(struct ksmbd_session *sess,
 110                                                   unsigned int id)
 111 {
 112         return xa_load(&sess->tree_conns, id);
 113 }
 114
 115 struct ksmbd_share_config *ksmbd_tree_conn_share(struct ksmbd_session *sess,
 116                                                  unsigned int id)
 117 {
 118         struct ksmbd_tree_connect *tc;
 119
 120         tc = ksmbd_tree_conn_lookup(sess, id);
 121         if (tc)
 122                 return tc->share_conf;
 123         return NULL;
 124 }
 125
 126 int ksmbd_tree_conn_session_logoff(struct ksmbd_session *sess)
 127 {
 128         int ret = 0;
 129         struct ksmbd_tree_connect *tc;
 130         unsigned long id;
 131
 132         xa_for_each(&sess->tree_conns, id, tc)
 133                 ret |= ksmbd_tree_conn_disconnect(sess, tc);
 134         xa_destroy(&sess->tree_conns);
 135         return ret;
 136 }