MyWackoSite: КурсОперационныеСистемы/ПрактикумPosixThreads/PthreadLectures/CodesamplesAio

Примеры кода для асинхронного ввода-вывода

Простой асинхронный ввод из сокета.

Обратите внимание, что мы формируем запрос на чтение раньше, чем отправляем запрос HTTP.

Это позволяет нам убедиться, что все честно, т.е. запрос отрабатывает строго после того как был сформирован.

gcc -lrt -lsocket -lnsl -lresolv sockaio.c

#include <aio.h> #include <sys/types.h> #include <sys/socket.h> #include <stdio.h> #include <netdb.h> const char req[]="GET / HTTP/1.0\r\n\r\n"; int main() { int s; static struct aiocb readrq; static const struct aiocb *readrqv[2]={&readrq, NULL}; char buf[4096]; char buf2[4096]; size_t size; struct sockaddr_in addr; struct hostent *srv; s=socket(PF_INET, SOCK_STREAM, 0); srv=gethostbyname("www.nsu.ru"); addr.sin_family=AF_INET; addr.sin_port=htons(80); if (srv) { addr.sin_addr.s_addr=*((unsigned long*)srv->h_addr); } else { inet_aton("193.124.215.195", &addr.sin_addr); } if (connect(s, (struct sockaddr*)&addr, sizeof addr)) { perror("connecting to server"); exit(1); } memset(&readrq, 0, sizeof readrq); readrq.aio_fildes=s; readrq.aio_buf=buf; readrq.aio_nbytes=sizeof buf; if (aio_read(&readrq)) { perror("aio_read"); exit(1); } write(s, req, (sizeof req)-1); while(1) { aio_suspend(readrqv, 1, NULL); size=aio_return(&readrq); if (size>0) { write(1, buf, size); aio_read(&readrq); } else if (size==0) { break; } else if (size < 0 && errno!=EINPROGRESS) { perror("reading from socket"); } } }

Асинхронный ввод из сокета на сигналах

Структура программы похожа на то, как обычно пишут драйверы: основная нить инициирует первую операцию в цепочке и спит в ожидании прерывания (ну, так как написано у меня – основной цикл будет просыпаться после каждого прерывания). А обработчик прерывания уже отвечает за инициацию следующих операций в цепочке.

#include <aio.h> #include <sys/types.h> #include <sys/socket.h> #include <stdio.h> #include <netdb.h> #include <signal.h> #include <setjmp.h> const char req[]="GET / HTTP/1.0\r\n\r\n"; sigjmp_buf toexit; void sigiohandler(int signo, siginfo_t *info, void *context) { struct aiocb * req; if (signo!=SIGIO || info->si_signo!=SIGIO) return; req=(struct aiocb *)info->si_value.sival_ptr; if (aio_error(req)==0) { /* it's complete!!! */ size_t size; size=aio_return(req); if (size==0) { /* EOF */ siglongjmp(toexit, 1); } write(1, req->aio_buf, size); /* schedule next read */ aio_read(req); } } int main() { int s; static struct aiocb readrq; static const struct aiocb *readrqv[2]={&readrq, NULL}; char buf[4096]; char buf2[4096]; size_t size; struct sockaddr_in addr; struct hostent *srv; struct sigaction sigiohandleraction; sigset_t set; s=socket(PF_INET, SOCK_STREAM, 0); srv=gethostbyname("www.nsu.ru"); addr.sin_family=AF_INET; addr.sin_port=htons(80); if (srv) { addr.sin_addr.s_addr=*((unsigned long*)srv->h_addr); } else { inet_aton("193.124.215.195", &addr.sin_addr); } if (connect(s, (struct sockaddr*)&addr, sizeof addr)) { perror("connecting to server"); exit(1); } /* block SIGIO from happening at wrong time */ sigemptyset(&set); sigaddset(&set, SIGIO); sigprocmask(SIG_BLOCK, &set, NULL); /* set SIGIO handler */ memset(&sigiohandleraction, 0, sizeof sigiohandleraction); sigiohandleraction.sa_sigaction=sigiohandler; sigiohandleraction.sa_flags=SA_SIGINFO; sigiohandleraction.sa_mask=set; sigaction(SIGIO, &sigiohandleraction, NULL); /* form AIO request */ memset(&readrq, 0, sizeof readrq); readrq.aio_fildes=s; readrq.aio_buf=buf; readrq.aio_nbytes=sizeof buf; readrq.aio_sigevent.sigev_notify=SIGEV_SIGNAL; readrq.aio_sigevent.sigev_signo=SIGIO; readrq.aio_sigevent.sigev_value.sival_ptr=&readrq; if (aio_read(&readrq)) { perror("aio_read"); exit(1); } /* Everything is ready, send HTTP request */ write(s, req, (sizeof req)-1); /* main loop :) */ if (!sigsetjmp(toexit, 1)) { while(1) sigpause(SIGIO); } write(2, "That's all, folks\n", 18); }

To Do:

Не тестировалось на страницах, которые не влезают в один буфер.

Код обоих примеров надо бы подчистить.

Дурацкий vim мне табуляций понавтыкал куда не надо, и порядок инклудов нечеловеческий.