Groups | Search | Server Info | Keyboard shortcuts | Login | Register [http] [https] [nntp] [nntps]

Groups > comp.programming.threads > #6931

Re: Sum server, finished, better, without usleep [Swift]

Show all headers | View raw

On 2021-10-06, Branimir Maksimovic <branimir.maksimovic@icloud.com> wrote:
import Foundation
signal(SIGPIPE,SIG_IGN)
let s = Server();
s.start()
var factor = 4
var bval:UnsafeMutablePointer<UnsafeMutableRawPointer?>? = nil
var ende:Bool? = nil
var semacnt:Int? = nil
var accarrFD:[Int32]? = nil
class Server {
  init() {
      let task = Process()
      task.executableURL = URL(fileURLWithPath: "/usr/sbin/sysctl")
      task.arguments = ["hw.ncpu"]
      let out = Pipe()
      task.standardOutput = out
      let res: ()? = try? task.run()
      if let _ = res {
          let outputData = out.fileHandleForReading.readDataToEndOfFile()
          let s = String(decoding: outputData, as: UTF8.self)
          print("got from sysctl ",s)
          let sep = CharacterSet(charactersIn: " \n")
          let components = s.components(separatedBy: sep)
          factor = Int(components[1]) ?? 4
      }
    if (CommandLine.argc > 1) {
      tmpSvcPort = Int(CommandLine.arguments[1]) ?? 0
      if (tmpSvcPort > 0){ servicePort = String(tmpSvcPort) }
    }
    semacnt! = 0
    ende! = false
    accarrFD! = Array(repeating:Int32(0),count:0)
    bval = UnsafeMutablePointer.allocate(capacity:1)
    bval!.pointee = UnsafeMutableRawPointer.allocate(byteCount:1,alignment:1)
    print("assigning")
    bval!.pointee!.storeBytes(of:0,as: UInt8.self)
    print("assigned")
  }
  deinit {
    bval!.pointee!.deallocate()
    bval!.deallocate()
  }
  var tmpSvcPort: Int = 0
  var servicePort = "1234"
  class Worker:Thread{
    override init(){
      super.init()
      semacnt! += 1
    }
    deinit {
      semacnt! -= 1
    }
    override func main(){
      let MTU = 65536
      let buffer = UnsafeMutablePointer<CChar>.allocate(capacity:MTU)
      defer {
        buffer.deallocate()
      }
      while(true){
      /***
       * interrestingly CAS doesn't work, because of
       * some weird Apple memory scheme.
       * OSAtomicCompareAndSwapPtr: doesn't work
      */
      var val:UInt8 = 0
      repeat {
          usleep(1)
          OSMemoryBarrier()
          val = bval!.pointee!.load(as:UInt8.self)
      }while ( val == 1 || val == 2)
      OSMemoryBarrier()
      bval!.pointee!.storeBytes(of:1,as: UInt8.self)
      repeat {
          usleep(1)
          OSMemoryBarrier()
          val = bval!.pointee!.load(as:UInt8.self)
      }while ( val == 2)
      OSMemoryBarrier()
      bval!.pointee!.storeBytes(of:2,as: UInt8.self)
      print("acquired")
      var oneFD = Array(repeating:Int32(0),count:0)
      var acqCount = 0
      while (!accarrFD!.isEmpty && acqCount < factor*4){
        /// collect rest of fds
        acqCount += 1
        oneFD.append(accarrFD![accarrFD!.count-1])
        accarrFD!.remove(at:accarrFD!.count-1)
      }
      OSMemoryBarrier()
      bval!.pointee!.storeBytes(of:0,as: UInt8.self)
      print("releazed\n")
      let toproc = oneFD
      for fd in toproc{
          let flags = fcntl(fd,F_GETFL)
          let _ = fcntl(fd,F_SETFL,flags | O_NONBLOCK)
          
        let sockKqueue = kqueue()
        if sockKqueue == -1 {
          print("Error creating kqueue")
        }

        defer {close(fd);close(sockKqueue)}
        var sockKevent = kevent(
          ident: UInt(fd),
          filter: Int16(EVFILT_READ),
          flags: UInt16(EV_ADD | EV_ENABLE),
          fflags: 0,
          data: 0,
          udata: nil)
        kevent(sockKqueue, &sockKevent, 1, nil, 0, nil)
        var event = kevent()
        let status = kevent(sockKqueue, nil, 0, &event, 1, nil)
        if  status == 0 {
          print("Timeout")
          continue
        } else if status > 0 {
            if (event.flags & UInt16(EV_EOF)) == EV_EOF {
              print("The socket (\(fd)) has been closed.")
              continue
            }
            print("File descriptor: \(fd) - has \(event.data) characters for reading")
          }
          else {
            print("Error reading kevent")
            continue
          }
        let readResult = read(fd, buffer, MTU)

        if (readResult == 0) {
          continue;  // end of file
        } else if (readResult == -1) {
          print("Error reading form client(\(fd)) - \(errno)")
          continue;  // error
        } else {
          let strResult =
              String(cString: buffer)
          let cset = CharacterSet(charactersIn: " \r\n")
          let numbers = strResult.components(separatedBy: cset)
          var sum = 0
          if numbers.count > 1 {
            if numbers[0].lowercased() == "quit" {
              ende! = true
              continue
            } else {
              let num1 = Int(numbers[0]) ?? 0
              let num2 = Int(numbers[1]) ?? 0
              sum = num1 + num2
            }
          }
          print("Received form client(\(fd)): \(strResult)")
          let out = String(format:"suma %d\n",sum)
          let _ = out.withCString {
            var sockKevent = kevent(
              ident: UInt(fd),
              filter: Int16(EVFILT_WRITE),
              flags: UInt16(EV_ADD | EV_ENABLE),
              fflags: 0,
              data: 0,
              udata: nil)
            kevent(sockKqueue, &sockKevent, 1, nil, 0, nil)
            var event = kevent()
            let status = kevent(sockKqueue, nil, 0, &event, 1, nil)
            if  status == 0 {
              print("Timeout")
            } else if status > 0 {
                if (event.flags & UInt16(EV_EOF)) == EV_EOF {
                  print("The socket (\(fd)) has been closed.")
                }
                print("File descriptor: \(fd) - can write")
                write(fd, $0, out.count)
              }
              else {
                print("Error reading kevent")
              }
            }
          }
        }
      }
    }
  }
  func start() {
    print("Server starting...")

    let socketFD = socket(AF_INET6,    //Domain [AF_INET,AF_INET6, AF_UNIX]
                          SOCK_STREAM, //Type [SOCK_STREAM, SOCK_DGRAM,
                                       // SOCK_SEQPACKET, SOCK_RAW]
                          IPPROTO_TCP  //Protocol [IPPROTO_TCP, IPPROTO_SCTP,
                                       // IPPROTO_UDP, IPPROTO_DCCP]
                          )            //Return a FileDescriptor -1 = error
    if socketFD == -1 {
      print("Error creating BSD Socket")
      return
    }

    var hints = addrinfo(
      ai_flags: AI_PASSIVE,    // Assign the address of the local host to the
                               // socket structures
      ai_family: AF_UNSPEC,    // Either IPv4 or IPv6
      ai_socktype: SOCK_STREAM,// TCP
      ai_protocol: 0,
      ai_addrlen: 0,
      ai_canonname: nil,
      ai_addr: nil,
      ai_next: nil)

    var servinfo: UnsafeMutablePointer<addrinfo>? = nil
    let addrInfoResult = getaddrinfo(
      nil,                        // Any interface
      servicePort,                // The port on which will be listenend
      &hints,                     // Protocol configuration as per above
      &servinfo)

    if addrInfoResult != 0 {
      print("Error getting address info: \(errno)")
      return
    }
    setsockopt(socketFD,SOL_SOCKET,SO_REUSEADDR,servinfo!.pointee.ai_addr, socklen_t(servinfo!.pointee.ai_addrlen))
    let bindResult = bind(socketFD, servinfo!.pointee.ai_addr, socklen_t(servinfo!.pointee.ai_addrlen))

    if bindResult == -1 {
      print("Error binding socket to Address: \(errno)")
      return
    }

    let listenResult = listen(socketFD, //Socket File descriptor
                              8         // adequate
                                        // The backlog argument defines the
                                        // maximum length the queue of pending
                                        // connections may grow to
    )

    let flags = fcntl(socketFD,F_GETFL)
    let _ = fcntl(socketFD,F_SETFL,flags | O_NONBLOCK)

    let sockKqueue = kqueue()
    if sockKqueue == -1 {
        print("Error creating kqueue")
      }
      var sockKevent = kevent(
        ident: UInt(socketFD),
        filter: Int16(EVFILT_READ),
        flags: UInt16(EV_ADD | EV_ENABLE),
        fflags: 0,
        data: 0,
        udata: nil)
      kevent(sockKqueue, &sockKevent, 1, nil, 0, nil)
      var event = kevent()
      if listenResult == -1 {
      print("Error setting our socket to listen")
      return
    }
    while (!ende!) {
      var addr = sockaddr()
      var addr_len :socklen_t = 0
      while (!ende!) {
          print("About to accept")
          let status = kevent(sockKqueue, nil, 0, &event, 1, nil)
          if  status == 0 {
            print("Timeout")
            continue
          } else if status > 0 {
              if (event.flags & UInt16(EV_EOF)) == EV_EOF {
                print("The socket (\(socketFD)) has been closed.")
                continue
              }
              print("File descriptor: \(socketFD) - has \(event.data) characters for reading")
            }
            else {
              print("Error reading kevent")
              continue
            }
          let accFD = accept(socketFD, &addr, &addr_len)
          if (accFD > 0){
              print("Accepted new client with file descriptor: \(accFD)")
              var val:UInt8 = 0
              repeat {
                  usleep(1)
                  OSMemoryBarrier()
                  val = bval!.pointee!.load(as:UInt8.self)
              }while ( val == 1 || val == 2)
              OSMemoryBarrier()
              bval!.pointee!.storeBytes(of:1,as: UInt8.self)
              repeat {
                  usleep(1)
                  OSMemoryBarrier()
                  val = bval!.pointee!.load(as:UInt8.self)
              }while ( val == 2)
              OSMemoryBarrier()
              bval!.pointee!.storeBytes(of:2,as: UInt8.self)
              accarrFD!.append(accFD)
              OSMemoryBarrier()
              bval!.pointee!.storeBytes(of:0,as:UInt8.self)
              if (semacnt! < factor) {
                let handle = Worker();
                handle.start()
              }
            } else {
                print("Error accepting connection")
            }
            OSMemoryBarrier() /// barrier needed here
        }
    }
    exit(EXIT_SUCCESS)
  }

}


