using System;
using System.Text;
using System.IO;
using System.Xml;
using System.Security.Cryptography;
using System.Collections;
using System.Runtime.InteropServices;
using System.Data;
using System.Management;
using Comm;

namespace CreateLicence
{

    /// <summary>
    /// LicenceMake 的摘要说明。
    /// </summary>
    [ClassInterfaceAttribute(ClassInterfaceType.AutoDual)]
    public class LicenceMake
    {
        internal const string TABLENAME = "Value";

        private const string PublicAndPrivateKey = "<RSAKeyValue><Modulus>wD+BfPFh06qyNMMjyIZfbSeNSVGppoH4VCR6iVTnsrWm+Czjg24EycKvQfZd9SVYHJMASHuKr0VlBCPye2REs2b9UCT3b+yLqhFrde0v7k6m6FTZe4kH4USmhsQaJstcU94LCRDM3PNfWS3r9qZ11ZtJDmrGH1zFtOJsYBWdZAM=</Modulus><Exponent>AQAB</Exponent><P>7NMGb03H7zLLLYZ71RTKte6mJ7TonQfzx6qYmRCCwBsESolDoAInJBe07iFsQ324BCMFNcNrguHaRXH9KDkmow==</P><Q>z9B9NJ3fpMsBNv+J8ynaI18gQHBFcQeRRZw6hnGvN23xeIhf+G2i9RJUPHksY+xc5LVsyMc5gcq/LzeaRD6DIQ==</Q><DP>vwHsvLTlPt2e1dcT5WZhoJ8c0AWvX4fcbPwBG+MfMcYYje1r55AU9Z1RRocahMzPMDb2qsg03FlA7KXlYx3FNQ==</DP><DQ>co281upHw7FbxBNs2auyYVkokBzTSpEfVT3fTQkRyv5B421ZaboV3/3wjT1GaPZ75l7TzoFzg8ZdSQvfb8r9wQ==</DQ><InverseQ>D3R8Pf6C+4C3iS2eqr4CqYyISuIP/YDFm2lIUFJS+jkqOZNv47/IIWXbY4nGK2qpzwjDtcFyO1DiHhOaEMqd8A==</InverseQ><D>CicQao3A0dHQ6XJowF1ASt5s2rNiVzXD7IWkpv4B4nEvaSnNthIoVJNgeAZPvfOD7LZirHFByLHMfOR9GMEZ05wgc+lYuvLaQxuvXjXFPFDjLSR3/SOCXDRcbHtDvx00q2KXSyGJIMa/TANw15rxfvKtecVJetOON+GbCMfOA0E=</D></RSAKeyValue>";

        private DataSet m_ds;
        public DataSet dataSet
        {
            get
            {
                return m_ds;
            }
        }
        
        public LicenceMake()
        {
            InitDataSet();
        }

        private void InitDataSet()
        {
            m_ds = new DataSet();
            DataTable m_dt = m_ds.Tables.Add(TABLENAME);
            m_dt.Columns.Add("DiskID", typeof(System.String));
            m_dt.Columns.Add("CompanyName", typeof(System.String));
            m_dt.Columns.Add("UserCount", typeof(System.String));
            m_dt.Columns.Add("ProductName", typeof(System.String));
            m_dt.Columns.Add("Version", typeof(System.String));
            m_dt.Columns.Add("FromDate", typeof(System.String));
            m_dt.Columns.Add("ToDate", typeof(System.String));
            m_dt.Columns.Add("ConfigueDepot", typeof(System.String));
        }

        private string RSASignData(string unSignedString)
        {
            string signedString = String.Empty;

            try
            {
                UnicodeEncoding ByteConverter = new UnicodeEncoding();
                byte[] unSignedByte = ByteConverter.GetBytes(unSignedString);

                byte[] buffer = new byte[unSignedByte.Length];

                using (MemoryStream mStream = new MemoryStream(buffer))
                {
                    mStream.Write(unSignedByte, 0, unSignedByte.Length);

                    mStream.Flush();
                    mStream.Position = 0;

                    RSACryptoServiceProvider RSAalg = new RSACryptoServiceProvider();
                    RSAalg.FromXmlString(PublicAndPrivateKey);
                    signedString = Convert.ToBase64String(RSAalg.SignData(mStream, new SHA1CryptoServiceProvider()));
                }
            }
            catch
            {
                throw;
            }
            return signedString;
        }

