summaryrefslogtreecommitdiff
path: root/testsuite/tests/primops/should_run/CStringLength.hs
blob: b580e61934a1872df68ca90ab84fe5af2ae957e3 (plain)
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
{-# LANGUAGE BangPatterns #-}
{-# LANGUAGE MagicHash #-}
{-# LANGUAGE UnboxedTuples #-}

import GHC.Exts

main :: IO ()
main = do
  putStr "A: "
  print $
    I# (cstringLength# "hello_world"#)
    ==
    naiveStrlen "hello_world"# 0
  putStr "B: "
  print $
    I# (cstringLength# "aaaaaaaaaaaaa\x00b"#)
    ==
    naiveStrlen "aaaaaaaaaaaaa\x00b"# 0
  putStr "C: "
  print $
    I# (cstringLength# "cccccccccccccccccc\x00b"#)
    ==
    naiveStrlen "cccccccccccccccccc\x00b"# 0
  putStr "D: "
  print $
    I# (cstringLength# "araña\NULb"#)
    ==
    naiveStrlen "araña\NULb"# 0

naiveStrlen :: Addr# -> Int -> Int
naiveStrlen addr !n = case indexWord8OffAddr# addr 0# of
  0## -> n
  _ -> naiveStrlen (plusAddr# addr 1#) (n + 1)