Add client Key hash to shm_area
[umurmur.git] / src / sharedmemory.c
index 56e9cea8b9aee84026b8fd30f0f8e20bad37a3c7..02b27c275856e912e96630fd5b91536cd89b3774 100644 (file)
-#include "sharedmemory.h"\r
-\r
-void Sharedmemory_init(void) \r
-{\r
-\r
-  key_t key = 0x53021d79;\r
-  int server_max_clients = getIntConf(MAX_CLIENTS);  \r
-  int shmptr_size =  sizeof( shm_t  ) + (sizeof( shmclient_t ) * server_max_clients);\r
-  \r
-  //key =  ftok( ".", 'S' );  //MJP BUG this needs fixing.. also how to handle multi copys of umurmur bound to diff ports or IPs  option to pass key on cmdline?  \r
-  shmid = shmget( key, shmptr_size, IPC_CREAT | 0666 );    \r
-\r
-  Log_info("SHM_KEY: 0x%x", key );\r
-\r
-        if( ( shmptr = ( shm_t *) shmat( shmid, 0, 0 ) ) == (shm_t *) (-1) )\r
-        {\r
-                perror("shmat");\r
-                exit(1); //MJP BUG should report error and just not use shm dont exit\r
-        }\r
-        memset( shmptr, 0, shmptr_size );                                     \r
-  shmptr->umurmurd_pid = getpid();\r
-  shmptr->server_max_clients = server_max_clients;  \r
-}\r
-\r
-void Sharedmemory_update(void) \r
-{\r
-\r
-  static size_t bt_end = sizeof(  bool_t  ) * 8,   //compute once\r
-                et_end = sizeof( etimer_t ) * 3,\r
-                pa_end = sizeof(  float   ) * 4,\r
-                pc_end = sizeof( uint32_t ) * 2;\r
-                \r
- unsigned int cc = 0;\r
- client_t *client_itr = NULL;\r
\r
- memset( &shmptr->client[0], 0, sizeof( shmclient_t ) * shmptr->server_max_clients );\r
- shmptr->clientcount = Client_count();\r
\r
\r
\r
- while( Client_iterate(&client_itr) != NULL ) {\r
\r
-\r
-                                                                                              \r
-      if( client_itr->authenticated )\r
-      {\r
-        \r
-        channel_t *channel = client_itr->channel;\r
-                                                                                                                                  \r
-        strncpy( shmptr->client[cc].username, client_itr->username, 120 );\r
-        strncpy( shmptr->client[cc].ipaddress, Util_clientAddressToString( client_itr ) , 45 );\r
-        shmptr->client[cc].tcp_port = Util_clientAddressToPortTCP( client_itr );\r
-        shmptr->client[cc].udp_port = Util_clientAddressToPortUDP( client_itr );\r
-        strncpy( shmptr->client[cc].channel, channel->name, 120 );\r
-        memcpy( &shmptr->client[cc].bUDP, &client_itr->bUDP, bt_end );\r
-        memcpy( &shmptr->client[cc].lastActivity, &client_itr->lastActivity, et_end );\r
-        memcpy( &shmptr->client[cc].UDPPingAvg, &client_itr->UDPPingAvg, pa_end );\r
-        memcpy( &shmptr->client[cc].UDPPackets, &client_itr->UDPPackets, pc_end );\r
-      }  \r
-      cc++;\r
-      \r
- }\r
\r
-}\r
-\r
-void Sharedmemory_deinit(void) \r
-{\r
-  shmctl( shmid, IPC_RMID, 0 );   //Mark shmid for removal.\r
-  shmdt( shmptr );\r
-}
\ No newline at end of file
+#include "sharedmemory.h"
+
+int shm_fd;
+shm_t *shmptr = NULL;
+char shm_file_name[128];
+
+void Sharedmemory_init( int bindport, int bindport6 )
+{
+
+       int server_max_clients = getIntConf(MAX_CLIENTS);
+       int shmtotal_size =  sizeof( shm_t  ) + (sizeof( shmclient_t ) * server_max_clients);
+
+       if( !bindport )
+       {
+               bindport = getIntConf(BINDPORT);
+       }
+
+       sprintf( shm_file_name, "/umurmurd:%i", bindport );
+       Log_info("SHM_API: shm_fd=\"%s\"", shm_file_name  );
+
+       shm_fd = shm_open( shm_file_name, O_CREAT | O_RDWR, 0660 );
+       if(shm_fd == -1)
+       {
+               Log_fatal( "SHM_API: Open failed:%s\n", strerror(errno));
+               exit(EXIT_FAILURE);
+       }
+
+       if( ftruncate( shm_fd, shmtotal_size ) == -1 )
+       {
+               Sharedmemory_deinit();
+               Log_fatal( "SHM_API: ftruncate : %s\n", strerror(errno));
+               exit(EXIT_FAILURE);
+       }
+
+       shmptr = mmap( 0, shmtotal_size, PROT_READ | PROT_WRITE, MAP_SHARED, shm_fd, 0 );
+       if (shmptr == MAP_FAILED)
+       {
+               Log_fatal( "SHM_API: mmap failed : %s\n", strerror(errno));
+               exit(EXIT_FAILURE);
+       }
+
+       memset( shmptr, 0, shmtotal_size );
+
+       shmptr->umurmurd_pid = getpid();
+       shmptr->server_max_clients = server_max_clients;
+       shmptr->shmtotal_size = shmtotal_size;
+       shmptr->shmclient_size = sizeof( shmclient_t ) * shmptr->server_max_clients;
+
+}
+
+void Sharedmemory_update(void)
+{
+
+       uint64_t now;
+       unsigned int cc = 0;
+       client_t *client_itr = NULL;
+
+       memset( &shmptr->client[0], 0, shmptr->shmclient_size );
+       shmptr->clientcount = Client_count();
+
+       if( shmptr->clientcount )
+       {
+               Timer_init( &now );
+               while( Client_iterate(&client_itr) != NULL )
+               {
+                       if( client_itr->authenticated && !client_itr->shutdown_wait )
+                       {
+                               channel_t *channel = client_itr->channel;
+
+                               char* clientAddressString = Util_clientAddressToString( client_itr );
+
+                               strncpy( shmptr->client[cc].username, client_itr->username, 120 );
+                               strncpy( shmptr->client[cc].ipaddress, clientAddressString, INET6_ADDRSTRLEN - 1 );
+                               strncpy( shmptr->client[cc].channel, channel->name, 120 );
+
+                               strncpy( shmptr->client[cc].os, client_itr->os, 120 );
+                               strncpy( shmptr->client[cc].release, client_itr->release, 120 );
+                               strncpy( shmptr->client[cc].os_version, client_itr->os_version, 120 );
+
+        strncpy( shmptr->client[cc].hash, client_itr->hash, 20 );
+        
+                               shmptr->client[cc].tcp_port = Util_clientAddressToPortTCP( client_itr );
+                               shmptr->client[cc].udp_port = Util_clientAddressToPortUDP( client_itr );
+
+                               shmptr->client[cc].online_secs = ( now - client_itr->connectTime ) / 1000000LL;
+                               shmptr->client[cc].idle_secs = ( now - client_itr->idleTime ) / 1000000LL;
+
+                               shmptr->client[cc].bUDP  = client_itr->bUDP;
+                               shmptr->client[cc].deaf  = client_itr->deaf;
+                               shmptr->client[cc].mute  = client_itr->mute;
+                               shmptr->client[cc].bOpus  = client_itr->bOpus;
+                               shmptr->client[cc].self_deaf  = client_itr->self_deaf;
+                               shmptr->client[cc].self_mute  = client_itr->self_mute;
+                               shmptr->client[cc].recording  = client_itr->recording;
+                               shmptr->client[cc].authenticated  = client_itr->authenticated;
+
+                               shmptr->client[cc].availableBandwidth  = client_itr->availableBandwidth;
+
+                               shmptr->client[cc].UDPPingAvg = client_itr->UDPPingAvg;
+                               shmptr->client[cc].UDPPingVar = client_itr->UDPPingVar;
+                               shmptr->client[cc].TCPPingAvg = client_itr->TCPPingAvg;
+                               shmptr->client[cc].TCPPingVar = client_itr->TCPPingVar;
+
+                               
+
+                               shmptr->client[cc].isAdmin = client_itr->isAdmin;
+                               shmptr->client[cc].isSuppressed = client_itr->isSuppressed;
+
+                               shmptr->client[cc].UDPPackets = client_itr->UDPPackets;
+                               shmptr->client[cc].TCPPackets = client_itr->TCPPackets;
+
+                               free(clientAddressString);
+                       }
+                       cc++;
+               }
+       }
+}
+void Sharedmemory_alivetick(void)
+{
+       shmptr->alive++;
+}
+
+void Sharedmemory_deinit(void)
+{
+       close( shm_fd );
+       shm_unlink( shm_file_name );
+       shmptr->umurmurd_pid = 0;
+}