/*
 * Copyright ©2026 Naomi Alterman.  All rights reserved.  Permission is
 * hereby granted to students registered for University of Washington
 * CSE 333 for use solely during Spring Quarter 2026 for purposes of
 * the course.  No other use, copying, distribution, or modification
 * is permitted without prior written consent. Copyrights for
 * third-party components of this work must be honored.  Instructors
 * interested in reusing these course materials should contact the
 * author.
 */

#ifndef SOCKETUTIL_H_
#define SOCKETUTIL_H_

#include <arpa/inet.h>
#include <netdb.h>
#include <errno.h>
#include <sys/socket.h>
#include <sys/types.h>

// Finds a local IP address for the given port "portnum" and
// creates a socket. Binds the socket and sets it up to listen
// for client connections.
//
// On success, returns a file descriptor for the created listening
// socket. The caller is responsible for eventually close()'ing
// the socket. Also returns the network address family through the
// output parameter "sock_family" (i.e., AF_INET or AF_INET6).
//
// On failure, returns -1.
int Listen(char* portnum, int* sock_family);

// A wrapper around "read" that shields the caller from dealing
// with the ugly issues of partial reads, EINTR, EAGAIN, and so
// on.
//
// Attempts to read "readlen" bytes from the file descriptor fd
// into the buffer "buf".  Returns the number of bytes actually
// read.  On fatal error, returns -1.  If EOF is hit and no
// bytes have been read, returns 0.  Might read fewer bytes
// than requested (for instance, if we are close to the end of
// the file).
// You may assume that buf has enough space for writelen bytes.
int WrappedRead(int fd, unsigned char* buf, int readlen);

// A wrapper around "write" that shields the caller from dealing
// with the ugly issues of partial writes, EINTR, EAGAIN, and so
// on.
//
// Writes "writelen" bytes to the file descriptor fd from
// the buffer "buf".  Blocks the caller until either writelen
// bytes have been written, or an error is encountered.  Returns
// the total number of bytes written. If this number is less
// than writelen, it's because some fatal error was encountered,
// like the connection being dropped.
// You may assume that buf has enough space for writelen bytes.
int WrappedWrite(int fd, unsigned char* buf, int writelen);

#endif  // SOCKETUTIL_H_