        public bool LicenceList(string path)
        {
            try
            {
                string directory = path + @"\Licence\";

                if (!Directory.Exists(directory)) //如果不存在就创建file文件夹
                {
                    Directory.CreateDirectory(directory);
                }
                string file = directory + "LicenceList.xml";

                //创建一个XmlDocument对象
                XmlDocument xmlDoc = new XmlDocument();
                XmlNode xmlNode;
                XmlElement elem;

                try
                {
                    xmlDoc.Load(file);
                }
                catch
                {
                    //加入一个根元素
                    elem = xmlDoc.CreateElement("", "LicenceList", "");
                    //XmlNode xmltext = xmlDoc.CreateTextNode(product);
                    //elem.AppendChild(xmltext);
                    xmlDoc.AppendChild(elem);
                }
                XmlNode root = xmlDoc.FirstChild;
                //第一个节点 ,保存记录数
                xmlNode = root.SelectSingleNode("LicenceCount");
                if (xmlNode == null)
                {
                    elem = xmlDoc.CreateElement("LicenceCount");
                    elem.SetAttribute("Value", "0");
                    root.PrependChild(elem);
                }
                xmlNode = root.SelectSingleNode("LicenceCount");
                xmlNode.Attributes["Value"].Value = (Convert.ToInt32(xmlNode.Attributes["Value"].Value) + 1).ToString();
                string licenceCount = xmlNode.Attributes["Value"].Value;
                xmlNode = xmlDoc.CreateElement("Licence" + licenceCount);
                root.AppendChild(xmlNode);

                root = root.SelectSingleNode("Licence" + licenceCount);

                for (int i = 0; i < m_ds.Tables[TABLENAME].Columns.Count; i++)
                {
                    elem = xmlDoc.CreateElement(m_ds.Tables[TABLENAME].Columns[i].ColumnName.Trim());
                    elem.SetAttribute("Value", m_ds.Tables[TABLENAME].Rows[0][i].ToString().Trim());
                    root.AppendChild(elem);
                }
                //保存这个文档到文件中
                xmlDoc.Save(file);
                return true;
            }
            catch (Exception ex)
            {
                return false;
            }
        }

        public bool CreateLicenceFile(string filePath)
        {
            FileInfo fileInfo = new FileInfo(filePath);
            if (fileInfo.Exists)
            {
                return false;
            }

            if (m_ds == null || m_ds.Tables[TABLENAME] == null || m_ds.Tables[TABLENAME].Rows.Count != 1)
            {
                return false;
            }

            XmlTextWriter writer = new XmlTextWriter(filePath, Encoding.Unicode);
            try
            {
                string unSignString = String.Empty;
                string signString = String.Empty;

                writer.Formatting = Formatting.Indented;
                writer.WriteStartElement("Licence");

                for (int i = 0; i < m_ds.Tables[TABLENAME].Columns.Count; i++)
                {
                    writer.WriteStartElement("add");
                    writer.WriteAttributeString("key", m_ds.Tables[TABLENAME].Columns[i].ColumnName.Trim());
                    writer.WriteAttributeString("value", m_ds.Tables[TABLENAME].Rows[0][i].ToString().Trim());
                    writer.WriteEndElement();
                    unSignString += m_ds.Tables[TABLENAME].Columns[i].ColumnName.Trim() + m_ds.Tables[TABLENAME].Rows[0][i].ToString().Trim();
                }

                signString = RSASignData(unSignString);

                if (String.IsNullOrEmpty(signString))
                    return false;

                writer.WriteStartElement("add");
                writer.WriteAttributeString("SignInfo", signString);
                writer.WriteEndElement();

                return true;
            }
            catch
            {
                return false;
            }
            finally
            {
                writer.Close();
            }
        }

