Rev 1642 | Rev 1647 | Go to most recent revision | Only display areas with differences | Regard whitespace | Details | Blame | Last modification | View Log | SVN | RSS feed
| Rev 1642 | Rev 1644 | ||
|---|---|---|---|
| 1 | /* |
1 | /* |
| 2 | * |
2 | * |
| 3 | * Copyright (C) 2010 Mattias Runge |
3 | * Copyright (C) 2010 Mattias Runge |
| 4 | * |
4 | * |
| 5 | * This program is free software; you can redistribute it and/or modify |
5 | * This program is free software; you can redistribute it and/or modify |
| 6 | * it under the terms of the GNU General Public License as published by |
6 | * it under the terms of the GNU General Public License as published by |
| 7 | * the Free Software Foundation; either version 2 of the License, or |
7 | * the Free Software Foundation; either version 2 of the License, or |
| 8 | * (at your option) any later version. |
8 | * (at your option) any later version. |
| 9 | * |
9 | * |
| 10 | * This program is distributed in the hope that it will be useful, |
10 | * This program is distributed in the hope that it will be useful, |
| 11 | * but WITHOUT ANY WARRANTY; without even the implied warranty of |
11 | * but WITHOUT ANY WARRANTY; without even the implied warranty of |
| 12 | * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the |
12 | * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the |
| 13 | * GNU General Public License for more details. |
13 | * GNU General Public License for more details. |
| 14 | * |
14 | * |
| 15 | * You should have received a copy of the GNU General Public License along |
15 | * You should have received a copy of the GNU General Public License along |
| 16 | * with this program; if not, write to the Free Software Foundation, Inc., |
16 | * with this program; if not, write to the Free Software Foundation, Inc., |
| 17 | * 51 Franklin Street, Fifth Floor, Boston, MA 02110-1301 USA. |
17 | * 51 Franklin Street, Fifth Floor, Boston, MA 02110-1301 USA. |
| 18 | * |
18 | * |
| 19 | */ |
19 | */ |
| 20 | 20 | ||
| 21 | #include "Manager.h" |
21 | #include "Manager.h" |
| 22 | 22 | ||
| 23 | #include <fstream> |
23 | #include <fstream> |
| 24 | 24 | ||
| 25 | #include <boost/lexical_cast.hpp> |
25 | #include <boost/lexical_cast.hpp> |
| 26 | #include <boost/bind.hpp> |
26 | #include <boost/bind.hpp> |
| 27 | 27 | ||
| 28 | namespace atom { |
28 | namespace atom { |
| 29 | namespace vm { |
29 | namespace vm { |
| 30 | 30 | ||
| 31 | Manager::Pointer Manager::instance_; |
31 | Manager::Pointer Manager::instance_; |
| 32 | 32 | ||
| 33 | Manager::Manager() : LOG("vm::Manager") |
33 | Manager::Manager() : LOG("vm::Manager") |
| 34 | { |
34 | { |
| 35 | } |
35 | } |
| 36 | 36 | ||
| 37 | Manager::~Manager() |
37 | Manager::~Manager() |
| 38 | { |
38 | { |
| 39 | this->io_service_.stop(); |
39 | this->io_service_.stop(); |
| 40 | 40 | ||
| 41 | this->plugins_.clear(); |
41 | this->plugins_.clear(); |
| 42 | this->context_.Dispose(); |
42 | this->context_.Dispose(); |
| 43 | 43 | ||
| 44 | for (ImportFunctionList::iterator it = this->functions_.begin(); it != this->functions_.end(); it++) |
44 | for (ImportFunctionList::iterator it = this->functions_.begin(); it != this->functions_.end(); it++) |
| 45 | { |
45 | { |
| 46 | it->second.Dispose(); |
46 | it->second.Dispose(); |
| 47 | } |
47 | } |
| 48 | } |
48 | } |
| 49 | 49 | ||
| 50 | void Manager::Create() |
50 | void Manager::Create() |
| 51 | { |
51 | { |
| 52 | Manager::instance_ = Manager::Pointer(new Manager()); |
52 | Manager::instance_ = Manager::Pointer(new Manager()); |
| 53 | } |
53 | } |
| 54 | 54 | ||
| 55 | Manager::Pointer Manager::Instance() |
55 | Manager::Pointer Manager::Instance() |
| 56 | { |
56 | { |
| 57 | return Manager::instance_; |
57 | return Manager::instance_; |
| 58 | } |
58 | } |
| 59 | 59 | ||
| 60 | void Manager::Delete() |
60 | void Manager::Delete() |
| 61 | { |
61 | { |
| 62 | Manager::instance_.reset(); |
62 | Manager::instance_.reset(); |
| 63 | } |
63 | } |
| 64 | 64 | ||
| 65 | v8::Persistent<v8::Context>& Manager::GetContext() |
65 | v8::Persistent<v8::Context>& Manager::GetContext() |
| 66 | { |
66 | { |
| 67 | return this->context_; |
67 | return this->context_; |
| 68 | } |
68 | } |
| 69 | 69 | ||
| 70 | void Manager::AddPlugin(Plugin::Pointer plugin) |
70 | void Manager::AddPlugin(Plugin::Pointer plugin) |
| 71 | { |
71 | { |
| 72 | this->plugins_.push_back(plugin); |
72 | this->plugins_.push_back(plugin); |
| 73 | } |
73 | } |
| 74 | 74 | ||
| 75 | void Manager::Start(std::string script_path) |
75 | void Manager::Start(std::string script_path) |
| 76 | { |
76 | { |
| 77 | this->script_path_ = script_path; |
77 | this->script_path_ = script_path; |
| 78 | this->io_service_.post(boost::bind(&Manager::StartHandler, this)); |
78 | this->io_service_.post(boost::bind(&Manager::StartHandler, this)); |
| 79 | } |
79 | } |
| 80 | 80 | ||
| 81 | void Manager::SendResponse(std::string plugin_name, unsigned int request_id, std::string response) |
81 | void Manager::SendResponse(std::string plugin_name, unsigned int request_id, std::string response) |
| 82 | { |
82 | { |
| 83 | for (unsigned int n = 0; n < this->plugins_.size(); n++) |
83 | for (unsigned int n = 0; n < this->plugins_.size(); n++) |
| 84 | { |
84 | { |
| 85 | if (this->plugins_[n]->GetName() == plugin_name) |
85 | if (this->plugins_[n]->GetName() == plugin_name) |
| 86 | { |
86 | { |
| 87 | this->plugins_[n]->ExecutionResult(response, request_id); |
87 | this->plugins_[n]->ExecutionResult(response, request_id); |
| 88 | return; |
88 | return; |
| 89 | } |
89 | } |
| 90 | } |
90 | } |
| 91 | } |
91 | } |
| 92 | 92 | ||
| 93 | void Manager::Call(std::string plugin_name, unsigned int request_id, std::string name, ArgumentListPointer arguments) |
93 | void Manager::Call(std::string plugin_name, unsigned int request_id, std::string name, ArgumentListPointer arguments) |
| 94 | { |
94 | { |
| 95 | this->io_service_.post(boost::bind(&Manager::CallHandler, this, plugin_name, request_id, name, arguments)); |
95 | this->io_service_.post(boost::bind(&Manager::CallHandler, this, plugin_name, request_id, name, arguments)); |
| 96 | } |
96 | } |
| 97 | 97 | ||
| 98 | void Manager::Execute(std::string plugin_name, unsigned int request_id, std::string code) |
98 | void Manager::Execute(std::string plugin_name, unsigned int request_id, std::string code) |
| 99 | { |
99 | { |
| 100 | this->io_service_.post(boost::bind(&Manager::ExecuteHandler, this, plugin_name, request_id, code)); |
100 | this->io_service_.post(boost::bind(&Manager::ExecuteHandler, this, plugin_name, request_id, code)); |
| 101 | } |
101 | } |
| 102 | 102 | ||
| 103 | void Manager::CallHandler(std::string plugin_name, unsigned int request_id, std::string name, ArgumentListPointer arguments) |
103 | void Manager::CallHandler(std::string plugin_name, unsigned int request_id, std::string name, ArgumentListPointer arguments) |
| 104 | { |
104 | { |
| 105 | v8::Context::Scope context_scope(this->context_); |
105 | v8::Context::Scope context_scope(this->context_); |
| 106 | v8::TryCatch try_catch; |
106 | v8::TryCatch try_catch; |
| 107 | 107 | ||
| 108 | if (this->functions_.find(name) == this->functions_.end()) |
108 | if (this->functions_.find(name) == this->functions_.end()) |
| 109 | { |
109 | { |
| 110 | this->SendResponse(plugin_name, request_id, "No such function found, " + name); |
110 | this->SendResponse(plugin_name, request_id, "No such function found, " + name); |
| 111 | return; |
111 | return; |
| 112 | } |
112 | } |
| 113 | 113 | ||
| 114 | //LOG.Debug("Call: " + name); |
114 | //LOG.Debug("Call: " + name); |
| 115 | 115 | ||
| 116 | Value result = this->functions_[name]->Call(this->context_->Global(), arguments->size(), arguments->data()); |
116 | Value result = this->functions_[name]->Call(this->context_->Global(), arguments->size(), arguments->data()); |
| 117 | 117 | ||
| 118 | if (result.IsEmpty()) |
118 | if (result.IsEmpty()) |
| 119 | { |
119 | { |
| 120 | this->SendResponse(plugin_name, request_id, this->FormatException(try_catch)); |
120 | this->SendResponse(plugin_name, request_id, this->FormatException(try_catch)); |
| 121 | return; |
121 | return; |
| 122 | } |
122 | } |
| 123 | 123 | ||
| 124 | v8::String::AsciiValue response(result->ToString()); |
124 | v8::String::AsciiValue response(result->ToString()); |
| 125 | 125 | ||
| 126 | this->SendResponse(plugin_name, request_id, std::string(*response)); |
126 | this->SendResponse(plugin_name, request_id, std::string(*response)); |
| 127 | } |
127 | } |
| 128 | 128 | ||
| 129 | void Manager::StartHandler() |
129 | void Manager::StartHandler() |
| 130 | { |
130 | { |
| 131 | v8::Handle<v8::ObjectTemplate> raw_template = v8::ObjectTemplate::New(); |
131 | v8::Handle<v8::ObjectTemplate> raw_template = v8::ObjectTemplate::New(); |
| 132 | 132 | ||
| 133 | for (unsigned int c = 0; c < this->plugins_.size(); c++) |
133 | for (unsigned int c = 0; c < this->plugins_.size(); c++) |
| 134 | { |
134 | { |
| 135 | ExportFunctionList export_functions = this->plugins_[c]->GetExportFunctions(); |
135 | ExportFunctionList export_functions = this->plugins_[c]->GetExportFunctions(); |
| 136 | 136 | ||
| 137 | for (ExportFunctionList::iterator it = export_functions.begin(); it != export_functions.end(); it++) |
137 | for (ExportFunctionList::iterator it = export_functions.begin(); it != export_functions.end(); it++) |
| 138 | { |
138 | { |
| 139 | raw_template->Set(v8::String::New(it->first.data()), v8::FunctionTemplate::New(it->second)); |
139 | raw_template->Set(v8::String::New(it->first.data()), v8::FunctionTemplate::New(it->second)); |
| 140 | } |
140 | } |
| 141 | } |
141 | } |
| 142 | 142 | ||
| 143 | this->context_ = v8::Context::New(NULL, raw_template); |
143 | this->context_ = v8::Context::New(NULL, raw_template); |
| 144 | 144 | ||
| 145 | v8::Context::Scope context_scope(this->context_); |
145 | v8::Context::Scope context_scope(this->context_); |
| 146 | 146 | ||
| 147 | for (unsigned int c = 0; c < this->plugins_.size(); c++) |
147 | for (unsigned int c = 0; c < this->plugins_.size(); c++) |
| 148 | { |
148 | { |
| 149 | this->LoadScript("plugin/" + this->plugins_[c]->GetName() + ".js"); |
149 | this->LoadScript("plugin/" + this->plugins_[c]->GetName() + ".js"); |
| 150 | } |
150 | } |
| 151 | 151 | ||
| 152 | this->object_template_ = v8::Persistent<v8::ObjectTemplate>::New(raw_template); |
152 | this->object_template_ = v8::Persistent<v8::ObjectTemplate>::New(raw_template); |
| 153 | 153 | ||
| 154 | for (unsigned int c = 0; c < this->plugins_.size(); c++) |
154 | for (unsigned int c = 0; c < this->plugins_.size(); c++) |
| 155 | { |
155 | { |
| 156 | this->plugins_[c]->InitializeDone(); |
156 | this->plugins_[c]->InitializeDone(); |
| 157 | } |
157 | } |
| 158 | 158 | ||
| 159 | LOG.Info("VM initialized."); |
159 | LOG.Info("VM initialized."); |
| 160 | } |
160 | } |
| 161 | 161 | ||
| 162 | bool Manager::ImportFunction(std::string functionname) |
162 | bool Manager::ImportFunction(std::string functionname) |
| 163 | { |
163 | { |
| 164 | v8::Context::Scope context_scope(this->context_); |
164 | v8::Context::Scope context_scope(this->context_); |
| 165 | 165 | ||
| 166 | v8::TryCatch try_catch; |
166 | v8::TryCatch try_catch; |
| 167 | 167 | ||
| 168 | v8::Handle<v8::String> process_name = v8::String::New(functionname.data()); |
168 | v8::Handle<v8::String> process_name = v8::String::New(functionname.data()); |
| 169 | v8::Handle<v8::Value> process_val = this->context_->Global()->Get(process_name); |
169 | v8::Handle<v8::Value> process_val = this->context_->Global()->Get(process_name); |
| 170 | 170 | ||
| 171 | if (!process_val->IsFunction()) |
171 | if (!process_val->IsFunction()) |
| 172 | { |
172 | { |
| 173 | LOG.Error(functionname + " was not found to be a function!"); |
173 | LOG.Error(functionname + " was not found to be a function!"); |
| 174 | return false; |
174 | return false; |
| 175 | } |
175 | } |
| 176 | 176 | ||
| 177 | v8::Handle<v8::Function> process_fun = v8::Handle<v8::Function>::Cast(process_val); |
177 | v8::Handle<v8::Function> process_fun = v8::Handle<v8::Function>::Cast(process_val); |
| 178 | 178 | ||
| 179 | this->functions_[functionname] = v8::Persistent<v8::Function>::New(process_fun); |
179 | this->functions_[functionname] = v8::Persistent<v8::Function>::New(process_fun); |
| 180 | 180 | ||
| 181 | return true; |
181 | return true; |
| 182 | } |
182 | } |
| 183 | 183 | ||
| 184 | bool Manager::LoadScript(std::string scriptname) |
184 | bool Manager::LoadScript(std::string scriptname) |
| 185 | { |
185 | { |
| - | 186 | for (unsigned int n = 0; n < this->loaded_scripts_.size(); n++) |
|
| - | 187 | { |
|
| - | 188 | if (this->loaded_scripts_[n] == scriptname) |
|
| - | 189 | { |
|
| - | 190 | LOG.Info(scriptname + " is already loaded."); |
|
| - | 191 | return true; |
|
| - | 192 | } |
|
| - | 193 | } |
|
| - | 194 | ||
| 186 | std::string path = this->script_path_ + scriptname; |
195 | std::string path = this->script_path_ + scriptname; |
| 187 | std::ifstream file(path.data()); |
196 | std::ifstream file(path.data()); |
| 188 | 197 | ||
| 189 | if (!file.is_open()) |
198 | if (!file.is_open()) |
| 190 | { |
199 | { |
| 191 | file.close(); |
200 | file.close(); |
| 192 | LOG.Warning("Could not load " + scriptname + "."); |
201 | LOG.Warning("Could not load " + scriptname + "."); |
| 193 | return false; |
202 | return false; |
| 194 | } |
203 | } |
| 195 | 204 | ||
| 196 | std::string code = ""; |
205 | std::string code = ""; |
| 197 | std::string line; |
206 | std::string line; |
| 198 | 207 | ||
| 199 | while (!file.eof()) |
208 | while (!file.eof()) |
| 200 | { |
209 | { |
| 201 | std::getline(file, line); |
210 | std::getline(file, line); |
| 202 | 211 | ||
| 203 | if (file.eof()) |
212 | if (file.eof()) |
| 204 | { |
213 | { |
| 205 | break; |
214 | break; |
| 206 | } |
215 | } |
| 207 | 216 | ||
| 208 | code += line + "\n"; |
217 | code += line + "\n"; |
| 209 | } |
218 | } |
| 210 | 219 | ||
| 211 | code += "\n"; |
220 | code += "\n"; |
| 212 | 221 | ||
| 213 | file.close(); |
222 | file.close(); |
| 214 | 223 | ||
| 215 | v8::Context::Scope context_scope(this->context_); |
224 | v8::Context::Scope context_scope(this->context_); |
| 216 | 225 | ||
| 217 | v8::TryCatch try_catch; |
226 | v8::TryCatch try_catch; |
| 218 | 227 | ||
| 219 | v8::Handle<v8::Script> script = v8::Script::Compile(v8::String::New(code.data(), code.length()), |
228 | v8::Handle<v8::Script> script = v8::Script::Compile(v8::String::New(code.data(), code.length()), |
| 220 | v8::String::New(scriptname.data(), scriptname.length())); |
229 | v8::String::New(scriptname.data(), scriptname.length())); |
| 221 | 230 | ||
| 222 | if (script.IsEmpty()) |
231 | if (script.IsEmpty()) |
| 223 | { |
232 | { |
| 224 | LOG.Error(this->FormatException(try_catch)); |
233 | LOG.Error(this->FormatException(try_catch)); |
| 225 | return false; |
234 | return false; |
| 226 | } |
235 | } |
| 227 | else |
236 | else |
| 228 | { |
237 | { |
| 229 | Value result = script->Run(); |
238 | Value result = script->Run(); |
| 230 | 239 | ||
| 231 | if (result.IsEmpty()) |
240 | if (result.IsEmpty()) |
| 232 | { |
241 | { |
| 233 | LOG.Error(this->FormatException(try_catch)); |
242 | LOG.Error(this->FormatException(try_catch)); |
| 234 | return false; |
243 | return false; |
| 235 | } |
244 | } |
| 236 | } |
245 | } |
| - | 246 | ||
| - | 247 | this->loaded_scripts_.push_back(scriptname); |
|
| 237 | 248 | ||
| 238 | LOG.Info("Loaded " + scriptname + " successfully."); |
249 | LOG.Info("Loaded " + scriptname + " successfully."); |
| 239 | return true; |
250 | return true; |
| 240 | } |
251 | } |
| 241 | 252 | ||
| 242 | void Manager::ExecuteHandler(std::string plugin_name, unsigned int request_id, std::string code) |
253 | void Manager::ExecuteHandler(std::string plugin_name, unsigned int request_id, std::string code) |
| 243 | { |
254 | { |
| 244 | std::string scriptname = "execute_" + plugin_name + ".js"; |
255 | std::string scriptname = "execute_" + plugin_name + ".js"; |
| 245 | 256 | ||
| 246 | v8::Context::Scope context_scope(this->context_); |
257 | v8::Context::Scope context_scope(this->context_); |
| 247 | 258 | ||
| 248 | v8::TryCatch try_catch; |
259 | v8::TryCatch try_catch; |
| 249 | 260 | ||
| 250 | v8::Handle<v8::Script> script = v8::Script::Compile(v8::String::New(code.data(), code.length()), |
261 | v8::Handle<v8::Script> script = v8::Script::Compile(v8::String::New(code.data(), code.length()), |
| 251 | v8::String::New(scriptname.data(), scriptname.length())); |
262 | v8::String::New(scriptname.data(), scriptname.length())); |
| 252 | 263 | ||
| 253 | if (script.IsEmpty()) |
264 | if (script.IsEmpty()) |
| 254 | { |
265 | { |
| 255 | this->SendResponse(plugin_name, request_id, this->FormatException(try_catch)); |
266 | this->SendResponse(plugin_name, request_id, this->FormatException(try_catch)); |
| 256 | } |
267 | } |
| 257 | else |
268 | else |
| 258 | { |
269 | { |
| 259 | Value result = script->Run(); |
270 | Value result = script->Run(); |
| 260 | 271 | ||
| 261 | if (result.IsEmpty()) |
272 | if (result.IsEmpty()) |
| 262 | { |
273 | { |
| 263 | this->SendResponse(plugin_name, request_id, this->FormatException(try_catch)); |
274 | this->SendResponse(plugin_name, request_id, this->FormatException(try_catch)); |
| 264 | } |
275 | } |
| 265 | else |
276 | else |
| 266 | { |
277 | { |
| 267 | v8::String::AsciiValue response(result->ToString()); |
278 | v8::String::AsciiValue response(result->ToString()); |
| 268 | 279 | ||
| 269 | this->SendResponse(plugin_name, request_id, std::string(*response)); |
280 | this->SendResponse(plugin_name, request_id, std::string(*response)); |
| 270 | } |
281 | } |
| 271 | } |
282 | } |
| 272 | } |
283 | } |
| 273 | 284 | ||
| 274 | std::string Manager::FormatException(v8::TryCatch& try_catch) |
285 | std::string Manager::FormatException(v8::TryCatch& try_catch) |
| 275 | { |
286 | { |
| 276 | v8::Context::Scope context_scope(this->context_); |
287 | v8::Context::Scope context_scope(this->context_); |
| 277 | 288 | ||
| 278 | std::string result; |
289 | std::string result; |
| 279 | 290 | ||
| 280 | v8::HandleScope handle_scope; |
291 | v8::HandleScope handle_scope; |
| 281 | v8::String::Utf8Value exception(try_catch.Exception()); |
292 | v8::String::Utf8Value exception(try_catch.Exception()); |
| 282 | v8::Handle<v8::Message> message = try_catch.Message(); |
293 | v8::Handle<v8::Message> message = try_catch.Message(); |
| 283 | 294 | ||
| 284 | if (message.IsEmpty()) |
295 | if (message.IsEmpty()) |
| 285 | { |
296 | { |
| 286 | result = *exception; |
297 | result = *exception; |
| 287 | } |
298 | } |
| 288 | else |
299 | else |
| 289 | { |
300 | { |
| 290 | v8::String::Utf8Value filename(message->GetScriptResourceName()); |
301 | v8::String::Utf8Value filename(message->GetScriptResourceName()); |
| 291 | 302 | ||
| 292 | result = std::string(*filename) + ":" + boost::lexical_cast<std::string>(message->GetLineNumber()) + ": " + std::string(*exception); |
303 | result = std::string(*filename) + ":" + boost::lexical_cast<std::string>(message->GetLineNumber()) + ": " + std::string(*exception); |
| 293 | 304 | ||
| 294 | /*v8::String::Utf8Value sourceline(message->GetSourceLine()); |
305 | /*v8::String::Utf8Value sourceline(message->GetSourceLine()); |
| 295 | string strSourceline = *sourceline; |
306 | string strSourceline = *sourceline; |
| 296 | |
307 | |
| 297 | result += strSourceline + "\n"; |
308 | result += strSourceline + "\n"; |
| 298 | |
309 | |
| 299 | string underline = rpad("", message->GetStartColumn(), ' '); |
310 | string underline = rpad("", message->GetStartColumn(), ' '); |
| 300 | underline = rpad(underline, message->GetEndColumn(), '^'); |
311 | underline = rpad(underline, message->GetEndColumn(), '^'); |
| 301 | |
312 | |
| 302 | result += underline + "\n";*/ |
313 | result += underline + "\n";*/ |
| 303 | } |
314 | } |
| 304 | 315 | ||
| 305 | return result; |
316 | return result; |
| 306 | } |
317 | } |
| 307 | 318 | ||
| 308 | }; // namespace vm |
319 | }; // namespace vm |
| 309 | }; // namespace atom |
320 | }; // namespace atom |
| 310 | 321 | ||