00001 /* 00002 * Scale Stack 00003 * 00004 * Copyright 2010 Eric Day 00005 * 00006 * Licensed under the Apache License, Version 2.0 (the "License"); 00007 * you may not use this file except in compliance with the License. 00008 * You may obtain a copy of the License at 00009 * 00010 * http://www.apache.org/licenses/LICENSE-2.0 00011 * 00012 * Unless required by applicable law or agreed to in writing, software 00013 * distributed under the License is distributed on an "AS IS" BASIS, 00014 * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied. 00015 * See the License for the specific language governing permissions and 00016 * limitations under the License. 00017 */ 00018 00024 #include "config.h" 00025 00026 #include <memory> 00027 00028 #include <scalestack/echo/flood/stream_service.h> 00029 #include <scalestack/kernel/core.h> 00030 #include <scalestack/kernel/module.h> 00031 #include <scalestack/kernel/option.h> 00032 #include <scalestack/network/ip/tcp/connection_service.h> 00033 00034 using namespace std; 00035 using namespace scalestack; 00036 00037 static void options(kernel::module& module) 00038 { 00039 echo::flood::stream_service::options(module); 00040 network::ip::tcp::connection_service::options(module, "localhost:7"); 00041 } 00042 00043 static void start(kernel::module& module) 00044 { 00045 auto_ptr<network::stream_service> stream_service( 00046 new echo::flood::stream_service(module)); 00047 auto_ptr<network::ip::tcp::connection_service> connection_service( 00048 new network::ip::tcp::connection_service(module, stream_service, "7")); 00049 00050 for (size_t count = module.get_option("count").get_size_value(); 00051 count != 0; 00052 --count) 00053 { 00054 connection_service->add_connections(); 00055 } 00056 00057 module.get_core().add_service("echo::flood::tcp", connection_service.get()); 00058 connection_service.release(); 00059 } 00060 00061 static void stop(kernel::module& module) 00062 { 00063 delete module.get_core().remove_service("echo::flood::tcp"); 00064 } 00065 00066 SCALESTACK_KERNEL_MODULE(options, start, stop, NULL, NULL);
1.6.3