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/datagram_service.h> 00029 #include <scalestack/kernel/core.h> 00030 #include <scalestack/kernel/module.h> 00031 #include <scalestack/kernel/option.h> 00032 #include <scalestack/network/datagram_service.h> 00033 #include <scalestack/network/ip/udp/socket_service.h> 00034 00035 using namespace std; 00036 using namespace scalestack; 00037 00038 static void options(kernel::module& module) 00039 { 00040 echo::flood::datagram_service::options(module); 00041 network::ip::udp::socket_service::peer_options(module, "localhost:7"); 00042 } 00043 00044 static void start(kernel::module& module) 00045 { 00046 auto_ptr<network::datagram_service> datagram_service( 00047 new echo::flood::datagram_service(module)); 00048 auto_ptr<network::ip::udp::socket_service> socket_service( 00049 new network::ip::udp::socket_service(module, datagram_service, 00050 "localhost", "7")); 00051 00052 for (size_t count = module.get_option("count").get_size_value(); 00053 count != 0; 00054 --count) 00055 { 00056 socket_service->add_peer_sockets(); 00057 } 00058 00059 module.get_core().add_service("echo::flood::udp", socket_service.get()); 00060 socket_service.release(); 00061 } 00062 00063 static void stop(kernel::module& module) 00064 { 00065 delete module.get_core().remove_service("echo::flood::udp"); 00066 } 00067 00068 SCALESTACK_KERNEL_MODULE(options, start, stop, NULL, NULL);
1.6.3