-- 

7-77-777
Evil Sinner!
to weak you should be meek, and you should brainfuck stronger
https://github.com/rofl0r/chaos-pp

Back to comp.programming.threads | Previous | Next — Previous in thread | Next in thread | Find similar | Unroll thread

Thread

Sum server [Swift] Branimir Maksimovic <branimir.maksimovic@icloud.com> - 2021-10-03 19:07 +0000
  Re: Sum server,improved slightly [Swift] Branimir Maksimovic <branimir.maksimovic@icloud.com> - 2021-10-04 04:30 +0000
    Re: Sum server, mutch improved [Swift] Branimir Maksimovic <branimir.maksimovic@icloud.com> - 2021-10-05 02:17 +0000
      Re: Sum server, finished [Swift] Branimir Maksimovic <branimir.maksimovic@icloud.com> - 2021-10-05 19:22 +0000
        Re: Sum server, finished, bugs corrected, accept now nonblocking as well [Swift] Branimir Maksimovic <branimir.maksimovic@icloud.com> - 2021-10-05 22:51 +0000
          Re: Sum server, finished, bugs corrected, accept now nonblocking as well, tweaked [Swift] Branimir Maksimovic <branimir.maksimovic@icloud.com> - 2021-10-06 02:49 +0000
            Re: Sum server, finished, better, without usleep [Swift] Branimir Maksimovic <branimir.maksimovic@icloud.com> - 2021-10-06 09:34 +0000
              Re: Sum server, finished, better, without usleep [Swift] Branimir Maksimovic <branimir.maksimovic@icloud.com> - 2021-10-06 17:03 +0000

csiph-web