        #region Old
        /*
        private string _RSAKey = "<RSAKeyValue><Modulus>wD+BfPFh06qyNMMjyIZfbSeNSVGppoH4VCR6iVTnsrWm+Czjg24EycKvQfZd9SVYHJMASHuKr0VlBCPye2REs2b9UCT3b+yLqhFrde0v7k6m6FTZe4kH4USmhsQaJstcU94LCRDM3PNfWS3r9qZ11ZtJDmrGH1zFtOJsYBWdZAM=</Modulus><Exponent>AQAB</Exponent></RSAKeyValue>";
        //private string _RSAKey = "<RSAKeyValue><Modulus>wD+BfPFh06qyNMMjyIZfbSeNSVGppoH4VCR6iVTnsrWm+Czjg24EycKvQfZd9SVYHJMASHuKr0VlBCPye2REs2b9UCT3b+yLqhFrde0v7k6m6FTZe4kH4USmhsQaJstcU94LCRDM3PNfWS3r9qZ11ZtJDmrGH1zFtOJsYBWdZAM=</Modulus><Exponent>AQAB</Exponent><P>7NMGb03H7zLLLYZ71RTKte6mJ7TonQfzx6qYmRCCwBsESolDoAInJBe07iFsQ324BCMFNcNrguHaRXH9KDkmow==</P><Q>z9B9NJ3fpMsBNv+J8ynaI18gQHBFcQeRRZw6hnGvN23xeIhf+G2i9RJUPHksY+xc5LVsyMc5gcq/LzeaRD6DIQ==</Q><DP>vwHsvLTlPt2e1dcT5WZhoJ8c0AWvX4fcbPwBG+MfMcYYje1r55AU9Z1RRocahMzPMDb2qsg03FlA7KXlYx3FNQ==</DP><DQ>co281upHw7FbxBNs2auyYVkokBzTSpEfVT3fTQkRyv5B421ZaboV3/3wjT1GaPZ75l7TzoFzg8ZdSQvfb8r9wQ==</DQ><InverseQ>D3R8Pf6C+4C3iS2eqr4CqYyISuIP/YDFm2lIUFJS+jkqOZNv47/IIWXbY4nGK2qpzwjDtcFyO1DiHhOaEMqd8A==</InverseQ><D>CicQao3A0dHQ6XJowF1ASt5s2rNiVzXD7IWkpv4B4nEvaSnNthIoVJNgeAZPvfOD7LZirHFByLHMfOR9GMEZ05wgc+lYuvLaQxuvXjXFPFDjLSR3/SOCXDRcbHtDvx00q2KXSyGJIMa/TANw15rxfvKtecVJetOON+GbCMfOA0E=</D></RSAKeyValue>";
        private string _SignedInfo = "";
        private const int KEYLENGTH = 1024;
        UnicodeEncoding ByteConverter = new UnicodeEncoding();

        

        /// <summary>
        /// 生成Licence序列号
        /// </summary>
        /// <param name="f_UnSignedInfo">需要签名的信息</param>
        /// <returns>已签名的信息</returns>
        private string MakeLicence(string f_UnSignedInfo)
        {
            this._SignedInfo = "";
            byte[] f_SignedInfoByte;
            if (this._RSAKey == "")
            {
                this._SignedInfo = "";
                return this._SignedInfo;
            }
            try
            {
                //// xfpan [5/20/2004]
                //CspParameters CSPParam = new CspParameters();
                //CSPParam.Flags = CspProviderFlags.UseMachineKeyStore;
                //RSACryptoServiceProvider f_RSA = new RSACryptoServiceProvider(CSPParam);
                //// xfpan [5/20/2004]
                //f_RSA.FromXmlString(this._RSAKey);
                //byte[] f_UnSignedInfoByte = ByteConverter.GetBytes(f_UnSignedInfo);
                //HashAlgorithm sha = new SHA1CryptoServiceProvider();
                //f_SignedInfoByte = f_RSA.SignData(f_UnSignedInfoByte, sha);
                //this._SignedInfo = ByteConverter.GetString(f_SignedInfoByte);

                byte[] f_UnSignedInfoByte = ByteConverter.GetBytes(f_UnSignedInfo);

                // Create a buffer for the memory stream.   
                byte[] buffer = new byte[f_UnSignedInfoByte.Length];

                // Create a MemoryStream.   
                MemoryStream mStream = new MemoryStream(buffer);

                // Write the bytes to the stream and flush it.   
                mStream.Write(f_UnSignedInfoByte, 0, f_UnSignedInfoByte.Length);

                mStream.Flush();

                // Hash and sign the data.   
                byte[] signedData = HashAndSignBytes(mStream);

                this._SignedInfo = Convert.ToBase64String(signedData);

            }
            catch
            {
                this._SignedInfo = "";
                return this._SignedInfo;
            }
            return this._SignedInfo;
        }

        /// <summary>
        /// 将信息写入文件
        /// </summary>
        /// <param name="ds">信息</param>
        /// <param name="f_LicenceFilePath">文件路径及全名</param>
        /// <returns>成功返回True ,失败返回false</returns>
        public bool CreateLicenceFile(DataSet ds, string f_LicenceFilePath)
        {
            string f_UnSignedInfo = "";
            FileInfo fi = new FileInfo(f_LicenceFilePath);
            if (fi.Exists)
            {
                return false;
            }
            try
            {
                XmlTextWriter writer = null;
                writer = new XmlTextWriter(f_LicenceFilePath, System.Text.Encoding.Unicode);
                writer.Formatting = Formatting.Indented;
                writer.WriteStartElement("Licence");
                if (ds == null || ds.Tables[_TABLENAME] == null || ds.Tables[_TABLENAME].Rows.Count != 1)
                {
                    return false;
                }
                for (int i = 0; i < ds.Tables[_TABLENAME].Columns.Count; i++)
                {
                    writer.WriteStartElement("add");
                    writer.WriteAttributeString("key", ds.Tables[_TABLENAME].Columns[i].ColumnName);
                    writer.WriteAttributeString("value", ds.Tables[_TABLENAME].Rows[0][i].ToString());
                    writer.WriteEndElement();
                    f_UnSignedInfo += ds.Tables[_TABLENAME].Columns[i].ColumnName + ds.Tables[_TABLENAME].Rows[0][i].ToString();
                }
                writer.Close();
            }
            catch
            {
                return false;
            }

            //f_UnSignedInfo = "DiskIDMA-00-15-58-E8-CC-E4";//yyb
            string f_SignedInfo = MakeLicence(f_UnSignedInfo);
            if (f_SignedInfo == "")
            {
                return false;
            }
            try
            {
                StreamWriter f_swfs = new StreamWriter(f_LicenceFilePath, true, System.Text.Encoding.Unicode);
                f_swfs.WriteLine("");
                f_swfs.WriteLine(f_SignedInfo);
                f_swfs.Close();
            }
            catch
            {
                return false;
            }
            return true;
        }

        // xfpan [5/17/2004]
        /// <summary>
        /// 将信息写入文件
        /// </summary>
        /// <param name="DiskID">硬盘序列号</param>
        /// <param name="CompanyName">公司名称</param>
        /// <param name="Version">版本号</param>
        /// <param name="DirPath">路径</param>
        /// <param name="DirPath">配置库</param>
        /// <returns>成功返回文件名,失败返回空</returns>
        public string CreateLicenceFile1(string DiskID, string CompanyName, string UserCount, string ProductName, string Version, String DirPath, string ConfigueDepot)
        {
            return CreateLicenceFile2(DiskID, CompanyName, UserCount, ProductName, Version, DateTime.MinValue, DateTime.MaxValue, DirPath, ConfigueDepot);
        }

        /// <summary>
        /// 将信息写入文件
        /// </summary>
        /// <param name="DiskID">硬盘序列号</param>
        /// <param name="CompanyName">公司名称</param>
        /// <param name="Version">版本号</param>
        /// <param name="dateFrom">起始时间</param>
        /// <param name="dateTo">截止时间</param>
        /// <param name="DirPath">路径</param>
        /// <param name="DirPath">配置库</param>
        /// <returns>成功返回文件名,失败返回空</returns>
        public string CreateLicenceFile2(String DiskID, String CompanyName, string UserCount, string ProductName, String Version, DateTime dateFrom, DateTime dateTo, String DirPath, String ConfigueDepot)
        {
            if (m_ds == null || m_ds.Tables[_TABLENAME] == null)
            {
                return "";
            }

            try
            {
                m_ds.Tables[_TABLENAME].Clear();
                DataRow m_dr = m_ds.Tables[_TABLENAME].NewRow();
                m_dr["DiskID"] = DiskID;
                m_dr["CompanyName"] = CompanyName;
                m_dr["UserCount"] = UserCount;
                m_dr["ProductName"] = ProductName;
                m_dr["Version"] = Version;
                m_dr["FromDate"] = dateFrom.ToShortDateString();
                m_dr["ToDate"] = dateTo.ToShortDateString();
                m_dr["ConfigueDepot"] = ConfigueDepot;
                m_ds.Tables[_TABLENAME].Rows.Add(m_dr);
            }
            catch
            {
                return "";
            }

            // 生成随机文件名称
            String FileName = GetFileName();
            String FilePath = DirPath + "\\" + FileName;

            bool bRet = this.CreateLicenceFile(m_ds, FilePath);
            if (bRet == false)
            {
                FileName = "";
            }
            return FileName;
        }

        // 测试使用
        
        private void WriteLog(String ErrInfos)
        {
            FileStream fs = new FileStream("c:\\ReadMe.txt", FileMode.OpenOrCreate);
            StreamWriter sw = new StreamWriter(fs);
            sw.WriteLine(ErrInfos);
            sw.Close();
        }
        

        private string GetFileName()
        {
            String FileName = DateTime.Now.ToShortDateString() + " ";
            char cTmp = 'a';
            Random ra = new Random(unchecked((int)DateTime.Now.Ticks));
            for (int i = 0; i < 8; i++)
            {
                cTmp = (char)('a' + ra.Next(0, 26));
                FileName += cTmp.ToString();
            }
            FileName += ".lic";

            return FileName;
        }
        // xfpan [5/17/2004]

        public bool CreateRSAKey(out string f_PublicKey, out string f_PrivateKey)
        {
            try
            {
                // xfpan [5/20/2004]
                CspParameters CSPParam = new CspParameters();
                CSPParam.Flags = CspProviderFlags.UseMachineKeyStore;
                RSACryptoServiceProvider f_RSA = new RSACryptoServiceProvider(CSPParam);
                // xfpan [5/20/2004]
                f_PublicKey = f_RSA.ToXmlString(false);
                f_PrivateKey = f_RSA.ToXmlString(true);
            }
            catch
            {
                f_PublicKey = "";
                f_PrivateKey = "";
                return false;
            }
            return true;
        }

        /// <summary>
        /// 已签名的信息
        /// </summary>
        public string SignedInfo
        {
            get
            {
                return _SignedInfo;
            }
        }

        /// <summary>
        /// 返回DataSet
        /// </summary>
        public DataSet dataSet
        {
            get
            {
                return m_ds;
            }
        }

        public string TABLENAME
        {
            get
            {
                return _TABLENAME;
            }
        }

        /// <summary>
        /// 
        /// </summary>
        internal string RSAKey
        {
            set
            {
                _RSAKey = value;
            }
        }

        public static string PublicKey = "<RSAKeyValue><Modulus>wD+BfPFh06qyNMMjyIZfbSeNSVGppoH4VCR6iVTnsrWm+Czjg24EycKvQfZd9SVYHJMASHuKr0VlBCPye2REs2b9UCT3b+yLqhFrde0v7k6m6FTZe4kH4USmhsQaJstcU94LCRDM3PNfWS3r9qZ11ZtJDmrGH1zFtOJsYBWdZAM=</Modulus><Exponent>AQAB</Exponent></RSAKeyValue>";

        public static byte[] HashAndSignBytes(Stream DataStream)
        {
            try
            {
                // Reset the current position in the stream to    
                // the beginning of the stream (0). RSACryptoServiceProvider   
                // can't verify the data unless the the stream position   
                // is set to the starting position of the data.   
                DataStream.Position = 0;

                // Create a new instance of RSACryptoServiceProvider using the    
                // key from RSAParameters.     
                RSACryptoServiceProvider RSAalg = new RSACryptoServiceProvider();

                RSAalg.FromXmlString(PublicAndPrivateKey);
                //RSAalg.ImportParameters(Key);

                // Hash and sign the data. Pass a new instance of SHA1CryptoServiceProvider   
                // to specify the use of SHA1 for hashing.   
                return RSAalg.SignData(DataStream, new SHA1CryptoServiceProvider());
            }
            catch (CryptographicException e)
            {
                Console.WriteLine(e.Message);

                return null;
            }
        }
        */
        #endregion

