Hola,
Yo tengo Android 2.2 con la rom original y número de compilación FRF91 y nunca he tenido problemas de desconexión del WiFi cuando apago la pantalla de la Nexus One.
Para alternar entre que se desconecte o que se quede conectado el WiFi uso una aplicación gratuíta del market que se llama "Wi-Fi Lock". Tengo un acceso directo en el escritorio y al ejecutarlo cambia la "Wi-Fi sleep policity" entre "Never" o "Default".
Ignoraba que hubiese problemas con este asunto.
Un saludo.