/*
  Copyright (c) 2004-2015 by Jakob Schröter <js@camaya.net>
  This file is part of the gloox library. http://camaya.net/gloox

  This software is distributed under a license. The full license
  agreement can be found in the file LICENSE in this distribution.
  This software may not be copied, modified, sold or distributed
  other than expressed in the named license agreement.

  This software is distributed without any warranty.
*/

#include "prep.h"

#include "config.h"

#ifdef HAVE_LIBIDN
# include <stringprep.h>
# include <idna.h>
#endif

#include <cstdlib>
#include <string>

#include <string.h>

#define JID_PORTION_SIZE 1023

namespace gloox
{

  namespace prep
  {

#ifdef HAVE_LIBIDN
    /**
     * Applies a Stringprep profile to a string. This function does the actual
     * work behind nodeprep, nameprep and resourceprep.
     * @param s The string to apply the profile to.
     * @param out Contains the prepped string if prepping was successful, else untouched.
     * @param profile The Stringprep profile to apply.
     * @return Returns @b true if prepping was successful, @b false otherwise.
     */
    static bool prepare( const std::string& s, std::string& out, const Stringprep_profile* profile )
    {
      if( s.empty() || s.length() > JID_PORTION_SIZE )
        return false;

      char* p = static_cast<char*>( calloc( JID_PORTION_SIZE, sizeof( char ) ) );
      strncpy( p, s.c_str(), s.length() );
      int rc = stringprep( p, JID_PORTION_SIZE, (Stringprep_profile_flags)0, profile );
      if( rc == STRINGPREP_OK )
        out = p;
      free( p );
      return rc == STRINGPREP_OK;
    }
#endif

    bool nodeprep( const std::string& node, std::string& out )
    {
#ifdef HAVE_LIBIDN
      return prepare( node, out, stringprep_xmpp_nodeprep );
#else
      if( node.length() > JID_PORTION_SIZE )
        return false;
      out = node;
      return true;
#endif
    }

    bool nameprep( const std::string& domain, std::string& out )
    {
#ifdef HAVE_LIBIDN
      return prepare( domain, out, stringprep_nameprep );
#else
      if( domain.length() > JID_PORTION_SIZE )
        return false;
      out = domain;
      return true;
#endif
    }

    bool resourceprep( const std::string& resource, std::string& out )
    {
#ifdef HAVE_LIBIDN
      return prepare( resource, out, stringprep_xmpp_resourceprep );
#else
      if( resource.length() > JID_PORTION_SIZE )
        return false;
      out = resource;
      return true;
#endif
    }

    bool saslprep( const std::string& input, std::string& out )
    {
      #ifdef HAVE_LIBIDN
      return prepare( input, out, stringprep_saslprep );
      #else
      if( input.length() > JID_PORTION_SIZE )
        return false;
      out = input;
      return true;
      #endif
    }

    bool idna( const std::string& domain, std::string& out )
    {
#ifdef HAVE_LIBIDN
      if( domain.empty() || domain.length() > JID_PORTION_SIZE )
        return false;

      char* prepped;
      int rc = idna_to_ascii_8z( domain.c_str(), &prepped, (Idna_flags)IDNA_USE_STD3_ASCII_RULES );
      if( rc == IDNA_SUCCESS )
      {
        out = prepped;
        free( prepped );
        return true;
      }
      if( rc != IDNA_MALLOC_ERROR )
        free( prepped );
      return false;
#else
      if( domain.length() > JID_PORTION_SIZE )
        return false;
      out = domain;
      return true;
#endif
    }

  }

}