← Back to branch summary

~zulcss/samba/server-dailies-3.4

« back to all changes in this revision

Viewing changes to source3/smbd/avahi_register.c

  • Committer: Chuck Short
  • Date: 2010-09-28 20:38:39 UTC
  • Revision ID: zulcss@ubuntu.com-20100928203839-pgjulytsi9ue63x1
Initial version

Show diffs side-by-side

added added

removed removed

Lines of Context:
source3/smbd/avahi_register.c
 
1
/*
 
2
 * Unix SMB/CIFS implementation.
 
3
 * Register _smb._tcp with avahi
 
4
 *
 
5
 * Copyright (C) Volker Lendecke 2009
 
6
 *
 
7
 * This program is free software; you can redistribute it and/or modify
 
8
 * it under the terms of the GNU General Public License as published by
 
9
 * the Free Software Foundation; either version 3 of the License, or
 
10
 * (at your option) any later version.
 
11
 *
 
12
 * This program is distributed in the hope that it will be useful,
 
13
 * but WITHOUT ANY WARRANTY; without even the implied warranty of
 
14
 * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
 
15
 * GNU General Public License for more details.
 
16
 *
 
17
 * You should have received a copy of the GNU General Public License
 
18
 * along with this program; if not, see <http://www.gnu.org/licenses/>.
 
19
 */
 
20
 
 
21
#include "includes.h"
 
22
 
 
23
#include <avahi-client/client.h>
 
24
#include <avahi-client/publish.h>
 
25
#include <avahi-common/error.h>
 
26
 
 
27
struct avahi_state_struct {
 
28
        struct AvahiPoll *poll;
 
29
        AvahiClient *client;
 
30
        AvahiEntryGroup *entry_group;
 
31
        uint16_t port;
 
32
};
 
33
 
 
34
static void avahi_entry_group_callback(AvahiEntryGroup *g,
 
35
                                       AvahiEntryGroupState status,
 
36
                                       void *userdata)
 
37
{
 
38
        struct avahi_state_struct *state = talloc_get_type_abort(
 
39
                userdata, struct avahi_state_struct);
 
40
        int error;
 
41
 
 
42
        switch (status) {
 
43
        case AVAHI_ENTRY_GROUP_ESTABLISHED:
 
44
                DEBUG(10, ("avahi_entry_group_callback: "
 
45
                           "AVAHI_ENTRY_GROUP_ESTABLISHED\n"));
 
46
                break;
 
47
        case AVAHI_ENTRY_GROUP_FAILURE:
 
48
                error = avahi_client_errno(state->client);
 
49
 
 
50
                DEBUG(10, ("avahi_entry_group_callback: "
 
51
                           "AVAHI_ENTRY_GROUP_FAILURE: %s\n",
 
52
                           avahi_strerror(error)));
 
53
                break;
 
54
        case AVAHI_ENTRY_GROUP_COLLISION:
 
55
                DEBUG(10, ("avahi_entry_group_callback: "
 
56
                           "AVAHI_ENTRY_GROUP_COLLISION\n"));
 
57
                break;
 
58
        case AVAHI_ENTRY_GROUP_UNCOMMITED:
 
59
                DEBUG(10, ("avahi_entry_group_callback: "
 
60
                           "AVAHI_ENTRY_GROUP_UNCOMMITED\n"));
 
61
                break;
 
62
        case AVAHI_ENTRY_GROUP_REGISTERING:
 
63
                DEBUG(10, ("avahi_entry_group_callback: "
 
64
                           "AVAHI_ENTRY_GROUP_REGISTERING\n"));
 
65
                break;
 
66
        }
 
67
}
 
68
 
 
69
static void avahi_client_callback(AvahiClient *c, AvahiClientState status,
 
70
                                  void *userdata)
 
