net/dccp/ccid.c

   1 /*
   2  *  net/dccp/ccid.c
   3  *
   4  *  An implementation of the DCCP protocol
   5  *  Arnaldo Carvalho de Melo <acme@conectiva.com.br>
   6  *
   7  *  CCID infrastructure
   8  *
   9  *      This program is free software; you can redistribute it and/or modify it
  10  *      under the terms of the GNU General Public License version 2 as
  11  *      published by the Free Software Foundation.
  12  */
  13
  14 #include "ccid.h"
  15
  16 static struct ccid *ccids[CCID_MAX];
  17 #if defined(CONFIG_SMP) || defined(CONFIG_PREEMPT)
  18 static atomic_t ccids_lockct = ATOMIC_INIT(0);
  19 static DEFINE_SPINLOCK(ccids_lock);
  20
  21 /*
  22  * The strategy is: modifications ccids vector are short, do not sleep and
  23  * veeery rare, but read access should be free of any exclusive locks.
  24  */
  25 static void ccids_write_lock(void)
  26 {
  27         spin_lock(&ccids_lock);
  28         while (atomic_read(&ccids_lockct) != 0) {
  29                 spin_unlock(&ccids_lock);
  30                 yield();
  31                 spin_lock(&ccids_lock);
  32         }
  33 }
  34
  35 static inline void ccids_write_unlock(void)
  36 {
  37         spin_unlock(&ccids_lock);
  38 }
  39
  40 static inline void ccids_read_lock(void)
  41 {
  42         atomic_inc(&ccids_lockct);
  43         spin_unlock_wait(&ccids_lock);
  44 }
  45
  46 static inline void ccids_read_unlock(void)
  47 {
  48         atomic_dec(&ccids_lockct);
  49 }
  50
  51 #else
  52 #define ccids_write_lock() do { } while(0)
  53 #define ccids_write_unlock() do { } while(0)
  54 #define ccids_read_lock() do { } while(0)
  55 #define ccids_read_unlock() do { } while(0)
  56 #endif
  57
  58 int ccid_register(struct ccid *ccid)
  59 {
  60         int err;
  61
  62         if (ccid->ccid_init == NULL)
  63                 return -1;
  64
  65         ccids_write_lock();
  66         err = -EEXIST;
  67         if (ccids[ccid->ccid_id] == NULL) {
  68                 ccids[ccid->ccid_id] = ccid;
  69                 err = 0;
  70         }
  71         ccids_write_unlock();
  72         if (err == 0)
  73                 pr_info("CCID: Registered CCID %d (%s)\n",
  74                         ccid->ccid_id, ccid->ccid_name);
  75         return err;
  76 }
  77
  78 EXPORT_SYMBOL_GPL(ccid_register);
  79
  80 int ccid_unregister(struct ccid *ccid)
  81 {
  82         ccids_write_lock();
  83         ccids[ccid->ccid_id] = NULL;
  84         ccids_write_unlock();
  85         pr_info("CCID: Unregistered CCID %d (%s)\n",
  86                 ccid->ccid_id, ccid->ccid_name);
  87         return 0;
  88 }
  89
  90 EXPORT_SYMBOL_GPL(ccid_unregister);
  91
  92 struct ccid *ccid_init(unsigned char id, struct sock *sk)
  93 {
  94         struct ccid *ccid;
  95
  96 #ifdef CONFIG_KMOD
  97         if (ccids[id] == NULL)
  98                 request_module("net-dccp-ccid-%d", id);
  99 #endif
 100         ccids_read_lock();
 101
 102         ccid = ccids[id];
 103         if (ccid == NULL)
 104                 goto out;
 105
 106         if (!try_module_get(ccid->ccid_owner))
 107                 goto out_err;
 108
 109         if (ccid->ccid_init(sk) != 0)
 110                 goto out_module_put;
 111 out:
 112         ccids_read_unlock();
 113         return ccid;
 114 out_module_put:
 115         module_put(ccid->ccid_owner);
 116 out_err:
 117         ccid = NULL;
 118         goto out;
 119 }
 120
 121 EXPORT_SYMBOL_GPL(ccid_init);
 122
 123 void ccid_exit(struct ccid *ccid, struct sock *sk)
 124 {
 125         if (ccid == NULL)
 126                 return;
 127
 128         ccids_read_lock();
 129
 130         if (ccids[ccid->ccid_id] != NULL) {
 131                 if (ccid->ccid_exit != NULL)
 132                         ccid->ccid_exit(sk);
 133                 module_put(ccid->ccid_owner);
 134         }
 135
 136         ccids_read_unlock();
 137 }
 138
 139 EXPORT_SYMBOL_GPL(ccid_exit);