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
64
65
66
67
68
69
70
71
72
73
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
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
121
122
123
124
125
126
127
128
129
130
131
132
133
134
135
136
137
138
139
140
141
142
143
144
145
146
147
148
149
150
151
152
153
154
155
156
157
158
159
160
161
162
163
164
165
166
167
168
169
170
171
172
173
174
175
176
177
178
179
180
181
182
183
184
185
186
187
188
189
190
191
192
193
194
195
196
197
198
199
200
201
202
203
204
205
206
207
208
209
210
211
212
213
214
215
216
217
218
219
220
221
222
223
224
225
226
227
228
229
230
|
/* -*- Mode: C++; tab-width: 2; indent-tabs-mode: nil; c-basic-offset: 2 -*- */
/* vim: set et sw=2 ts=2: */
/***************************************************************************
* messagehandler.cc
*
* Mon Mar 26 15:35:24 CEST 2012
* Copyright 2012 Jonas Suhr Christensen
* jsc@umbraculum.org
****************************************************************************/
/*
* This file is part of Munia.
*
* Munia 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.
*
* Munia 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 Munia; if not, write to the Free Software
* Foundation, Inc., 59 Temple Place, Suite 330, Boston, MA 02111-1307 USA.
*/
#include "messagehandler.h"
#include "hugin.hpp"
// Defines global node_manager object
#include "nodemanager.h"
#include "messageparser.h"
#include "errorcodes.h"
MessageList handle_msg(MessageList msgList, clientid_t wsi)
{
MessageList outmsgs;
MessageList::iterator it;
for(it = msgList.begin(); it != msgList.end(); it++)
{
message_t &m = *it;
if(m.cmd == cmd::login)
{
connection_handler.login(wsi, m.login.user, m.login.password);
if(!connection_handler.authenticated(wsi))
{
outmsgs.push_back(create_msg_error(ErrorCode::LoginFailed,
"Bad username or password.", wsi));
continue;
}
}
// If client is not authenticated; do not continue beyond this point...
if(!connection_handler.authenticated(wsi))
{
outmsgs.push_back(create_msg_error(ErrorCode::Unauthorized,
"Not authorized.", wsi));
continue;
}
switch(m.cmd)
{
case cmd::login:
// Already handled, before this switch ... case
INFO(messagehandler, "Handling login command\n");
break;
case cmd::logout:
INFO(messagehandler, "Handling logout command\n");
connection_handler.logout(wsi);
break;
case cmd::create_with_attributes:
// Never sent to the server
INFO(messagehandler, "Handling create_with_attributes command\n");
break;
case cmd::create:
{
INFO(messagehandler, "Handling create command\n");
try
{
m.nodes = node_manager.createNode(m.create.parentid, m.create.id,
m.create.insertbeforeid);
outmsgs.push_back(m);
}
catch (const ErrorMessage& e)
{
// Set client id (wsi) and forward upstream
outmsgs.push_back(create_msg_error(e, wsi));
}
catch (...)
{
outmsgs.push_back(create_msg_error(ErrorCode::Unknown,
"Unknown create error", wsi));
}
}
break;
case cmd::remove:
{
INFO(messagehandler, "Handling remove command\n");
try
{
NodeIdList ids = node_manager.subNodes(m.remove.id);
NodeIdList::reverse_iterator id = ids.rbegin();
while(id != ids.rend())
{
node_t node = node_manager.node(*id);
message_t m = create_msg_remove(node);
m.nodes = node_manager.removeNode(node.id);
outmsgs.push_back(m);
id++;
}
}
catch (const ErrorMessage& e)
{
// Set client id (wsi) and forward upstream
outmsgs.push_back(create_msg_error(e, wsi));
}
catch (...)
{
outmsgs.push_back(create_msg_error(ErrorCode::Unknown,
"Unknown remove error", wsi));
}
}
break;
case cmd::move:
{
INFO(messagehandler, "Handling move command\n");
try
{
node_t removenode = node_manager.node(m.move.id);
NodeIdListPair tilpair =
node_manager.moveNode(m.move.id, m.move.parentid,
m.move.insertbeforeid);
node_t createnode = node_manager.node(m.move.id);
message_t removemsg = create_msg_remove(removenode);
removemsg.nodes = tilpair.first;
message_t createmsg =
create_msg_create_with_attributes(createnode,
m.move.insertbeforeid);
createmsg.create.attributes = createnode.attributes;
for(const auto& child_node_id : node_manager.subNodes(m.move.id))
{
if(child_node_id == m.move.id)
{
continue;
}
auto child_node = node_manager.node(child_node_id);
create_t child;
child.id = child_node.id;
child.parentid = child_node.parentid;
child.insertbeforeid = -1;
child.attributes = child_node.attributes;
createmsg.create.children.push_back(child);
}
createmsg.nodes = tilpair.second;
outmsgs.push_back(removemsg);
outmsgs.push_back(createmsg);
}
catch (const ErrorMessage& e)
{
// Set client id (wsi) and forward upstream
outmsgs.push_back(create_msg_error(e, wsi));
}
catch (...)
{
outmsgs.push_back(create_msg_error(ErrorCode::Unknown,
"Unknown move error", wsi));
}
}
break;
case cmd::subscribe:
INFO(messagehandler, "Handling subscribe command\n");
outmsgs.push_back(m);
break;
case cmd::unsubscribe:
INFO(messagehandler, "Handling unsubscribe command\n");
outmsgs.push_back(m);
break;
case cmd::update:
{
INFO(messagehandler, "Handling update command\n");
try
{
m.nodes = node_manager.updateNode(m.update.id,
m.update.attribute,
m.update.value);
outmsgs.push_back(m);
}
catch (const ErrorMessage& e)
{
// Set client id (wsi) and forward upstream
outmsgs.push_back(create_msg_error(e, wsi));
}
catch (...)
{
outmsgs.push_back(create_msg_error(ErrorCode::Unknown,
"Unknown update error", wsi));
}
}
break;
case cmd::error:
INFO(messagehandler, "Handling error command\n");
outmsgs.push_back(m); // Just forward the error message
break;
}
}
return outmsgs;
}
|
