~ubuntu-branches/ubuntu/wily/bluez/wily

* gcc -g -Wall -I../src -I../lib/ -I../include -DSTORAGEDIR=\"/var/lib/bluetooth\" -o sixpair sixpair.c ../src/storage.c ../common/libhelper.a -I../common `pkg-config --libs --cflags glib-2.0 libusb-1.0` -lbluetooth

#include <unistd.h>

#include <stdio.h>

#include <inttypes.h>

#include <sdp.h>

#include <bluetooth/bluetooth.h>

#include <bluetooth/sdp_lib.h>

#include <glib.h>

#include <libusb.h>

#include "storage.h"

/* Vendor and product ID for the Sixaxis PS3 controller */

#define VENDOR 0x054c

#define PRODUCT 0x0268

#define PS3_PNP_RECORD "3601920900000A000100000900013503191124090004350D35061901000900113503190011090006350909656E09006A0901000900093508350619112409010009000D350F350D350619010009001335031900110901002513576972656C65737320436F6E74726F6C6C65720901012513576972656C65737320436F6E74726F6C6C6572090102251B536F6E7920436F6D707574657220456E7465727461696E6D656E740902000901000902010901000902020800090203082109020428010902052801090206359A35980822259405010904A101A102850175089501150026FF00810375019513150025013500450105091901291381027501950D0600FF8103150026FF0005010901A10075089504350046FF0009300931093209358102C0050175089527090181027508953009019102750895300901B102C0A1028502750895300901B102C0A10285EE750895300901B102C0A10285EF750895300901B102C0C0090207350835060904090901000902082800090209280109020A280109020B09010009020C093E8009020D280009020E2800"

gboolean option_get_master = TRUE;

char *option_master= NULL;

gboolean option_store_info = TRUE;

const char *option_device = NULL;

gboolean option_quiet = FALSE;

const GOptionEntry options[] = {

{ "get-master", '\0', 0, G_OPTION_ARG_NONE, &option_get_master, "Get currently set master address", NULL },

{ "set-master", '\0', 0, G_OPTION_ARG_STRING, &option_master, "Set master address (\"auto\" for automatic)", NULL },

{ "store-info", '\0', 0, G_OPTION_ARG_NONE, &option_store_info, "Store the HID info into the input database", NULL },

{ "device", '\0', 0, G_OPTION_ARG_STRING, &option_device, "Only handle one device (default, all supported", NULL },

{ "quiet", 'q', 0, G_OPTION_ARG_NONE, &option_quiet, "Quieten the output", NULL },

{ NULL }

};

static gboolean

show_master (libusb_device_handle *devh, int itfnum)

{

unsigned char msg[8];

int res;

res = libusb_control_transfer (devh,

LIBUSB_ENDPOINT_IN | LIBUSB_REQUEST_TYPE_CLASS | LIBUSB_RECIPIENT_INTERFACE,

0x01, 0x03f5, itfnum,

(void*) msg, sizeof(msg),

5000);

if (res < 0) {

g_warning ("Getting the master Bluetooth address failed");

return FALSE;

}

g_print ("Current Bluetooth master: %02X:%02X:%02X:%02X:%02X:%02X\n",

msg[2], msg[3], msg[4], msg[5], msg[6], msg[7]);

return TRUE;

}

static char *

get_bdaddr (libusb_device_handle *devh, int itfnum)

{

unsigned char msg[17];

char *address;

int res;

res = libusb_control_transfer (devh,

LIBUSB_ENDPOINT_IN | LIBUSB_REQUEST_TYPE_CLASS | LIBUSB_RECIPIENT_INTERFACE,

0x01, 0x03f2, itfnum,

(void*) msg, sizeof(msg),

5000);

if (res < 0) {

g_warning ("Getting the device Bluetooth address failed");

return NULL;

}

address = g_strdup_printf ("%02X:%02X:%02X:%02X:%02X:%02X",

msg[4], msg[5], msg[6], msg[7], msg[8], msg[9]);

if (option_quiet == FALSE) {

g_print ("Device Bluetooth address: %s\n", address);

}

return address;

}

static gboolean

set_master_bdaddr (libusb_device_handle *devh, int itfnum, char *host)

{

unsigned char msg[8];

int mac[6];

int res;

if (sscanf(host, "%X:%X:%X:%X:%X:%X",

&mac[0],&mac[1],&mac[2],&mac[3],&mac[4],&mac[5]) != 6) {

return FALSE;

}

100

msg[0] = 0x01;

101

msg[1] = 0x00;

102

msg[2] = mac[0];

103

msg[3] = mac[1];

104

msg[4] = mac[2];

105

msg[5] = mac[3];

106

msg[6] = mac[4];

107

msg[7] = mac[5];

108

109

res = libusb_control_transfer (devh,

110

LIBUSB_ENDPOINT_OUT | LIBUSB_REQUEST_TYPE_CLASS | LIBUSB_RECIPIENT_INTERFACE,

111

0x09, 0x03f5, itfnum,

112

(void*) msg, sizeof(msg),

113

5000);

114

115

if (res < 0) {

116

g_warning ("Setting the master Bluetooth address failed");

117

return FALSE;

118

}

119

120

return TRUE;

121

}

122

123

static char *

124

get_host_bdaddr (void)

125

