enroll.c 4.5 KB
Newer Older
Daniel Drake's avatar
Daniel Drake committed
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
/*
 * fprintd example to enroll right index finger
 * Copyright (C) 2008 Daniel Drake <dsd@gentoo.org>
 *
 * This program is free software; you can redistribute it and/or modify
 * it under the terms of the GNU General Public License as published by
 * the Free Software Foundation; either version 2 of the License, or
 * (at your option) any later version.
 * 
 * This program is distributed in the hope that it will be useful,
 * but WITHOUT ANY WARRANTY; without even the implied warranty of
 * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
 * GNU General Public License for more details.
 * 
 * You should have received a copy of the GNU General Public License along
 * with this program; if not, write to the Free Software Foundation, Inc.,
 * 51 Franklin Street, Fifth Floor, Boston, MA 02110-1301 USA.
 */

#include <stdlib.h>
#include <dbus/dbus-glib-bindings.h>
#include "manager-dbus-glue.h"
#include "device-dbus-glue.h"

static DBusGProxy *manager = NULL;
static DBusGConnection *connection = NULL;

enum enroll_result {
	ENROLL_COMPLETE = 1,
	ENROLL_FAIL,
	ENROLL_PASS,
	ENROLL_RETRY = 100,
	ENROLL_RETRY_TOO_SHORT = 101,
	ENROLL_RETRY_CENTER_FINGER = 102,
	ENROLL_RETRY_REMOVE_FINGER = 103,
};

static const char *enroll_result_str(int result)
{
	switch (result) {
	case ENROLL_COMPLETE:
		return "Enroll completed.";
	case ENROLL_FAIL:
		return "Enroll failed :(";
	case ENROLL_PASS:
		return "Enroll stage passed. Please scan again for next stage.";
	case ENROLL_RETRY:
		return "Retry scan";
	case ENROLL_RETRY_TOO_SHORT:
		return "Swipe too short, please retry";
	case ENROLL_RETRY_CENTER_FINGER:
		return "Finger not centered, please retry";
	case ENROLL_RETRY_REMOVE_FINGER:
		return "Please remove finger and retry";
	default:
		return "Unknown";
	}
}

static void create_manager(void)
{
	GError *error = NULL;

64
	connection = dbus_g_bus_get(DBUS_BUS_SYSTEM, &error);
Daniel Drake's avatar
Daniel Drake committed
65
66
67
	if (connection == NULL)
		g_error("Failed to connect to session bus: %s", error->message);

68
	manager = dbus_g_proxy_new_for_name(connection,
Daniel Drake's avatar
Daniel Drake committed
69
		"net.reactivated.Fprint", "/net/reactivated/Fprint/Manager",
70
		"net.reactivated.Fprint.Manager");
Daniel Drake's avatar
Daniel Drake committed
71
72
}

73
static DBusGProxy *open_device(const char *username)
Daniel Drake's avatar
Daniel Drake committed
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
{
	GError *error = NULL;
	GPtrArray *devices;
	gchar *path;
	DBusGProxy *dev;
	guint i;

	if (!net_reactivated_Fprint_Manager_get_devices(manager, &devices, &error))
		g_error("list_devices failed: %s", error->message);
	
	if (devices->len == 0) {
		g_print("No devices found\n");
		exit(1);
	}

	g_print("found %d devices\n", devices->len);
	for (i = 0; i < devices->len; i++) {
		path = g_ptr_array_index(devices, i);
		g_print("Device at %s\n", path);
	}

	path = g_ptr_array_index(devices, 0);
	g_print("Using device %s\n", path);

	/* FIXME use for_name_owner?? */
	dev = dbus_g_proxy_new_for_name(connection, "net.reactivated.Fprint",
		path, "net.reactivated.Fprint.Device");

	g_ptr_array_foreach(devices, (GFunc) g_free, NULL);
	g_ptr_array_free(devices, TRUE);

105
	if (!net_reactivated_Fprint_Device_claim(dev, username, &error))
Daniel Drake's avatar
Daniel Drake committed
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
121
122
123
124
125
126
127
		g_error("failed to claim device: %s", error->message);
	return dev;
}

static void enroll_result(GObject *object, int result, void *user_data)
{
	gboolean *enroll_completed = user_data;
	g_print("Enroll result: %s (%d)\n", enroll_result_str(result), result);
	if (result == ENROLL_COMPLETE || result == ENROLL_FAIL)
		*enroll_completed = TRUE;
}

static void do_enroll(DBusGProxy *dev)
{
	GError *error;
	gboolean enroll_completed = FALSE;

	dbus_g_proxy_add_signal(dev, "EnrollStatus", G_TYPE_INT, NULL);
	dbus_g_proxy_connect_signal(dev, "EnrollStatus", G_CALLBACK(enroll_result),
		&enroll_completed, NULL);

	g_print("Enrolling right index finger.\n");
128
	if (!net_reactivated_Fprint_Device_enroll_start(dev, "right-index-finger", &error))
Daniel Drake's avatar
Daniel Drake committed
129
130
131
132
133
134
135
136
137
138
139
140
141
142
143
144
145
146
147
148
149
150
151
		g_error("EnrollStart failed: %s", error->message);

	while (!enroll_completed)
		g_main_context_iteration(NULL, TRUE);

	dbus_g_proxy_disconnect_signal(dev, "EnrollStatus",
		G_CALLBACK(enroll_result), &enroll_completed);

	if (!net_reactivated_Fprint_Device_enroll_stop(dev, &error))
		g_error("VerifyStop failed: %s", error->message);
}

static void release_device(DBusGProxy *dev)
{
	GError *error = NULL;
	if (!net_reactivated_Fprint_Device_release(dev, &error))
		g_error("ReleaseDevice failed: %s", error->message);
}

int main(int argc, char **argv)
{
	GMainLoop *loop;
	DBusGProxy *dev;
152
	char *username;
Daniel Drake's avatar
Daniel Drake committed
153
154
155
156
157

	g_type_init();
	loop = g_main_loop_new(NULL, FALSE);
	create_manager();

158
159
160
161
	username = NULL;
	if (argc == 2)
		username = argv[1];
	dev = open_device(username);
Daniel Drake's avatar
Daniel Drake committed
162
163
164
165
166
	do_enroll(dev);
	release_device(dev);
	return 0;
}