diff options
Diffstat (limited to 'core/io/ip.cpp')
| -rw-r--r-- | core/io/ip.cpp | 270 |
1 files changed, 270 insertions, 0 deletions
diff --git a/core/io/ip.cpp b/core/io/ip.cpp new file mode 100644 index 0000000000..503a009444 --- /dev/null +++ b/core/io/ip.cpp @@ -0,0 +1,270 @@ +/*************************************************************************/ +/* ip.cpp */ +/*************************************************************************/ +/* This file is part of: */ +/* GODOT ENGINE */ +/* http://www.godotengine.org */ +/*************************************************************************/ +/* Copyright (c) 2007-2014 Juan Linietsky, Ariel Manzur. */ +/* */ +/* Permission is hereby granted, free of charge, to any person obtaining */ +/* a copy of this software and associated documentation files (the */ +/* "Software"), to deal in the Software without restriction, including */ +/* without limitation the rights to use, copy, modify, merge, publish, */ +/* distribute, sublicense, and/or sell copies of the Software, and to */ +/* permit persons to whom the Software is furnished to do so, subject to */ +/* the following conditions: */ +/* */ +/* The above copyright notice and this permission notice shall be */ +/* included in all copies or substantial portions of the Software. */ +/* */ +/* THE SOFTWARE IS PROVIDED "AS IS", WITHOUT WARRANTY OF ANY KIND, */ +/* EXPRESS OR IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF */ +/* MERCHANTABILITY, FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT.*/ +/* IN NO EVENT SHALL THE AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY */ +/* CLAIM, DAMAGES OR OTHER LIABILITY, WHETHER IN AN ACTION OF CONTRACT, */ +/* TORT OR OTHERWISE, ARISING FROM, OUT OF OR IN CONNECTION WITH THE */ +/* SOFTWARE OR THE USE OR OTHER DEALINGS IN THE SOFTWARE. */ +/*************************************************************************/ +#include "ip.h" +#include "os/thread.h" +#include "os/semaphore.h" +#include "hash_map.h" + + + +/************* RESOLVER ******************/ + + +struct _IP_ResolverPrivate { + + struct QueueItem { + + volatile IP::ResolverStatus status; + IP_Address response; + String hostname; + + void clear() { + status = IP::RESOLVER_STATUS_NONE; + response = IP_Address(); + hostname=""; + }; + + QueueItem() { + clear(); + }; + }; + + QueueItem queue[IP::RESOLVER_MAX_QUERIES]; + + IP::ResolverID find_empty_id() const { + + for(int i=0;i<IP::RESOLVER_MAX_QUERIES;i++) { + if (queue[i].status==IP::RESOLVER_STATUS_NONE) + return i; + } + return IP::RESOLVER_INVALID_ID; + } + + Semaphore *sem; + + Thread* thread; + //Semaphore* semaphore; + bool thread_abort; + + void resolve_queues() { + + for(int i=0;i<IP::RESOLVER_MAX_QUERIES;i++) { + + if (queue[i].status!=IP::RESOLVER_STATUS_WAITING) + continue; + queue[i].response=IP::get_singleton()->resolve_hostname(queue[i].hostname); + + if (queue[i].response.host==0) + queue[i].status=IP::RESOLVER_STATUS_ERROR; + else + queue[i].status=IP::RESOLVER_STATUS_DONE; + + } + } + + + static void _thread_function(void *self) { + + _IP_ResolverPrivate *ipr=(_IP_ResolverPrivate*)self; + + while(!ipr->thread_abort) { + + ipr->sem->wait(); + GLOBAL_LOCK_FUNCTION; + ipr->resolve_queues(); + + } + + } + + HashMap<String, IP_Address> cache; + +}; + + + +IP_Address IP::resolve_hostname(const String& p_hostname) { + + GLOBAL_LOCK_FUNCTION + + if (resolver->cache.has(p_hostname)) + return resolver->cache[p_hostname]; + + IP_Address res = _resolve_hostname(p_hostname); + resolver->cache[p_hostname]=res; + return res; + +} +IP::ResolverID IP::resolve_hostname_queue_item(const String& p_hostname) { + + GLOBAL_LOCK_FUNCTION + + ResolverID id = resolver->find_empty_id(); + + if (id==RESOLVER_INVALID_ID) { + WARN_PRINT("Out of resolver queries"); + return id; + } + + resolver->queue[id].hostname=p_hostname; + if (resolver->cache.has(p_hostname)) { + resolver->queue[id].response=resolver->cache[p_hostname]; + resolver->queue[id].status=IP::RESOLVER_STATUS_DONE; + } else { + resolver->queue[id].response=IP_Address(); + resolver->queue[id].status=IP::RESOLVER_STATUS_WAITING; + if (resolver->thread) + resolver->sem->post(); + else + resolver->resolve_queues(); + } + + + + + + return id; +} + +IP::ResolverStatus IP::get_resolve_item_status(ResolverID p_id) const { + + ERR_FAIL_INDEX_V(p_id,IP::RESOLVER_MAX_QUERIES,IP::RESOLVER_STATUS_NONE); + + GLOBAL_LOCK_FUNCTION; + ERR_FAIL_COND_V(resolver->queue[p_id].status==IP::RESOLVER_STATUS_NONE,IP::RESOLVER_STATUS_NONE); + + return resolver->queue[p_id].status; + +} +IP_Address IP::get_resolve_item_address(ResolverID p_id) const { + + ERR_FAIL_INDEX_V(p_id,IP::RESOLVER_MAX_QUERIES,IP_Address()); + + GLOBAL_LOCK_FUNCTION; + + if (resolver->queue[p_id].status!=IP::RESOLVER_STATUS_DONE) { + ERR_EXPLAIN("Resolve of '"+resolver->queue[p_id].hostname+"'' didn't complete yet."); + ERR_FAIL_COND_V(resolver->queue[p_id].status!=IP::RESOLVER_STATUS_DONE,IP_Address()); + } + + + return resolver->queue[p_id].response; + +} +void IP::erase_resolve_item(ResolverID p_id) { + + ERR_FAIL_INDEX(p_id,IP::RESOLVER_MAX_QUERIES); + + GLOBAL_LOCK_FUNCTION; + + resolver->queue[p_id].status=IP::RESOLVER_STATUS_DONE; + +} + + +void IP::_bind_methods() { + + ObjectTypeDB::bind_method(_MD("resolve_hostname","host"),&IP::resolve_hostname); + ObjectTypeDB::bind_method(_MD("resolve_hostname_queue_item","host"),&IP::resolve_hostname_queue_item); + ObjectTypeDB::bind_method(_MD("get_resolve_item_status","id"),&IP::get_resolve_item_status); + ObjectTypeDB::bind_method(_MD("get_resolve_item_address","id"),&IP::get_resolve_item_address); + ObjectTypeDB::bind_method(_MD("erase_resolve_item","id"),&IP::erase_resolve_item); + + BIND_CONSTANT( RESOLVER_STATUS_NONE ); + BIND_CONSTANT( RESOLVER_STATUS_WAITING ); + BIND_CONSTANT( RESOLVER_STATUS_DONE ); + BIND_CONSTANT( RESOLVER_STATUS_ERROR ); + + BIND_CONSTANT( RESOLVER_MAX_QUERIES ); + BIND_CONSTANT( RESOLVER_INVALID_ID ); + +} + + +IP*IP::singleton=NULL; + +IP* IP::get_singleton() { + + return singleton; +} + + +IP* (*IP::_create)()=NULL; + +IP* IP::create() { + + ERR_FAIL_COND_V(singleton,NULL); + ERR_FAIL_COND_V(!_create,NULL); + return _create(); +} + +IP::IP() { + + singleton=this; + resolver = memnew( _IP_ResolverPrivate ); + resolver->sem=NULL; + +#ifndef NO_THREADS + + //resolver->sem = Semaphore::create(); + + resolver->sem=NULL; + if (resolver->sem) { + resolver->thread_abort=false; + + resolver->thread = Thread::create( _IP_ResolverPrivate::_thread_function,resolver ); + + if (!resolver->thread) + memdelete(resolver->sem); //wtf + } else { + resolver->thread=NULL; + } +#else + resolver->sem = NULL; + resolver->thread=NULL; +#endif + + +} + +IP::~IP() { + +#ifndef NO_THREADS + if (resolver->thread) { + resolver->thread_abort=true; + resolver->sem->post(); + Thread::wait_to_finish(resolver->thread); + memdelete( resolver->thread ); + memdelete( resolver->sem); + } + memdelete(resolver); + +#endif + +} |