        // 获取硬盘物理序列号
        [DllImport("DiskSerialNo.dll")]
        private static extern string GetDiskSerialNo();

        // 获取网卡序列号
        [DllImport("DiskSerialNo.dll")]
        private static extern string GetMacAddr();

        /// <summary>
        /// 获取机器唯一标志
        /// </summary>
        /// <returns>机器唯一标志</returns>
        public static string GetDiskID()
        {
            string identity = String.Empty;
            try
            {
                identity = GetMacAddr().Trim();
                if (!String.IsNullOrEmpty(identity))
                {
                    identity = "MA-" + identity;
                    return identity;
                }

                identity = GetDiskSerialNo().Trim();
                if (!String.IsNullOrEmpty(identity))
                {
                    identity = "HD-" + identity;
                    return identity;
                }

                //identity = GetCPUId();
                //if (String.IsNullOrEmpty(identity))
                //{
                //    identity = "CPU-" + identity;
                //}
            }
            catch
            {
                throw;
            }
            return identity;
        }

        /// <summary>
        /// 获取本机CPUID
        /// </summary>
        private static string GetCPUId()
        {
            String HDid = String.Empty;
            //ManagementClass cimobject = new ManagementClass("Win32_Processor");
            //try
            //{
            //    ManagementObjectCollection moc = cimobject.GetInstances();
            //    foreach (ManagementObject mo in moc)
            //    {
            //        //HDid = mo["UniqueId"].ToString().Trim();
            //        //if (!String.IsNullOrEmpty(HDid))
            //        //    break;
            //    }
            //}
            //catch
            //{
            //    throw;
            //}
            return HDid;
        }
    }
}