71
{
 
72
        struct avahi_state_struct *state = talloc_get_type_abort(
 
73
                userdata, struct avahi_state_struct);
 
74
        int error;
 
75
 
 
76
        switch (status) {
 
77
        case AVAHI_CLIENT_S_RUNNING:
 
78
                DEBUG(10, ("avahi_client_callback: AVAHI_CLIENT_S_RUNNING\n"));
 
79
 
 
80
                state->entry_group = avahi_entry_group_new(
 
81
                        c, avahi_entry_group_callback, state);
 
82
                if (state->entry_group == NULL) {
 
83
                        error = avahi_client_errno(c);
 
84
                        DEBUG(10, ("avahi_entry_group_new failed: %s\n",
 
85
                                   avahi_strerror(error)));
 
86
                        break;
 
87
                }
 
88
                if (avahi_entry_group_add_service(
 
89
                            state->entry_group, AVAHI_IF_UNSPEC,
 
90
                            AVAHI_PROTO_UNSPEC, 0, global_myname(),
 
91
                            "_smb._tcp", NULL, NULL, state->port, NULL) < 0) {
 
92
                        error = avahi_client_errno(c);
 
93
                        DEBUG(10, ("avahi_entry_group_add_service failed: "
 
94
                                   "%s\n", avahi_strerror(error)));
 
95
                        avahi_entry_group_free(state->entry_group);
 
96
                        state->entry_group = NULL;
 
97
                        break;
 
98
                }
 
99
                if (avahi_entry_group_commit(state->entry_group) < 0) {
 
100
                        error = avahi_client_errno(c);
 
101
                        DEBUG(10, ("avahi_entry_group_commit failed: "
 
102
                                   "%s\n", avahi_strerror(error)));
 
103
                        avahi_entry_group_free(state->entry_group);
 
104
                        state->entry_group = NULL;
 
105
                        break;
 
106
                }
 
107
                break;
 
108
        case AVAHI_CLIENT_FAILURE:
 
109
                error = avahi_client_errno(c);
 
110
 
 
111
                DEBUG(10, ("avahi_client_callback: AVAHI_CLIENT_FAILURE: %s\n",
 
112
                           avahi_strerror(error)));
 
113
 
 
114
                if (error != AVAHI_ERR_DISCONNECTED) {
 
115
                        break;
 
116
                }
 
117
                avahi_client_free(c);
 
118
                state->client = avahi_client_new(state->poll, AVAHI_CLIENT_NO_FAIL,
 
119
                                                 avahi_client_callback, state,
 
120
                                                 &error);
 
121
                if (state->client == NULL) {
 
122
                        DEBUG(10, ("avahi_client_new failed: %s\n",
 
123
                                   avahi_strerror(error)));
 
124
                        break;
 
125
                }
 
126
                break;
 
127
        case AVAHI_CLIENT_S_COLLISION:
 
128
                DEBUG(10, ("avahi_client_callback: "
 
129
                           "AVAHI_CLIENT_S_COLLISION\n"));
 
130
                break;
 
131
        case AVAHI_CLIENT_S_REGISTERING:
 
132
                DEBUG(10, ("avahi_client_callback: "
 
133
                           "AVAHI_CLIENT_S_REGISTERING\n"));
 
134
                break;
 
135
        case AVAHI_CLIENT_CONNECTING:
 
136
                DEBUG(10, ("avahi_client_callback: "
 
137
                           "AVAHI_CLIENT_CONNECTING\n"));
 
138
                break;
 
139
        }
 
140
}
 
141
 
 
142
void *avahi_start_register(TALLOC_CTX *mem_ctx, struct tevent_context *ev,
 
143
                           uint16_t port)
 
144
{
 
145
        struct avahi_state_struct *state;
 
146
        int error;
 
147
 
 
148
        state = talloc(mem_ctx, struct avahi_state_struct);
 
149
        if (state == NULL) {
 
150
                return state;
 
151
        }
 
152
        state->port = port;
 
153
        state->poll = tevent_avahi_poll(state, ev);
 
154
        if (state->poll == NULL) {
 
155
                goto fail;
 
156
        }
 
157
        state->client = avahi_client_new(state->poll, AVAHI_CLIENT_NO_FAIL,
 
158
                                         avahi_client_callback, state,
 
159
                                         &error);
 
160
        if (state->client == NULL) {
 
161
                DEBUG(10, ("avahi_client_new failed: %s\n",
 
162
                           avahi_strerror(error)));
 
163
                goto fail;
 
164
        }
 
165
        return state;
 
166
 
 
167
 fail:
 
168
        TALLOC_FREE(state);
 
169
        return NULL;
 
170
}