src/driver.c

   1 /*
   2  * driver.c: Helpers for loading drivers
   3  *
   4  * Copyright (C) 2006-2011 Red Hat, Inc.
   5  *
   6  * This library is free software; you can redistribute it and/or
   7  * modify it under the terms of the GNU Lesser General Public
   8  * License as published by the Free Software Foundation; either
   9  * version 2.1 of the License, or (at your option) any later version.
  10  *
  11  * This library is distributed in the hope that it will be useful,
  12  * but WITHOUT ANY WARRANTY; without even the implied warranty of
  13  * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the GNU
  14  * Lesser General Public License for more details.
  15  *
  16  * You should have received a copy of the GNU Lesser General Public
  17  * License along with this library.  If not, see
  18  * <http://www.gnu.org/licenses/>.
  19  *
  20  */
  21
  22
  23 #include <config.h>
  24
  25 #include <unistd.h>
  26
  27 #include "driver.h"
  28 #include "viralloc.h"
  29 #include "virfile.h"
  30 #include "virlog.h"
  31 #include "virmodule.h"
  32 #include "virthread.h"
  33 #include "configmake.h"
  34
  35 VIR_LOG_INIT("driver");
  36
  37 #define VIR_FROM_THIS VIR_FROM_NONE
  38
  39 /* XXX re-implement this for other OS, or use libtools helper lib ? */
  40 #define DEFAULT_DRIVER_DIR LIBDIR "/libvirt/connection-driver"
  41
  42
  43
  44 int
  45 virDriverLoadModule(const char *name,
  46                     const char *regfunc,
  47                     bool required)
  48 {
  49     char *modfile = NULL;
  50     int ret;
  51
  52     VIR_DEBUG("Module load %s", name);
  53
  54     if (!(modfile = virFileFindResourceFull(name,
  55                                             "libvirt_driver_",
  56                                             ".so",
  57                                             abs_top_builddir "/src/.libs",
  58                                             DEFAULT_DRIVER_DIR,
  59                                             "LIBVIRT_DRIVER_DIR")))
  60         return -1;
  61
  62     ret = virModuleLoad(modfile, regfunc, required);
  63
  64     VIR_FREE(modfile);
  65
  66     return ret;
  67 }
  68
  69
  70 /* XXX unload modules, but we can't until we can unregister libvirt drivers */
  71
  72 virThreadLocal connectInterface;
  73 virThreadLocal connectNetwork;
  74 virThreadLocal connectNWFilter;
  75 virThreadLocal connectNodeDev;
  76 virThreadLocal connectSecret;
  77 virThreadLocal connectStorage;
  78
  79 static int
  80 virConnectCacheOnceInit(void)
  81 {
  82     if (virThreadLocalInit(&connectInterface, NULL) < 0)
  83         return -1;
  84     if (virThreadLocalInit(&connectNetwork, NULL) < 0)
  85         return -1;
  86     if (virThreadLocalInit(&connectNWFilter, NULL) < 0)
  87         return -1;
  88     if (virThreadLocalInit(&connectNodeDev, NULL) < 0)
  89         return -1;
  90     if (virThreadLocalInit(&connectSecret, NULL) < 0)
  91         return -1;
  92     if (virThreadLocalInit(&connectStorage, NULL) < 0)
  93         return -1;
  94     return 0;
  95 }
  96
  97 VIR_ONCE_GLOBAL_INIT(virConnectCache);
  98
  99 virConnectPtr virGetConnectInterface(void)
 100 {
 101     virConnectPtr conn;
 102
 103     if (virConnectCacheInitialize() < 0)
 104         return NULL;
 105
 106     conn = virThreadLocalGet(&connectInterface);
 107     if (conn) {
 108         VIR_DEBUG("Return cached interface connection %p", conn);
 109         virObjectRef(conn);
 110     } else {
 111         conn = virConnectOpen(geteuid() == 0 ? "interface:///system" : "interface:///session");
 112         VIR_DEBUG("Opened new interface connection %p", conn);
 113     }
 114     return conn;
 115 }
 116
 117 virConnectPtr virGetConnectNetwork(void)
 118 {
 119     virConnectPtr conn;
 120
 121     if (virConnectCacheInitialize() < 0)
 122         return NULL;
 123
 124     conn = virThreadLocalGet(&connectNetwork);
 125     if (conn) {
 126         VIR_DEBUG("Return cached network connection %p", conn);
 127         virObjectRef(conn);
 128     } else {
 129         conn = virConnectOpen(geteuid() == 0 ? "network:///system" : "network:///session");
 130         VIR_DEBUG("Opened new network connection %p", conn);
 131     }
 132     return conn;
 133 }
 134
 135 virConnectPtr virGetConnectNWFilter(void)
 136 {
 137     virConnectPtr conn;
 138
 139     if (virConnectCacheInitialize() < 0)
 140         return NULL;
 141
 142     conn = virThreadLocalGet(&connectNWFilter);
 143     if (conn) {
 144         VIR_DEBUG("Return cached nwfilter connection %p", conn);
 145         virObjectRef(conn);
 146     } else {
 147         conn = virConnectOpen(geteuid() == 0 ? "nwfilter:///system" : "nwfilter:///session");
 148         VIR_DEBUG("Opened new nwfilter connection %p", conn);
 149     }
 150     return conn;
 151 }
 152
 153 virConnectPtr virGetConnectNodeDev(void)
 154 {
 155     virConnectPtr conn;
 156
 157     if (virConnectCacheInitialize() < 0)
 158         return NULL;
 159
 160     conn = virThreadLocalGet(&connectNodeDev);
 161     if (conn) {
 162         VIR_DEBUG("Return cached nodedev connection %p", conn);
 163         virObjectRef(conn);
 164     } else {
 165         conn = virConnectOpen(geteuid() == 0 ? "nodedev:///system" : "nodedev:///session");
 166         VIR_DEBUG("Opened new nodedev connection %p", conn);
 167     }
 168     return conn;
 169 }
 170
 171 virConnectPtr virGetConnectSecret(void)
 172 {
 173     virConnectPtr conn;
 174
 175     if (virConnectCacheInitialize() < 0)
 176         return NULL;
 177
 178     conn = virThreadLocalGet(&connectSecret);
 179     if (conn) {
 180         VIR_DEBUG("Return cached secret connection %p", conn);
 181         virObjectRef(conn);
 182     } else {
 183         conn = virConnectOpen(geteuid() == 0 ? "secret:///system" : "secret:///session");
 184         VIR_DEBUG("Opened new secret connection %p", conn);
 185     }
 186     return conn;
 187 }
 188
 189 virConnectPtr virGetConnectStorage(void)
 190 {
 191     virConnectPtr conn;
 192
 193     if (virConnectCacheInitialize() < 0)
 194         return NULL;
 195
 196     conn = virThreadLocalGet(&connectStorage);
 197     if (conn) {
 198         VIR_DEBUG("Return cached storage connection %p", conn);
 199         virObjectRef(conn);
 200     } else {
 201         conn = virConnectOpen(geteuid() == 0 ? "storage:///system" : "storage:///session");
 202         VIR_DEBUG("Opened new storage connection %p", conn);
 203     }
 204     return conn;
 205 }
 206
 207
 208 int
 209 virSetConnectInterface(virConnectPtr conn)
 210 {
 211     if (virConnectCacheInitialize() < 0)
 212         return -1;
 213
 214     VIR_DEBUG("Override interface connection with %p", conn);
 215     return virThreadLocalSet(&connectInterface, conn);
 216 }
 217
 218
 219 int
 220 virSetConnectNetwork(virConnectPtr conn)
 221 {
 222     if (virConnectCacheInitialize() < 0)
 223         return -1;
 224
 225     VIR_DEBUG("Override network connection with %p", conn);
 226     return virThreadLocalSet(&connectNetwork, conn);
 227 }
 228
 229
 230 int
 231 virSetConnectNWFilter(virConnectPtr conn)
 232 {
 233     if (virConnectCacheInitialize() < 0)
 234         return -1;
 235
 236     VIR_DEBUG("Override nwfilter connection with %p", conn);
 237     return virThreadLocalSet(&connectNWFilter, conn);
 238 }
 239
 240
 241 int
 242 virSetConnectNodeDev(virConnectPtr conn)
 243 {
 244     if (virConnectCacheInitialize() < 0)
 245         return -1;
 246
 247     VIR_DEBUG("Override nodedev connection with %p", conn);
 248     return virThreadLocalSet(&connectNodeDev, conn);
 249 }
 250
 251
 252 int
 253 virSetConnectSecret(virConnectPtr conn)
 254 {
 255     if (virConnectCacheInitialize() < 0)
 256         return -1;
 257
 258     VIR_DEBUG("Override secret connection with %p", conn);
 259     return virThreadLocalSet(&connectSecret, conn);
 260 }
 261
 262
 263 int
 264 virSetConnectStorage(virConnectPtr conn)
 265 {
 266     if (virConnectCacheInitialize() < 0)
 267         return -1;
 268
 269     VIR_DEBUG("Override storage connection with %p", conn);
 270     return virThreadLocalSet(&connectStorage, conn);
 271 }