{

126

FILE *f;

127

int mac[6];

128

129

//FIXME use dbus to get the default adapter

130

131

f = popen("hcitool dev", "r");

132

133

if (f == NULL) {

134

//FIXME

135

return NULL;

136

}

137

if (fscanf(f, "%*s\n%*s %X:%X:%X:%X:%X:%X",

138

&mac[0],&mac[1],&mac[2],&mac[3],&mac[4],&mac[5]) != 6) {

139

//FIXME

140

return NULL;

141

}

142

143

return g_strdup_printf ("%02X:%02X:%02X:%02X:%02X:%02X", mac[0], mac[1], mac[2], mac[3], mac[4], mac[5]);

144

}

145

146

static int

147

handle_device (libusb_device *dev, struct libusb_config_descriptor *cfg, int itfnum, const struct libusb_interface_descriptor *alt)

148

{

149

libusb_device_handle *devh;

150

int res, retval;

151

152

retval = -1;

153

154

if (libusb_open (dev, &devh) < 0) {

155

g_warning ("Can't open device");

156

goto bail;

157

}

158

libusb_detach_kernel_driver (devh, itfnum);

159

160

res = libusb_claim_interface (devh, itfnum);

161

if (res < 0) {

162

g_warning ("Can't claim interface %d", itfnum);

163

goto bail;

164

}

165

166

if (option_get_master != FALSE) {

167

if (show_master (devh, itfnum) == FALSE)

168

goto bail;

169

retval = 0;

170

}

171

172

if (option_master != NULL) {

173

if (strcmp (option_master, "auto") == 0) {

174

g_free (option_master);

175

option_master = get_host_bdaddr ();

176

if (option_master == NULL) {

177

g_warning ("Can't get bdaddr from default device");

178

retval = -1;

179

goto bail;

180

}

181

}

182

} else {

183

option_master = get_host_bdaddr ();

184

if (option_master == NULL) {

185

g_warning ("Can't get bdaddr from default device");

186

retval = -1;

187

goto bail;

188

}

189

}

190

191

if (option_store_info != FALSE) {

192

sdp_record_t *rec;

193

char *device;

194

bdaddr_t dst, src;

195

196

device = get_bdaddr (devh, itfnum);

197

if (device == NULL) {

198

retval = -1;

199

goto bail;

200

}

201

202

rec = record_from_string (PS3_PNP_RECORD);

203

store_record(option_master, device, rec);

204

write_trust(option_master, device, "[all]", TRUE);

205

store_device_id(option_master, device, 0xffff, 0x054c, 0x0268, 0);

206

str2ba(option_master, &src);

207

str2ba(device, &dst);

208

write_device_profiles(&src, &dst, "");

209

write_device_name(&src, &dst, "PLAYSTATION(R)3 Controller");

210

sdp_record_free(rec);

211

212

if (set_master_bdaddr (devh, itfnum, option_master) == FALSE) {

213

retval = -1;

214

goto bail;

215

}

216

}

217

218

bail:

219

libusb_release_interface (devh, itfnum);

220

res = libusb_attach_kernel_driver(devh, itfnum);

221

if (res < 0) {

222

//FIXME sometimes the kernel tells us ENOENT, but succeeds anyway...

223

g_warning ("Reattaching the driver failed: %d", res);

224

}

225

if (devh != NULL)

226

libusb_close (devh);

227

228

return retval;

229

}

230

231

int main (int argc, char **argv)

232

{

233

GOptionContext *context;

234

GError *error = NULL;

235

libusb_device **list;

236

ssize_t num_devices, i;

237

238

context = g_option_context_new ("- Manage Sixaxis PS3 controllers");

239

g_option_context_add_main_entries (context, options, NULL);

240

if (g_option_context_parse (context, &argc, &argv, &error) == FALSE) {

241

g_warning ("Couldn't parse command-line options: %s", error->message);

242

return 1;

243

}

244

245

/* Check that the passed bdaddr is correct */

246

if (option_master != NULL && strcmp (option_master, "auto") != 0) {

247

//FIXME check bdaddr

248

}

249

250

libusb_init (NULL);

251

252

/* Find device(s) */

253

num_devices = libusb_get_device_list (NULL, &list);

254

if (num_devices < 0) {

255

g_warning ("libusb_get_device_list failed");

256

return 1;

257

}

258

259

for (i = 0; i < num_devices; i++) {

260

struct libusb_config_descriptor *cfg;

261

libusb_device *dev = list[i];

262

struct libusb_device_descriptor desc;

263

guint8 j;

264

265

if (libusb_get_device_descriptor (dev, &desc) < 0) {

266

g_warning ("libusb_get_device_descriptor failed");

267

continue;

268

}

269

270

/* Here we check for the supported devices */

271

if (desc.idVendor != VENDOR || desc.idProduct != PRODUCT)

272

continue;

273

274

/* Look for the interface number that interests us */

275

for (j = 0; j < desc.bNumConfigurations; j++) {

276

struct libusb_config_descriptor *config;

277

guint8 k;

278

279

libusb_get_config_descriptor (dev, j, &config);

280

281

for (k = 0; k < config->bNumInterfaces; k++) {

282

const struct libusb_interface *itf = &config->interface[k];

283

int l;

284

285

for (l = 0; l < itf->num_altsetting ; l++) {

286

struct libusb_interface_descriptor alt;

287

288

alt = itf->altsetting[l];

289

if (alt.bInterfaceClass == 3) {

290

handle_device (dev, cfg, l, &alt);

291

}

292

}

293

}

294

}

295

}

296

297

return 0;

298

}

299

Older »