CPunch/Laika
1
0
Fork 0
mirror of https://github.com/CPunch/Laika.git synced 2026-02-08 09:10:03 +00:00
Code Issues Projects Releases Wiki Activity

Fixed poll() path in lpolllist.c, fixed laikaM_rmvarray()

Browse Source
This commit is contained in:
CPunch
2022-01-31 13:27:12 -06:00
parent 9291dbf823
commit 6d799a7532
5 changed files with 10 additions and 10 deletions
Download Patch File Download Diff File Expand all files Collapse all files

2
lib/include/laika.h
View File

@@ -11,7 +11,7 @@
#define ARRAY_START 4
#ifdef DEBUG
#define LAIKA_DEBUG(...) printf("[~] " __VA_ARGS__);
#define LAIKA_DEBUG(...) printf("[~] " __VA_ARGS__); fflush(stdout);
#else
#define LAIKA_DEBUG(...)
#endif

2
lib/include/lmem.h
View File

@@ -16,7 +16,7 @@
/* moves array elements above indx down by numElem, removing numElem elements at indx */
#define laikaM_rmvarray(type, buf, count, indx, numElem) { \
int _i, _sz = ((count-indx)-numElem)*sizeof(type); \
int _i, _sz = ((count-indx)-numElem); \
for (_i = 0; _i < _sz; _i++) \
buf[indx+_i] = buf[indx+numElem+_i]; \
count -= numElem; \

3
lib/include/lpeer.h
View File

@@ -8,8 +8,7 @@
#include "lrsa.h"
typedef enum {
PEER_UNVERIFIED,
,
PEER_UNVERIFIED,
PEER_BOT,
PEER_CNC, /* cnc 2 cnc communication */
PEER_AUTH /* authorized peers can send commands to cnc */

7
lib/src/lpolllist.c
View File

@@ -151,6 +151,7 @@ struct sLaika_pollEvent *laikaP_poll(struct sLaika_pollList *pList, int timeout,
if (SOCKETERROR(nEvents))
LAIKA_ERROR("epoll_wait() failed!\n");
*_nevents = nEvents;
for (i = 0; i < nEvents; i++) {
/* add event to revent array */
laikaM_growarray(struct sLaika_pollEvent, pList->revents, 1, pList->reventCount, pList->reventCapacity);
@@ -166,17 +167,18 @@ struct sLaika_pollEvent *laikaP_poll(struct sLaika_pollList *pList, int timeout,
if (SOCKETERROR(nEvents))
LAIKA_ERROR("poll() failed!\n");
*_nevents = nEvents;
/* walk through the returned poll fds, if they have an event, add it to our revents array */
for (i = 0; i < pList->fdCount && nEvents > 0; i++) {
PollFD pfd = pList->fds[i];
if (pList->fds[i].revents != 0) {
/* grab socket from hashmap */
struct sLaika_socket *sock = hashmap_get(pList->sockets, &(tLaika_hashMapElem){.fd = (SOCKET)pfd.fd});
struct sLaika_hashMapElem *_sock = hashmap_get(pList->sockets, &(tLaika_hashMapElem){.fd = (SOCKET)pfd.fd});
/* insert event into revents array */
laikaM_growarray(struct sLaika_pollEvent, pList->revents, 1, pList->reventCount, pList->reventCapacity);
pList->revents[pList->reventCount++] = (struct sLaika_pollEvent){
.sock = sock,
.sock = _sock->sock,
.pollIn = pfd.revents & POLLIN,
.pollOut = pfd.revents & POLLOUT
};
@@ -187,6 +189,5 @@ struct sLaika_pollEvent *laikaP_poll(struct sLaika_pollList *pList, int timeout,
#endif
/* return revents array */
*_nevents = i;
return pList->revents;
}