#include #include #include #include #include #include "Handle.h" #include "config.h" Handle::Handle(const Handle & nh) : h(nh.h >= 0 ? dup(nh.h) : nh.h), closed(false), nonblock(false) { } Handle::~Handle() { if (h >= 0) { ::close(h); } } Handle::Handle(int nh) : h(nh), closed(false) { } int Handle::GetHandle() { return h; } ssize_t Handle::write(const void *buf, size_t count) throw (GeneralException) { ssize_t r, tr = 0; bool done, full = false; do { done = true; errno = 0; if ((r = ::write(h, buf, count)) < 0) { if ((!errno) || (errno = EAGAIN)) { // Avant de déclarer une erreur, on vérifie si ce n'est pas un // problème lié au fait qu'il n'y a plus de place libre. Cela peut // arriver si l'on agit sur un pipe ou un handle. Nous // attendons encore une fois avant de déclarer l'erreur, // grace au drapeau full. if (full) { throw IOException(GetName(), IO_WRITE, count); } else { done = false; full = true; if (nonblock) { throw TaskSwitch(); } else { sleep(1); } } } else { throw IOException(GetName(), IO_WRITE, count); } } else if (r != count) { full = done = false; ((char *)buf) += r; tr += r; } } while (!done); return r + tr; } ssize_t Handle::read(void *buf, size_t count) throw (GeneralException) { ssize_t r; errno = 0; if ((r = ::read(h, buf, count)) < 0) { if ((!errno) || (errno = EAGAIN)) { // Avant de déclarer une erreur, on vérifie si ce n'est pas un // problème lié au fait qu'il n'y a plus d'octets. throw TaskSwitch(); } else { throw IOException(GetName(), IO_READ, count); } } if (!r) { ::close(h); closed = true; } return r; } bool Handle::IsClosed(void) { return closed; } bool Handle::IsNonBlock(void) { return nonblock; } void Handle::SetNonBlock(void) { if ((h >= 0) || !nonblock) { fcntl(h, F_SETFL, O_NONBLOCK); } nonblock = true; } Handle & operator<<(Handle & h, const String & s) { char * p; p = s.to_charp(); h.write(p, strlen(p)); return h; } Handle & operator>>(Handle & h, String & s) { char t[BUFSIZ]; int i = 0, r; cerr << "Into 'Handle & >> String &'...\n"; while ((r = h.read(&(t[i]), 1)) && (i != (BUFSIZ - 1))) { // Il y a souvent des \r\n dans les sockets par exemple, // ou bien en lisant des fichiers au format MS-DOS. On // ignore le \r pour ne garder que le \n, standard sous Unix. if (t[i] == '\r') { continue; } if (t[i] == '\n') { break; } else { i++; } } t[i] = '\0'; s.set("%s", t); return h; } void Handle::close() { if (IsClosed()) { return; } if (h >= 0) { ::close(h); } closed = 1; } bool Handle::CanRead(void) { return false; } bool Handle::CanWrite(void) { return false; } String Handle::GetName(void) { return _("Bare Handle - should not happend"); }