mirror of
https://github.com/lucannez64/darling-lol.git
synced 2026-10-26 19:47:42 +00:00
69 lines
3.0 KiB
C++
69 lines
3.0 KiB
C++
#include <darlingserver/message.hpp>
|
|
#include <sys/wait.h>
|
|
#include <unistd.h>
|
|
#include <chrono>
|
|
#include <cstdio>
|
|
#include <thread>
|
|
#include <vector>
|
|
#include <atomic>
|
|
|
|
using DarlingServer::Message;
|
|
static bool same(const ucred& a,const ucred& b) {
|
|
return a.pid==b.pid&&a.uid==b.uid&&a.gid==b.gid;
|
|
}
|
|
static double benchmark(unsigned count) {
|
|
auto start=std::chrono::steady_clock::now();
|
|
for(unsigned i=0;i<count;++i) {Message message(1,0);}
|
|
return std::chrono::duration<double,std::nano>(std::chrono::steady_clock::now()-start).count()/count;
|
|
}
|
|
int main() {
|
|
const ucred expected={getpid(),getuid(),getgid()};
|
|
double live=benchmark(100000);
|
|
Message::initializeServerCredentials();
|
|
double cached=benchmark(100000);
|
|
std::atomic<bool> valid=true;
|
|
std::vector<std::thread> workers;
|
|
for(unsigned n=0;n<4;++n) workers.emplace_back([&]{
|
|
for(unsigned i=0;i<10000;++i) {
|
|
Message message(1,0);ucred observed;
|
|
if(!message.copyCredentialsOut(observed)||!same(observed,expected))valid=false;
|
|
}
|
|
});
|
|
for(auto& worker:workers)worker.join();
|
|
Message custom(1,0);ucred changed={expected.pid+1,expected.uid+1,expected.gid+1},observed;
|
|
custom.copyCredentialsIn(changed);
|
|
bool overrides=custom.copyCredentialsOut(observed)&&same(observed,changed);
|
|
custom.setPID(expected.pid);custom.setUID(expected.uid);custom.setGID(expected.gid);
|
|
overrides=overrides&&custom.copyCredentialsOut(observed)&&same(observed,expected);
|
|
// Fork after initialization: the child must never reuse its parent's PID.
|
|
pid_t child=fork();
|
|
if(child==0) {
|
|
Message message(1,0);ucred credentials;
|
|
const ucred current={getpid(),getuid(),getgid()};
|
|
_exit(message.copyCredentialsOut(credentials)&&same(credentials,current)?0:1);
|
|
}
|
|
int status=0;bool forkPass=child>0&&waitpid(child,&status,0)==child&&WIFEXITED(status)&&WEXITSTATUS(status)==0;
|
|
int sockets[2];
|
|
if(socketpair(AF_UNIX,SOCK_SEQPACKET,0,sockets)!=0)return 2;
|
|
int enabled=1;
|
|
if(setsockopt(sockets[1],SOL_SOCKET,SO_PASSCRED,&enabled,sizeof(enabled))!=0)return 3;
|
|
unsigned transferred=0;
|
|
for(unsigned i=0;i<200;++i) {
|
|
int pipeFDs[2];if(pipe(pipeFDs))return 4;
|
|
char pattern=42;if(write(pipeFDs[1],&pattern,1)!=1)return 5;close(pipeFDs[1]);
|
|
Message outgoing(1,0);outgoing.pushDescriptor(pipeFDs[0]);outgoing.data()[0]=7;
|
|
outgoing.rawHeader().msg_name=nullptr;outgoing.rawHeader().msg_namelen=0;
|
|
if(sendmsg(sockets[0],&outgoing.rawHeader(),0)!=1)return 6;
|
|
Message incoming(1,1);
|
|
if(recvmsg(sockets[1],&incoming.rawHeader(),0)!=1||incoming.rawHeader().msg_flags&MSG_CTRUNC)return 7;
|
|
if(!incoming.copyCredentialsOut(observed)||!same(observed,expected))return 8;
|
|
int received=incoming.extractDescriptorAtIndex(0);char value=0;
|
|
if(received<0||read(received,&value,1)!=1||value!=pattern||incoming.data()[0]!=7)return 9;
|
|
close(received);++transferred;
|
|
}
|
|
close(sockets[0]);close(sockets[1]);
|
|
bool pass=valid&&overrides&&forkPass&&transferred==200;
|
|
printf("HOST_MESSAGE pass=%d concurrent=40000 overrides=%d fork_identity=%d transferred=%u live_ns=%.1f cached_ns=%.1f\n",pass,overrides,forkPass,transferred,live,cached);
|
|
return pass?0:1;
|
|
}
|